custom-model-fields.txt 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785
  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/current/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 to 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. All of Django's fields (and when we say *fields* in this document, we always
  83. mean model fields and not :doc:`form fields </ref/forms/fields>`) are subclasses
  84. of :class:`django.db.models.Field`. Most of the information that Django records
  85. about a field is common to all fields -- name, help text, uniqueness and so
  86. forth. Storing all that information is handled by ``Field``. We'll get into the
  87. precise details of what ``Field`` can do later on; for now, suffice it to say
  88. that everything descends from ``Field`` and then customizes key pieces of the
  89. class behavior.
  90. It's important to realize that a Django field class is not what is stored in
  91. your model attributes. The model attributes contain normal Python objects. The
  92. field classes you define in a model are actually stored in the ``Meta`` class
  93. when the model class is created (the precise details of how this is done are
  94. unimportant here). This is because the field classes aren't necessary when
  95. you're just creating and modifying attributes. Instead, they provide the
  96. machinery for converting between the attribute value and what is stored in the
  97. database or sent to the :doc:`serializer </topics/serialization>`.
  98. Keep this in mind when creating your own custom fields. The Django ``Field``
  99. subclass you write provides the machinery for converting between your Python
  100. instances and the database/serializer values in various ways (there are
  101. differences between storing a value and using a value for lookups, for
  102. example). If this sounds a bit tricky, don't worry -- it will become clearer in
  103. the examples below. Just remember that you will often end up creating two
  104. classes when you want a custom field:
  105. * The first class is the Python object that your users will manipulate.
  106. They will assign it to the model attribute, they will read from it for
  107. displaying purposes, things like that. This is the ``Hand`` class in our
  108. example.
  109. * The second class is the ``Field`` subclass. This is the class that knows
  110. how to convert your first class back and forth between its permanent
  111. storage form and the Python form.
  112. Writing a field subclass
  113. ========================
  114. When planning your :class:`~django.db.models.Field` subclass, first give some
  115. thought to which existing :class:`~django.db.models.Field` class your new field
  116. is most similar to. Can you subclass an existing Django field and save yourself
  117. some work? If not, you should subclass the :class:`~django.db.models.Field`
  118. class, from which everything is descended.
  119. Initializing your new field is a matter of separating out any arguments that are
  120. specific to your case from the common arguments and passing the latter to the
  121. ``__init__()`` method of :class:`~django.db.models.Field` (or your parent
  122. class).
  123. In our example, we'll call our field ``HandField``. (It's a good idea to call
  124. your :class:`~django.db.models.Field` subclass ``<Something>Field``, so it's
  125. easily identifiable as a :class:`~django.db.models.Field` subclass.) It doesn't
  126. behave like any existing field, so we'll subclass directly from
  127. :class:`~django.db.models.Field`::
  128. from django.db import models
  129. class HandField(models.Field):
  130. description = "A hand of cards (bridge style)"
  131. def __init__(self, *args, **kwargs):
  132. kwargs['max_length'] = 104
  133. super(HandField, self).__init__(*args, **kwargs)
  134. Our ``HandField`` accepts most of the standard field options (see the list
  135. below), but we ensure it has a fixed length, since it only needs to hold 52
  136. card values plus their suits; 104 characters in total.
  137. .. note::
  138. Many of Django's model fields accept options that they don't do anything
  139. with. For example, you can pass both
  140. :attr:`~django.db.models.Field.editable` and
  141. :attr:`~django.db.models.DateField.auto_now` to a
  142. :class:`django.db.models.DateField` and it will simply ignore the
  143. :attr:`~django.db.models.Field.editable` parameter
  144. (:attr:`~django.db.models.DateField.auto_now` being set implies
  145. ``editable=False``). No error is raised in this case.
  146. This behavior simplifies the field classes, because they don't need to
  147. check for options that aren't necessary. They just pass all the options to
  148. the parent class and then don't use them later on. It's up to you whether
  149. you want your fields to be more strict about the options they select, or to
  150. use the simpler, more permissive behavior of the current fields.
  151. The ``Field.__init__()`` method takes the following parameters:
  152. * :attr:`~django.db.models.Field.verbose_name`
  153. * ``name``
  154. * :attr:`~django.db.models.Field.primary_key`
  155. * :attr:`~django.db.models.CharField.max_length`
  156. * :attr:`~django.db.models.Field.unique`
  157. * :attr:`~django.db.models.Field.blank`
  158. * :attr:`~django.db.models.Field.null`
  159. * :attr:`~django.db.models.Field.db_index`
  160. * ``rel``: Used for related fields (like :class:`ForeignKey`). For advanced
  161. use only.
  162. * :attr:`~django.db.models.Field.default`
  163. * :attr:`~django.db.models.Field.editable`
  164. * ``serialize``: If ``False``, the field will not be serialized when the model
  165. is passed to Django's :doc:`serializers </topics/serialization>`. Defaults to
  166. ``True``.
  167. * :attr:`~django.db.models.Field.unique_for_date`
  168. * :attr:`~django.db.models.Field.unique_for_month`
  169. * :attr:`~django.db.models.Field.unique_for_year`
  170. * :attr:`~django.db.models.Field.choices`
  171. * :attr:`~django.db.models.Field.help_text`
  172. * :attr:`~django.db.models.Field.db_column`
  173. * :attr:`~django.db.models.Field.db_tablespace`: Only for index creation, if the
  174. backend supports :doc:`tablespaces </topics/db/tablespaces>`. You can usually
  175. ignore this option.
  176. * :attr:`~django.db.models.Field.auto_created`: ``True`` if the field was
  177. automatically created, as for the :class:`~django.db.models.OneToOneField`
  178. used by model inheritance. For advanced use only.
  179. All of the options without an explanation in the above list have the same
  180. meaning they do for normal Django fields. See the :doc:`field documentation
  181. </ref/models/fields>` for examples and details.
  182. .. _custom-field-deconstruct-method:
  183. Field deconstruction
  184. --------------------
  185. .. versionadded:: 1.7
  186. ``deconstruct()`` is part of the migrations framework in Django 1.7 and
  187. above. If you have custom fields from previous versions they will
  188. need this method added before you can use them with migrations.
  189. The counterpoint to writing your ``__init__()`` method is writing the
  190. ``deconstruct()`` method. This method tells Django how to take an instance
  191. of your new field and reduce it to a serialized form - in particular, what
  192. arguments to pass to ``__init__()`` to re-create it.
  193. If you haven't added any extra options on top of the field you inherited from,
  194. then there's no need to write a new ``deconstruct()`` method. If, however,
  195. you're, changing the arguments passed in ``__init__()`` (like we are in
  196. ``HandField``), you'll need to supplement the values being passed.
  197. The contract of ``deconstruct()`` is simple; it returns a tuple of four items:
  198. the field's attribute name, the full import path of the field class, the
  199. positional arguments (as a list), and the keyword arguments (as a dict). Note
  200. this is different from the ``deconstruct()`` method :ref:`for custom classes
  201. <custom-deconstruct-method>` which returns a tuple of three things.
  202. As a custom field author, you don't need to care about the first two values;
  203. the base ``Field`` class has all the code to work out the field's attribute
  204. name and import path. You do, however, have to care about the positional
  205. and keyword arguments, as these are likely the things you are changing.
  206. For example, in our ``HandField`` class we're always forcibly setting
  207. max_length in ``__init__()``. The ``deconstruct()`` method on the base ``Field``
  208. class will see this and try to return it in the keyword arguments; thus,
  209. we can drop it from the keyword arguments for readability::
  210. from django.db import models
  211. class HandField(models.Field):
  212. def __init__(self, *args, **kwargs):
  213. kwargs['max_length'] = 104
  214. super(HandField, self).__init__(*args, **kwargs)
  215. def deconstruct(self):
  216. name, path, args, kwargs = super(HandField, self).deconstruct()
  217. del kwargs["max_length"]
  218. return name, path, args, kwargs
  219. If you add a new keyword argument, you need to write code to put its value
  220. into ``kwargs`` yourself::
  221. from django.db import models
  222. class CommaSepField(models.Field):
  223. "Implements comma-separated storage of lists"
  224. def __init__(self, separator=",", *args, **kwargs):
  225. self.separator = separator
  226. super(CommaSepField, self).__init__(*args, **kwargs)
  227. def deconstruct(self):
  228. name, path, args, kwargs = super(CommaSepField, self).deconstruct()
  229. # Only include kwarg if it's not the default
  230. if self.separator != ",":
  231. kwargs['separator'] = self.separator
  232. return name, path, args, kwargs
  233. More complex examples are beyond the scope of this document, but remember -
  234. for any configuration of your Field instance, ``deconstruct()`` must return
  235. arguments that you can pass to ``__init__`` to reconstruct that state.
  236. Pay extra attention if you set new default values for arguments in the
  237. ``Field`` superclass; you want to make sure they're always included, rather
  238. than disappearing if they take on the old default value.
  239. In addition, try to avoid returning values as positional arguments; where
  240. possible, return values as keyword arguments for maximum future compatibility.
  241. Of course, if you change the names of things more often than their position
  242. in the constructor's argument list, you might prefer positional, but bear in
  243. mind that people will be reconstructing your field from the serialized version
  244. for quite a while (possibly years), depending how long your migrations live for.
  245. You can see the results of deconstruction by looking in migrations that include
  246. the field, and you can test deconstruction in unit tests by just deconstructing
  247. and reconstructing the field::
  248. name, path, args, kwargs = my_field_instance.deconstruct()
  249. new_instance = MyField(*args, **kwargs)
  250. self.assertEqual(my_field_instance.some_attribute, new_instance.some_attribute)
  251. Documenting your custom field
  252. -----------------------------
  253. As always, you should document your field type, so users will know what it is.
  254. In addition to providing a docstring for it, which is useful for developers,
  255. you can also allow users of the admin app to see a short description of the
  256. field type via the :doc:`django.contrib.admindocs
  257. </ref/contrib/admin/admindocs>` application. To do this simply provide
  258. descriptive text in a :attr:`~Field.description` class attribute of your custom
  259. field. In the above example, the description displayed by the ``admindocs``
  260. application for a ``HandField`` will be 'A hand of cards (bridge style)'.
  261. In the :mod:`django.contrib.admindocs` display, the field description is
  262. interpolated with ``field.__dict__`` which allows the description to
  263. incorporate arguments of the field. For example, the description for
  264. :class:`~django.db.models.CharField` is::
  265. description = _("String (up to %(max_length)s)")
  266. Useful methods
  267. --------------
  268. Once you've created your :class:`~django.db.models.Field` subclass and set up
  269. the ``__metaclass__``, you might consider overriding a few standard methods,
  270. depending on your field's behavior. The list of methods below is in
  271. approximately decreasing order of importance, so start from the top.
  272. .. _custom-database-types:
  273. Custom database types
  274. ~~~~~~~~~~~~~~~~~~~~~
  275. Say you've created a PostgreSQL custom type called ``mytype``. You can
  276. subclass ``Field`` and implement the :meth:`~Field.db_type` method, like so::
  277. from django.db import models
  278. class MytypeField(models.Field):
  279. def db_type(self, connection):
  280. return 'mytype'
  281. Once you have ``MytypeField``, you can use it in any model, just like any other
  282. ``Field`` type::
  283. class Person(models.Model):
  284. name = models.CharField(max_length=80)
  285. something_else = MytypeField()
  286. If you aim to build a database-agnostic application, you should account for
  287. differences in database column types. For example, the date/time column type
  288. in PostgreSQL is called ``timestamp``, while the same column in MySQL is called
  289. ``datetime``. The simplest way to handle this in a :meth:`~Field.db_type`
  290. method is to check the ``connection.settings_dict['ENGINE']`` attribute.
  291. For example::
  292. class MyDateField(models.Field):
  293. def db_type(self, connection):
  294. if connection.settings_dict['ENGINE'] == 'django.db.backends.mysql':
  295. return 'datetime'
  296. else:
  297. return 'timestamp'
  298. The :meth:`~Field.db_type` method is called by Django when the framework
  299. constructs the ``CREATE TABLE`` statements for your application -- that is,
  300. when you first create your tables. It is also called when constructing a
  301. ``WHERE`` clause that includes the model field -- that is, when you retrieve data
  302. using QuerySet methods like ``get()``, ``filter()``, and ``exclude()`` and have
  303. the model field as an argument. It's not called at any other time, so it can afford to
  304. execute slightly complex code, such as the ``connection.settings_dict`` check in
  305. the above example.
  306. Some database column types accept parameters, such as ``CHAR(25)``, where the
  307. parameter ``25`` represents the maximum column length. In cases like these,
  308. it's more flexible if the parameter is specified in the model rather than being
  309. hard-coded in the ``db_type()`` method. For example, it wouldn't make much
  310. sense to have a ``CharMaxlength25Field``, shown here::
  311. # This is a silly example of hard-coded parameters.
  312. class CharMaxlength25Field(models.Field):
  313. def db_type(self, connection):
  314. return 'char(25)'
  315. # In the model:
  316. class MyModel(models.Model):
  317. # ...
  318. my_field = CharMaxlength25Field()
  319. The better way of doing this would be to make the parameter specifiable at run
  320. time -- i.e., when the class is instantiated. To do that, just implement
  321. ``Field.__init__()``, like so::
  322. # This is a much more flexible example.
  323. class BetterCharField(models.Field):
  324. def __init__(self, max_length, *args, **kwargs):
  325. self.max_length = max_length
  326. super(BetterCharField, self).__init__(*args, **kwargs)
  327. def db_type(self, connection):
  328. return 'char(%s)' % self.max_length
  329. # In the model:
  330. class MyModel(models.Model):
  331. # ...
  332. my_field = BetterCharField(25)
  333. Finally, if your column requires truly complex SQL setup, return ``None`` from
  334. :meth:`.db_type`. This will cause Django's SQL creation code to skip
  335. over this field. You are then responsible for creating the column in the right
  336. table in some other way, of course, but this gives you a way to tell Django to
  337. get out of the way.
  338. .. _converting-values-to-python-objects:
  339. Converting values to Python objects
  340. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  341. .. versionchanged:: 1.8
  342. Historically, Django provided a metaclass called ``SubfieldBase`` which
  343. always called :meth:`~Field.to_python` on assignment. This did not play
  344. nicely with custom database transformations, aggregation, or values
  345. queries, so it has been replaced with :meth:`~Field.from_db_value`.
  346. If your custom :class:`~Field` class deals with data structures that are more
  347. complex than strings, dates, integers, or floats, then you may need to override
  348. :meth:`~Field.from_db_value` and :meth:`~Field.to_python`.
  349. If present for the field subclass, ``from_db_value()`` will be called in all
  350. circumstances when the data is loaded from the database, including in
  351. aggregates and :meth:`~django.db.models.query.QuerySet.values` calls.
  352. ``to_python()`` is called by deserialization and during the
  353. :meth:`~django.db.models.Model.clean` method used from forms.
  354. As a general rule, ``to_python()`` should deal gracefully with any of the
  355. following arguments:
  356. * An instance of the correct type (e.g., ``Hand`` in our ongoing example).
  357. * A string
  358. * ``None`` (if the field allows ``null=True``)
  359. In our ``HandField`` class, we're storing the data as a VARCHAR field in the
  360. database, so we need to be able to process strings and ``None`` in the
  361. ``from_db_value()``. In ``to_python()``, we need to also handle ``Hand``
  362. instances::
  363. import re
  364. from django.core.exceptions import ValidationError
  365. from django.db import models
  366. def parse_hand(hand_string):
  367. """Takes a string of cards and splits into a full hand."""
  368. p1 = re.compile('.{26}')
  369. p2 = re.compile('..')
  370. args = [p2.findall(x) for x in p1.findall(hand_string)]
  371. if len(args) != 4:
  372. raise ValidationError("Invalid input for a Hand instance")
  373. return Hand(*args)
  374. class HandField(models.Field):
  375. # ...
  376. def from_db_value(self, value, connection):
  377. if value is None:
  378. return value
  379. return parse_hand(value)
  380. def to_python(self, value):
  381. if isinstance(value, Hand):
  382. return value
  383. if value is None:
  384. return value
  385. return parse_hand(value)
  386. Notice that we always return a ``Hand`` instance from these methods. That's the
  387. Python object type we want to store in the model's attribute.
  388. For ``to_python()``, if anything goes wrong during value conversion, you should
  389. raise a :exc:`~django.core.exceptions.ValidationError` exception.
  390. .. _converting-python-objects-to-query-values:
  391. Converting Python objects to query values
  392. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  393. Since using a database requires conversion in both ways, if you override
  394. :meth:`~Field.to_python` you also have to override :meth:`~Field.get_prep_value`
  395. to convert Python objects back to query values.
  396. For example::
  397. class HandField(models.Field):
  398. # ...
  399. def get_prep_value(self, value):
  400. return ''.join([''.join(l) for l in (value.north,
  401. value.east, value.south, value.west)])
  402. .. warning::
  403. If your custom field uses the ``CHAR``, ``VARCHAR`` or ``TEXT``
  404. types for MySQL, you must make sure that :meth:`.get_prep_value`
  405. always returns a string type. MySQL performs flexible and unexpected
  406. matching when a query is performed on these types and the provided
  407. value is an integer, which can cause queries to include unexpected
  408. objects in their results. This problem cannot occur if you always
  409. return a string type from :meth:`.get_prep_value`.
  410. .. _converting-query-values-to-database-values:
  411. Converting query values to database values
  412. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  413. Some data types (for example, dates) need to be in a specific format
  414. before they can be used by a database backend.
  415. :meth:`~Field.get_db_prep_value` is the method where those conversions should
  416. be made. The specific connection that will be used for the query is
  417. passed as the ``connection`` parameter. This allows you to use
  418. backend-specific conversion logic if it is required.
  419. For example, Django uses the following method for its
  420. :class:`BinaryField`::
  421. def get_db_prep_value(self, value, connection, prepared=False):
  422. value = super(BinaryField, self).get_db_prep_value(value, connection, prepared)
  423. if value is not None:
  424. return connection.Database.Binary(value)
  425. return value
  426. In case your custom field needs a special conversion when being saved that is
  427. not the same as the conversion used for normal query parameters, you can
  428. override :meth:`~Field.get_db_prep_save`.
  429. .. _preprocessing-values-before-saving:
  430. Preprocessing values before saving
  431. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  432. If you want to preprocess the value just before saving, you can use
  433. :meth:`~Field.pre_save`. For example, Django's
  434. :class:`~django.db.models.DateTimeField` uses this method to set the attribute
  435. correctly in the case of :attr:`~django.db.models.DateField.auto_now` or
  436. :attr:`~django.db.models.DateField.auto_now_add`.
  437. If you do override this method, you must return the value of the attribute at
  438. the end. You should also update the model's attribute if you make any changes
  439. to the value so that code holding references to the model will always see the
  440. correct value.
  441. .. _preparing-values-for-use-in-database-lookups:
  442. Preparing values for use in database lookups
  443. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  444. As with value conversions, preparing a value for database lookups is a
  445. two phase process.
  446. :meth:`.get_prep_lookup` performs the first phase of lookup preparation:
  447. type conversion and data validation.
  448. Prepares the ``value`` for passing to the database when used in a lookup (a
  449. ``WHERE`` constraint in SQL). The ``lookup_type`` parameter will be one of the
  450. valid Django filter lookups: ``exact``, ``iexact``, ``contains``, ``icontains``,
  451. ``gt``, ``gte``, ``lt``, ``lte``, ``in``, ``startswith``, ``istartswith``,
  452. ``endswith``, ``iendswith``, ``range``, ``year``, ``month``, ``day``,
  453. ``isnull``, ``search``, ``regex``, and ``iregex``.
  454. .. versionadded:: 1.7
  455. If you are using :doc:`Custom lookups </howto/custom-lookups>` the
  456. ``lookup_type`` can be any ``lookup_name`` used by the project's custom
  457. lookups.
  458. Your method must be prepared to handle all of these ``lookup_type`` values and
  459. should raise either a ``ValueError`` if the ``value`` is of the wrong sort (a
  460. list when you were expecting an object, for example) or a ``TypeError`` if
  461. your field does not support that type of lookup. For many fields, you can get
  462. by with handling the lookup types that need special handling for your field
  463. and pass the rest to the :meth:`~Field.get_db_prep_lookup` method of the parent
  464. class.
  465. If you needed to implement :meth:`.get_db_prep_save`, you will usually need to
  466. implement :meth:`.get_prep_lookup`. If you don't, :meth:`.get_prep_value` will
  467. be called by the default implementation, to manage ``exact``, ``gt``, ``gte``,
  468. ``lt``, ``lte``, ``in`` and ``range`` lookups.
  469. You may also want to implement this method to limit the lookup types that could
  470. be used with your custom field type.
  471. Note that, for ``"range"`` and ``"in"`` lookups, ``get_prep_lookup`` will receive
  472. a list of objects (presumably of the right type) and will need to convert them
  473. to a list of things of the right type for passing to the database. Most of the
  474. time, you can reuse ``get_prep_value()``, or at least factor out some common
  475. pieces.
  476. For example, the following code implements ``get_prep_lookup`` to limit the
  477. accepted lookup types to ``exact`` and ``in``::
  478. class HandField(models.Field):
  479. # ...
  480. def get_prep_lookup(self, lookup_type, value):
  481. # We only handle 'exact' and 'in'. All others are errors.
  482. if lookup_type == 'exact':
  483. return self.get_prep_value(value)
  484. elif lookup_type == 'in':
  485. return [self.get_prep_value(v) for v in value]
  486. else:
  487. raise TypeError('Lookup type %r not supported.' % lookup_type)
  488. For performing database-specific data conversions required by a lookup,
  489. you can override :meth:`~Field.get_db_prep_lookup`.
  490. .. _specifying-form-field-for-model-field:
  491. Specifying the form field for a model field
  492. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  493. To customize the form field used by :class:`~django.forms.ModelForm`, you can
  494. override :meth:`~Field.formfield`.
  495. The form field class can be specified via the ``form_class`` and
  496. ``choices_form_class`` arguments; the latter is used if the field has choices
  497. specified, the former otherwise. If these arguments are not provided,
  498. :class:`~django.forms.CharField` or :class:`~django.forms.TypedChoiceField`
  499. will be used.
  500. All of the ``kwargs`` dictionary is passed directly to the form field's
  501. ``__init__()`` method. Normally, all you need to do is set up a good default
  502. for the ``form_class`` (and maybe ``choices_form_class``) argument and then
  503. delegate further handling to the parent class. This might require you to write
  504. a custom form field (and even a form widget). See the :doc:`forms documentation
  505. </topics/forms/index>` for information about this.
  506. Continuing our ongoing example, we can write the :meth:`~Field.formfield` method
  507. as::
  508. class HandField(models.Field):
  509. # ...
  510. def formfield(self, **kwargs):
  511. # This is a fairly standard way to set up some defaults
  512. # while letting the caller override them.
  513. defaults = {'form_class': MyFormField}
  514. defaults.update(kwargs)
  515. return super(HandField, self).formfield(**defaults)
  516. This assumes we've imported a ``MyFormField`` field class (which has its own
  517. default widget). This document doesn't cover the details of writing custom form
  518. fields.
  519. .. _helper functions: ../forms/#generating-forms-for-models
  520. .. _forms documentation: ../forms/
  521. .. _emulating-built-in-field-types:
  522. Emulating built-in field types
  523. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  524. If you have created a :meth:`.db_type` method, you don't need to worry about
  525. :meth:`.get_internal_type` -- it won't be used much. Sometimes, though, your
  526. database storage is similar in type to some other field, so you can use that
  527. other field's logic to create the right column.
  528. For example::
  529. class HandField(models.Field):
  530. # ...
  531. def get_internal_type(self):
  532. return 'CharField'
  533. No matter which database backend we are using, this will mean that
  534. :djadmin:`migrate` and other SQL commands create the right column type for
  535. storing a string.
  536. If :meth:`.get_internal_type` returns a string that is not known to Django for
  537. the database backend you are using -- that is, it doesn't appear in
  538. ``django.db.backends.<db_name>.creation.data_types`` -- the string will still be
  539. used by the serializer, but the default :meth:`~Field.db_type` method will
  540. return ``None``. See the documentation of :meth:`~Field.db_type` for reasons why
  541. this might be useful. Putting a descriptive string in as the type of the field
  542. for the serializer is a useful idea if you're ever going to be using the
  543. serializer output in some other place, outside of Django.
  544. .. _converting-model-field-to-serialization:
  545. Converting field data for serialization
  546. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  547. To customize how the values are serialized by a serializer, you can override
  548. :meth:`~Field.value_to_string`. Calling ``Field._get_val_from_obj(obj)`` is the
  549. best way to get the value serialized. For example, since our ``HandField`` uses
  550. strings for its data storage anyway, we can reuse some existing conversion code::
  551. class HandField(models.Field):
  552. # ...
  553. def value_to_string(self, obj):
  554. value = self._get_val_from_obj(obj)
  555. return self.get_prep_value(value)
  556. Some general advice
  557. --------------------
  558. Writing a custom field can be a tricky process, particularly if you're doing
  559. complex conversions between your Python types and your database and
  560. serialization formats. Here are a couple of tips to make things go more
  561. smoothly:
  562. 1. Look at the existing Django fields (in
  563. :file:`django/db/models/fields/__init__.py`) for inspiration. Try to find
  564. a field that's similar to what you want and extend it a little bit,
  565. instead of creating an entirely new field from scratch.
  566. 2. Put a ``__str__()`` (``__unicode__()`` on Python 2) method on the class you're
  567. wrapping up as a field. There are a lot of places where the default
  568. behavior of the field code is to call
  569. :func:`~django.utils.encoding.force_text` on the value. (In our
  570. examples in this document, ``value`` would be a ``Hand`` instance, not a
  571. ``HandField``). So if your ``__str__()`` method (``__unicode__()`` on
  572. Python 2) automatically converts to the string form of your Python object,
  573. you can save yourself a lot of work.
  574. Writing a ``FileField`` subclass
  575. ================================
  576. In addition to the above methods, fields that deal with files have a few other
  577. special requirements which must be taken into account. The majority of the
  578. mechanics provided by ``FileField``, such as controlling database storage and
  579. retrieval, can remain unchanged, leaving subclasses to deal with the challenge
  580. of supporting a particular type of file.
  581. Django provides a ``File`` class, which is used as a proxy to the file's
  582. contents and operations. This can be subclassed to customize how the file is
  583. accessed, and what methods are available. It lives at
  584. ``django.db.models.fields.files``, and its default behavior is explained in the
  585. :doc:`file documentation </ref/files/file>`.
  586. Once a subclass of ``File`` is created, the new ``FileField`` subclass must be
  587. told to use it. To do so, simply assign the new ``File`` subclass to the special
  588. ``attr_class`` attribute of the ``FileField`` subclass.
  589. A few suggestions
  590. ------------------
  591. In addition to the above details, there are a few guidelines which can greatly
  592. improve the efficiency and readability of the field's code.
  593. 1. The source for Django's own ``ImageField`` (in
  594. ``django/db/models/fields/files.py``) is a great example of how to
  595. subclass ``FileField`` to support a particular type of file, as it
  596. incorporates all of the techniques described above.
  597. 2. Cache file attributes wherever possible. Since files may be stored in
  598. remote storage systems, retrieving them may cost extra time, or even
  599. money, that isn't always necessary. Once a file is retrieved to obtain
  600. some data about its content, cache as much of that data as possible to
  601. reduce the number of times the file must be retrieved on subsequent
  602. calls for that information.