custom-model-fields.txt 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748
  1. ===========================
  2. Writing custom model fields
  3. ===========================
  4. .. currentmodule:: django.db.models
  5. Introduction
  6. ============
  7. The :doc:`model reference </topics/db/models>` documentation explains how to use
  8. Django's standard field classes -- :class:`~django.db.models.CharField`,
  9. :class:`~django.db.models.DateField`, etc. For many purposes, those classes are
  10. all you'll need. Sometimes, though, the Django version won't meet your precise
  11. requirements, or you'll want to use a field that is entirely different from
  12. those shipped with Django.
  13. Django's built-in field types don't cover every possible database column type --
  14. only the common types, such as ``VARCHAR`` and ``INTEGER``. For more obscure
  15. column types, such as geographic polygons or even user-created types such as
  16. `PostgreSQL custom types`_, you can define your own Django ``Field`` subclasses.
  17. .. _PostgreSQL custom types: http://www.postgresql.org/docs/8.2/interactive/sql-createtype.html
  18. Alternatively, you may have a complex Python object that can somehow be
  19. serialized to fit into a standard database column type. This is another case
  20. where a ``Field`` subclass will help you use your object with your models.
  21. Our example object
  22. ------------------
  23. Creating custom fields requires a bit of attention to detail. To make things
  24. easier to follow, we'll use a consistent example throughout this document:
  25. wrapping a Python object representing the deal of cards in a hand of Bridge_.
  26. Don't worry, you don't have know how to play Bridge to follow this example.
  27. You only need to know that 52 cards are dealt out equally to four players, who
  28. are traditionally called *north*, *east*, *south* and *west*. Our class looks
  29. something like this::
  30. class Hand(object):
  31. """A hand of cards (bridge style)"""
  32. def __init__(self, north, east, south, west):
  33. # Input parameters are lists of cards ('Ah', '9s', etc)
  34. self.north = north
  35. self.east = east
  36. self.south = south
  37. self.west = west
  38. # ... (other possibly useful methods omitted) ...
  39. .. _Bridge: http://en.wikipedia.org/wiki/Contract_bridge
  40. This is just an ordinary Python class, with nothing Django-specific about it.
  41. We'd like to be able to do things like this in our models (we assume the
  42. ``hand`` attribute on the model is an instance of ``Hand``)::
  43. example = MyModel.objects.get(pk=1)
  44. print example.hand.north
  45. new_hand = Hand(north, east, south, west)
  46. example.hand = new_hand
  47. example.save()
  48. We assign to and retrieve from the ``hand`` attribute in our model just like
  49. any other Python class. The trick is to tell Django how to handle saving and
  50. loading such an object.
  51. In order to use the ``Hand`` class in our models, we **do not** have to change
  52. this class at all. This is ideal, because it means you can easily write
  53. model support for existing classes where you cannot change the source code.
  54. .. note::
  55. You might only be wanting to take advantage of custom database column
  56. types and deal with the data as standard Python types in your models;
  57. strings, or floats, for example. This case is similar to our ``Hand``
  58. example and we'll note any differences as we go along.
  59. Background theory
  60. =================
  61. Database storage
  62. ----------------
  63. The simplest way to think of a model field is that it provides a way to take a
  64. normal Python object -- string, boolean, ``datetime``, or something more
  65. complex like ``Hand`` -- and convert it to and from a format that is useful
  66. when dealing with the database (and serialization, but, as we'll see later,
  67. that falls out fairly naturally once you have the database side under control).
  68. Fields in a model must somehow be converted to fit into an existing database
  69. column type. Different databases provide different sets of valid column types,
  70. but the rule is still the same: those are the only types you have to work
  71. with. Anything you want to store in the database must fit into one of
  72. those types.
  73. Normally, you're either writing a Django field to match a particular database
  74. column type, or there's a fairly straightforward way to convert your data to,
  75. say, a string.
  76. For our ``Hand`` example, we could convert the card data to a string of 104
  77. characters by concatenating all the cards together in a pre-determined order --
  78. say, all the *north* cards first, then the *east*, *south* and *west* cards. So
  79. ``Hand`` objects can be saved to text or character columns in the database.
  80. What does a field class do?
  81. ---------------------------
  82. .. class:: Field
  83. All of Django's fields (and when we say *fields* in this document, we always
  84. mean model fields and not :doc:`form fields </ref/forms/fields>`) are subclasses
  85. of :class:`django.db.models.Field`. Most of the information that Django records
  86. about a field is common to all fields -- name, help text, uniqueness and so
  87. forth. Storing all that information is handled by ``Field``. We'll get into the
  88. precise details of what ``Field`` can do later on; for now, suffice it to say
  89. that everything descends from ``Field`` and then customizes key pieces of the
  90. class behavior.
  91. It's important to realize that a Django field class is not what is stored in
  92. your model attributes. The model attributes contain normal Python objects. The
  93. field classes you define in a model are actually stored in the ``Meta`` class
  94. when the model class is created (the precise details of how this is done are
  95. unimportant here). This is because the field classes aren't necessary when
  96. you're just creating and modifying attributes. Instead, they provide the
  97. machinery for converting between the attribute value and what is stored in the
  98. database or sent to the :doc:`serializer </topics/serialization>`.
  99. Keep this in mind when creating your own custom fields. The Django ``Field``
  100. subclass you write provides the machinery for converting between your Python
  101. instances and the database/serializer values in various ways (there are
  102. differences between storing a value and using a value for lookups, for
  103. example). If this sounds a bit tricky, don't worry -- it will become clearer in
  104. the examples below. Just remember that you will often end up creating two
  105. classes when you want a custom field:
  106. * The first class is the Python object that your users will manipulate.
  107. They will assign it to the model attribute, they will read from it for
  108. displaying purposes, things like that. This is the ``Hand`` class in our
  109. example.
  110. * The second class is the ``Field`` subclass. This is the class that knows
  111. how to convert your first class back and forth between its permanent
  112. storage form and the Python form.
  113. Writing a field subclass
  114. ========================
  115. When planning your :class:`~django.db.models.Field` subclass, first give some
  116. thought to which existing :class:`~django.db.models.Field` class your new field
  117. is most similar to. Can you subclass an existing Django field and save yourself
  118. some work? If not, you should subclass the :class:`~django.db.models.Field`
  119. class, from which everything is descended.
  120. Initializing your new field is a matter of separating out any arguments that are
  121. specific to your case from the common arguments and passing the latter to the
  122. :meth:`~django.db.models.Field.__init__` method of
  123. :class:`~django.db.models.Field` (or your parent class).
  124. In our example, we'll call our field ``HandField``. (It's a good idea to call
  125. your :class:`~django.db.models.Field` subclass ``<Something>Field``, so it's
  126. easily identifiable as a :class:`~django.db.models.Field` subclass.) It doesn't
  127. behave like any existing field, so we'll subclass directly from
  128. :class:`~django.db.models.Field`::
  129. from django.db import models
  130. class HandField(models.Field):
  131. description = "A hand of cards (bridge style)"
  132. def __init__(self, *args, **kwargs):
  133. kwargs['max_length'] = 104
  134. super(HandField, self).__init__(*args, **kwargs)
  135. Our ``HandField`` accepts most of the standard field options (see the list
  136. below), but we ensure it has a fixed length, since it only needs to hold 52
  137. card values plus their suits; 104 characters in total.
  138. .. note::
  139. Many of Django's model fields accept options that they don't do anything
  140. with. For example, you can pass both
  141. :attr:`~django.db.models.Field.editable` and
  142. :attr:`~django.db.models.Field.auto_now` to a
  143. :class:`django.db.models.DateField` and it will simply ignore the
  144. :attr:`~django.db.models.Field.editable` parameter
  145. (:attr:`~django.db.models.Field.auto_now` being set implies
  146. ``editable=False``). No error is raised in this case.
  147. This behavior simplifies the field classes, because they don't need to
  148. check for options that aren't necessary. They just pass all the options to
  149. the parent class and then don't use them later on. It's up to you whether
  150. you want your fields to be more strict about the options they select, or to
  151. use the simpler, more permissive behavior of the current fields.
  152. .. method:: Field.__init__
  153. The :meth:`~django.db.models.Field.__init__` method takes the following
  154. parameters:
  155. * :attr:`~django.db.models.Field.verbose_name`
  156. * :attr:`~django.db.models.Field.name`
  157. * :attr:`~django.db.models.Field.primary_key`
  158. * :attr:`~django.db.models.Field.max_length`
  159. * :attr:`~django.db.models.Field.unique`
  160. * :attr:`~django.db.models.Field.blank`
  161. * :attr:`~django.db.models.Field.null`
  162. * :attr:`~django.db.models.Field.db_index`
  163. * :attr:`~django.db.models.Field.rel`: Used for related fields (like
  164. :class:`ForeignKey`). For advanced use only.
  165. * :attr:`~django.db.models.Field.default`
  166. * :attr:`~django.db.models.Field.editable`
  167. * :attr:`~django.db.models.Field.serialize`: If ``False``, the field will
  168. not be serialized when the model is passed to Django's :doc:`serializers
  169. </topics/serialization>`. Defaults to ``True``.
  170. * :attr:`~django.db.models.Field.unique_for_date`
  171. * :attr:`~django.db.models.Field.unique_for_month`
  172. * :attr:`~django.db.models.Field.unique_for_year`
  173. * :attr:`~django.db.models.Field.choices`
  174. * :attr:`~django.db.models.Field.help_text`
  175. * :attr:`~django.db.models.Field.db_column`
  176. * :attr:`~django.db.models.Field.db_tablespace`: Currently only used with
  177. the Oracle backend and only for index creation. You can usually ignore
  178. this option.
  179. * :attr:`~django.db.models.Field.auto_created`: True if the field was
  180. automatically created, as for the `OneToOneField` used by model
  181. inheritance. For advanced use only.
  182. All of the options without an explanation in the above list have the same
  183. meaning they do for normal Django fields. See the :doc:`field documentation
  184. </ref/models/fields>` for examples and details.
  185. The ``SubfieldBase`` metaclass
  186. ------------------------------
  187. .. class:: django.db.models.SubfieldBase
  188. As we indicated in the introduction_, field subclasses are often needed for
  189. two reasons: either to take advantage of a custom database column type, or to
  190. handle complex Python types. Obviously, a combination of the two is also
  191. possible. If you're only working with custom database column types and your
  192. model fields appear in Python as standard Python types direct from the
  193. database backend, you don't need to worry about this section.
  194. If you're handling custom Python types, such as our ``Hand`` class, we need to
  195. make sure that when Django initializes an instance of our model and assigns a
  196. database value to our custom field attribute, we convert that value into the
  197. appropriate Python object. The details of how this happens internally are a
  198. little complex, but the code you need to write in your ``Field`` class is
  199. simple: make sure your field subclass uses a special metaclass:
  200. For example::
  201. class HandField(models.Field):
  202. description = "A hand of cards (bridge style)"
  203. __metaclass__ = models.SubfieldBase
  204. def __init__(self, *args, **kwargs):
  205. # ...
  206. This ensures that the :meth:`.to_python` method, documented below, will always
  207. be called when the attribute is initialized.
  208. ModelForms and custom fields
  209. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  210. If you use :class:`~django.db.models.SubfieldBase`, :meth:`.to_python`
  211. will be called every time an instance of the field is assigned a
  212. value. This means that whenever a value may be assigned to the field,
  213. you need to ensure that it will be of the correct datatype, or that
  214. you handle any exceptions.
  215. This is especially important if you use :doc:`ModelForms
  216. </topics/forms/modelforms>`. When saving a ModelForm, Django will use
  217. form values to instantiate model instances. However, if the cleaned
  218. form data can't be used as valid input to the field, the normal form
  219. validation process will break.
  220. Therefore, you must ensure that the form field used to represent your
  221. custom field performs whatever input validation and data cleaning is
  222. necessary to convert user-provided form input into a
  223. `to_python()`-compatible model field value. This may require writing a
  224. custom form field, and/or implementing the :meth:`.formfield` method on
  225. your field to return a form field class whose `to_python()` returns the
  226. correct datatype.
  227. Documenting your custom field
  228. -----------------------------
  229. .. attribute:: Field.description
  230. As always, you should document your field type, so users will know what it is.
  231. In addition to providing a docstring for it, which is useful for developers,
  232. you can also allow users of the admin app to see a short description of the
  233. field type via the :doc:`django.contrib.admindocs
  234. </ref/contrib/admin/admindocs>` application. To do this simply provide
  235. descriptive text in a ``description`` class attribute of your custom field. In
  236. the above example, the description displayed by the ``admindocs``
  237. application for a ``HandField`` will be 'A hand of cards (bridge style)'.
  238. Useful methods
  239. --------------
  240. Once you've created your :class:`~django.db.models.Field` subclass and set up
  241. the ``__metaclass__``, you might consider overriding a few standard methods,
  242. depending on your field's behavior. The list of methods below is in
  243. approximately decreasing order of importance, so start from the top.
  244. Custom database types
  245. ~~~~~~~~~~~~~~~~~~~~~
  246. .. method:: Field.db_type(self, connection)
  247. .. versionadded:: 1.2
  248. The ``connection`` argument was added to support multiple databases.
  249. Returns the database column data type for the :class:`~django.db.models.Field`,
  250. taking into account the connection object, and the settings associated with it.
  251. Say you've created a PostgreSQL custom type called ``mytype``. You can use this
  252. field with Django by subclassing ``Field`` and implementing the
  253. :meth:`.db_type` method, like so::
  254. from django.db import models
  255. class MytypeField(models.Field):
  256. def db_type(self, connection):
  257. return 'mytype'
  258. Once you have ``MytypeField``, you can use it in any model, just like any other
  259. ``Field`` type::
  260. class Person(models.Model):
  261. name = models.CharField(max_length=80)
  262. gender = models.CharField(max_length=1)
  263. something_else = MytypeField()
  264. If you aim to build a database-agnostic application, you should account for
  265. differences in database column types. For example, the date/time column type
  266. in PostgreSQL is called ``timestamp``, while the same column in MySQL is called
  267. ``datetime``. The simplest way to handle this in a :meth:`.db_type`
  268. method is to check the ``connection.settings_dict['ENGINE']`` attribute.
  269. For example::
  270. class MyDateField(models.Field):
  271. def db_type(self, connection):
  272. if connection.settings_dict['ENGINE'] == 'django.db.backends.mysql':
  273. return 'datetime'
  274. else:
  275. return 'timestamp'
  276. The :meth:`.db_type` method is only called by Django when the framework
  277. constructs the ``CREATE TABLE`` statements for your application -- that is,
  278. when you first create your tables. It's not called at any other time, so it can
  279. afford to execute slightly complex code, such as the
  280. ``connection.settings_dict`` check in the above example.
  281. Some database column types accept parameters, such as ``CHAR(25)``, where the
  282. parameter ``25`` represents the maximum column length. In cases like these,
  283. it's more flexible if the parameter is specified in the model rather than being
  284. hard-coded in the ``db_type()`` method. For example, it wouldn't make much
  285. sense to have a ``CharMaxlength25Field``, shown here::
  286. # This is a silly example of hard-coded parameters.
  287. class CharMaxlength25Field(models.Field):
  288. def db_type(self, connection):
  289. return 'char(25)'
  290. # In the model:
  291. class MyModel(models.Model):
  292. # ...
  293. my_field = CharMaxlength25Field()
  294. The better way of doing this would be to make the parameter specifiable at run
  295. time -- i.e., when the class is instantiated. To do that, just implement
  296. :meth:`django.db.models.Field.__init__`, like so::
  297. # This is a much more flexible example.
  298. class BetterCharField(models.Field):
  299. def __init__(self, max_length, *args, **kwargs):
  300. self.max_length = max_length
  301. super(BetterCharField, self).__init__(*args, **kwargs)
  302. def db_type(self, connection):
  303. return 'char(%s)' % self.max_length
  304. # In the model:
  305. class MyModel(models.Model):
  306. # ...
  307. my_field = BetterCharField(25)
  308. Finally, if your column requires truly complex SQL setup, return ``None`` from
  309. :meth:`.db_type`. This will cause Django's SQL creation code to skip
  310. over this field. You are then responsible for creating the column in the right
  311. table in some other way, of course, but this gives you a way to tell Django to
  312. get out of the way.
  313. Converting database values to Python objects
  314. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  315. .. method:: Field.to_python(self, value)
  316. Converts a value as returned by your database (or a serializer) to a Python
  317. object.
  318. The default implementation simply returns ``value``, for the common case in
  319. which the database backend already returns data in the correct format (as a
  320. Python string, for example).
  321. If your custom :class:`~django.db.models.Field` class deals with data structures
  322. that are more complex than strings, dates, integers or floats, then you'll need
  323. to override this method. As a general rule, the method should deal gracefully
  324. with any of the following arguments:
  325. * An instance of the correct type (e.g., ``Hand`` in our ongoing example).
  326. * A string (e.g., from a deserializer).
  327. * Whatever the database returns for the column type you're using.
  328. In our ``HandField`` class, we're storing the data as a VARCHAR field in the
  329. database, so we need to be able to process strings and ``Hand`` instances in
  330. :meth:`.to_python`::
  331. import re
  332. class HandField(models.Field):
  333. # ...
  334. def to_python(self, value):
  335. if isinstance(value, Hand):
  336. return value
  337. # The string case.
  338. p1 = re.compile('.{26}')
  339. p2 = re.compile('..')
  340. args = [p2.findall(x) for x in p1.findall(value)]
  341. return Hand(*args)
  342. Notice that we always return a ``Hand`` instance from this method. That's the
  343. Python object type we want to store in the model's attribute.
  344. **Remember:** If your custom field needs the :meth:`to_python` method to be
  345. called when it is created, you should be using `The SubfieldBase metaclass`_
  346. mentioned earlier. Otherwise :meth:`.to_python` won't be called
  347. automatically.
  348. Converting Python objects to query values
  349. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  350. .. method:: Field.get_prep_value(self, value)
  351. .. versionadded:: 1.2
  352. This method was factored out of ``get_db_prep_value()``
  353. This is the reverse of :meth:`.to_python` when working with the
  354. database backends (as opposed to serialization). The ``value``
  355. parameter is the current value of the model's attribute (a field has
  356. no reference to its containing model, so it cannot retrieve the value
  357. itself), and the method should return data in a format that has been
  358. prepared for use as a parameter in a query.
  359. This conversion should *not* include any database-specific
  360. conversions. If database-specific conversions are required, they
  361. should be made in the call to :meth:`.get_db_prep_value`.
  362. For example::
  363. class HandField(models.Field):
  364. # ...
  365. def get_prep_value(self, value):
  366. return ''.join([''.join(l) for l in (value.north,
  367. value.east, value.south, value.west)])
  368. Converting query values to database values
  369. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  370. .. method:: Field.get_db_prep_value(self, value, connection, prepared=False)
  371. .. versionadded:: 1.2
  372. The ``connection`` and ``prepared`` arguments were added to support multiple databases.
  373. Some data types (for example, dates) need to be in a specific format
  374. before they can be used by a database backend.
  375. :meth:`.get_db_prep_value` is the method where those conversions should
  376. be made. The specific connection that will be used for the query is
  377. passed as the ``connection`` parameter. This allows you to use
  378. backend-specific conversion logic if it is required.
  379. The ``prepared`` argument describes whether or not the value has
  380. already been passed through :meth:`.get_prep_value` conversions. When
  381. ``prepared`` is False, the default implementation of
  382. :meth:`.get_db_prep_value` will call :meth:`.get_prep_value` to do
  383. initial data conversions before performing any database-specific
  384. processing.
  385. .. method:: Field.get_db_prep_save(self, value, connection)
  386. .. versionadded:: 1.2
  387. The ``connection`` argument was added to support multiple databases.
  388. Same as the above, but called when the Field value must be *saved* to
  389. the database. As the default implementation just calls
  390. :meth:`.get_db_prep_value`, you shouldn't need to implement this method
  391. unless your custom field needs a special conversion when being saved
  392. that is not the same as the conversion used for normal query
  393. parameters (which is implemented by :meth:`.get_db_prep_value`).
  394. Preprocessing values before saving
  395. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  396. .. method:: Field.pre_save(self, model_instance, add)
  397. This method is called just prior to :meth:`.get_db_prep_save` and should return
  398. the value of the appropriate attribute from ``model_instance`` for this field.
  399. The attribute name is in ``self.attname`` (this is set up by
  400. :class:`~django.db.models.Field`). If the model is being saved to the database
  401. for the first time, the ``add`` parameter will be ``True``, otherwise it will be
  402. ``False``.
  403. You only need to override this method if you want to preprocess the value
  404. somehow, just before saving. For example, Django's
  405. :class:`~django.db.models.DateTimeField` uses this method to set the attribute
  406. correctly in the case of :attr:`~django.db.models.Field.auto_now` or
  407. :attr:`~django.db.models.Field.auto_now_add`.
  408. If you do override this method, you must return the value of the attribute at
  409. the end. You should also update the model's attribute if you make any changes
  410. to the value so that code holding references to the model will always see the
  411. correct value.
  412. Preparing values for use in database lookups
  413. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  414. As with value conversions, preparing a value for database lookups is a
  415. two phase process.
  416. .. method:: Field.get_prep_lookup(self, lookup_type, value)
  417. .. versionadded:: 1.2
  418. This method was factored out of ``get_db_prep_lookup()``
  419. :meth:`.get_prep_lookup` performs the first phase of lookup preparation,
  420. performing generic data validity checks
  421. Prepares the ``value`` for passing to the database when used in a lookup (a
  422. ``WHERE`` constraint in SQL). The ``lookup_type`` will be one of the valid
  423. Django filter lookups: ``exact``, ``iexact``, ``contains``, ``icontains``,
  424. ``gt``, ``gte``, ``lt``, ``lte``, ``in``, ``startswith``, ``istartswith``,
  425. ``endswith``, ``iendswith``, ``range``, ``year``, ``month``, ``day``,
  426. ``isnull``, ``search``, ``regex``, and ``iregex``.
  427. Your method must be prepared to handle all of these ``lookup_type`` values and
  428. should raise either a ``ValueError`` if the ``value`` is of the wrong sort (a
  429. list when you were expecting an object, for example) or a ``TypeError`` if
  430. your field does not support that type of lookup. For many fields, you can get
  431. by with handling the lookup types that need special handling for your field
  432. and pass the rest to the :meth:`.get_db_prep_lookup` method of the parent class.
  433. If you needed to implement ``get_db_prep_save()``, you will usually need to
  434. implement ``get_prep_lookup()``. If you don't, ``get_prep_value`` will be
  435. called by the default implementation, to manage ``exact``, ``gt``, ``gte``,
  436. ``lt``, ``lte``, ``in`` and ``range`` lookups.
  437. You may also want to implement this method to limit the lookup types that could
  438. be used with your custom field type.
  439. Note that, for ``range`` and ``in`` lookups, ``get_prep_lookup`` will receive
  440. a list of objects (presumably of the right type) and will need to convert them
  441. to a list of things of the right type for passing to the database. Most of the
  442. time, you can reuse ``get_prep_value()``, or at least factor out some common
  443. pieces.
  444. For example, the following code implements ``get_prep_lookup`` to limit the
  445. accepted lookup types to ``exact`` and ``in``::
  446. class HandField(models.Field):
  447. # ...
  448. def get_prep_lookup(self, lookup_type, value):
  449. # We only handle 'exact' and 'in'. All others are errors.
  450. if lookup_type == 'exact':
  451. return self.get_prep_value(value)
  452. elif lookup_type == 'in':
  453. return [self.get_prep_value(v) for v in value]
  454. else:
  455. raise TypeError('Lookup type %r not supported.' % lookup_type)
  456. .. method:: Field.get_db_prep_lookup(self, lookup_type, value, connection, prepared=False)
  457. .. versionadded:: 1.2
  458. The ``connection`` and ``prepared`` arguments were added to support multiple databases.
  459. Performs any database-specific data conversions required by a lookup.
  460. As with :meth:`.get_db_prep_value`, the specific connection that will
  461. be used for the query is passed as the ``connection`` parameter.
  462. The ``prepared`` argument describes whether the value has already been
  463. prepared with :meth:`.get_prep_lookup`.
  464. Specifying the form field for a model field
  465. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  466. .. method:: Field.formfield(self, form_class=forms.CharField, **kwargs)
  467. Returns the default form field to use when this field is displayed in a model.
  468. This method is called by the :class:`~django.forms.ModelForm` helper.
  469. All of the ``kwargs`` dictionary is passed directly to the form field's
  470. :meth:`~django.forms.Field__init__` method. Normally, all you need to do is
  471. set up a good default for the ``form_class`` argument and then delegate further
  472. handling to the parent class. This might require you to write a custom form
  473. field (and even a form widget). See the :doc:`forms documentation
  474. </topics/forms/index>` for information about this, and take a look at the code in
  475. :mod:`django.contrib.localflavor` for some examples of custom widgets.
  476. Continuing our ongoing example, we can write the :meth:`.formfield` method as::
  477. class HandField(models.Field):
  478. # ...
  479. def formfield(self, **kwargs):
  480. # This is a fairly standard way to set up some defaults
  481. # while letting the caller override them.
  482. defaults = {'form_class': MyFormField}
  483. defaults.update(kwargs)
  484. return super(HandField, self).formfield(**defaults)
  485. This assumes we've imported a ``MyFormField`` field class (which has its own
  486. default widget). This document doesn't cover the details of writing custom form
  487. fields.
  488. .. _helper functions: ../forms/#generating-forms-for-models
  489. .. _forms documentation: ../forms/
  490. Emulating built-in field types
  491. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  492. .. method:: Field.get_internal_type(self)
  493. Returns a string giving the name of the :class:`~django.db.models.Field`
  494. subclass we are emulating at the database level. This is used to determine the
  495. type of database column for simple cases.
  496. If you have created a :meth:`.db_type` method, you don't need to worry about
  497. :meth:`.get_internal_type` -- it won't be used much. Sometimes, though, your
  498. database storage is similar in type to some other field, so you can use that
  499. other field's logic to create the right column.
  500. For example::
  501. class HandField(models.Field):
  502. # ...
  503. def get_internal_type(self):
  504. return 'CharField'
  505. No matter which database backend we are using, this will mean that ``syncdb``
  506. and other SQL commands create the right column type for storing a string.
  507. If :meth:`.get_internal_type` returns a string that is not known to Django for
  508. the database backend you are using -- that is, it doesn't appear in
  509. ``django.db.backends.<db_name>.creation.DATA_TYPES`` -- the string will still be
  510. used by the serializer, but the default :meth:`.db_type` method will return
  511. ``None``. See the documentation of :meth:`.db_type` for reasons why this might be
  512. useful. Putting a descriptive string in as the type of the field for the
  513. serializer is a useful idea if you're ever going to be using the serializer
  514. output in some other place, outside of Django.
  515. Converting field data for serialization
  516. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  517. .. method:: Field.value_to_string(self, obj)
  518. This method is used by the serializers to convert the field into a string for
  519. output. Calling :meth:`Field._get_val_from_obj(obj)` is the best way to get the
  520. value to serialize. For example, since our ``HandField`` uses strings for its
  521. data storage anyway, we can reuse some existing conversion code::
  522. class HandField(models.Field):
  523. # ...
  524. def value_to_string(self, obj):
  525. value = self._get_val_from_obj(obj)
  526. return self.get_db_prep_value(value)
  527. Some general advice
  528. --------------------
  529. Writing a custom field can be a tricky process, particularly if you're doing
  530. complex conversions between your Python types and your database and
  531. serialization formats. Here are a couple of tips to make things go more
  532. smoothly:
  533. 1. Look at the existing Django fields (in
  534. :file:`django/db/models/fields/__init__.py`) for inspiration. Try to find
  535. a field that's similar to what you want and extend it a little bit,
  536. instead of creating an entirely new field from scratch.
  537. 2. Put a :meth:`__str__` or :meth:`__unicode__` method on the class you're
  538. wrapping up as a field. There are a lot of places where the default
  539. behavior of the field code is to call
  540. :func:`~django.utils.encoding.force_unicode` on the value. (In our
  541. examples in this document, ``value`` would be a ``Hand`` instance, not a
  542. ``HandField``). So if your :meth:`__unicode__` method automatically
  543. converts to the string form of your Python object, you can save yourself
  544. a lot of work.
  545. Writing a ``FileField`` subclass
  546. =================================
  547. In addition to the above methods, fields that deal with files have a few other
  548. special requirements which must be taken into account. The majority of the
  549. mechanics provided by ``FileField``, such as controlling database storage and
  550. retrieval, can remain unchanged, leaving subclasses to deal with the challenge
  551. of supporting a particular type of file.
  552. Django provides a ``File`` class, which is used as a proxy to the file's
  553. contents and operations. This can be subclassed to customize how the file is
  554. accessed, and what methods are available. It lives at
  555. ``django.db.models.fields.files``, and its default behavior is explained in the
  556. :doc:`file documentation </ref/files/file>`.
  557. Once a subclass of ``File`` is created, the new ``FileField`` subclass must be
  558. told to use it. To do so, simply assign the new ``File`` subclass to the special
  559. ``attr_class`` attribute of the ``FileField`` subclass.
  560. A few suggestions
  561. ------------------
  562. In addition to the above details, there are a few guidelines which can greatly
  563. improve the efficiency and readability of the field's code.
  564. 1. The source for Django's own ``ImageField`` (in
  565. ``django/db/models/fields/files.py``) is a great example of how to
  566. subclass ``FileField`` to support a particular type of file, as it
  567. incorporates all of the techniques described above.
  568. 2. Cache file attributes wherever possible. Since files may be stored in
  569. remote storage systems, retrieving them may cost extra time, or even
  570. money, that isn't always necessary. Once a file is retrieved to obtain
  571. some data about its content, cache as much of that data as possible to
  572. reduce the number of times the file must be retrieved on subsequent
  573. calls for that information.