expressions.txt 52 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392
  1. =================
  2. Query Expressions
  3. =================
  4. .. currentmodule:: django.db.models
  5. Query expressions describe a value or a computation that can be used as part of
  6. an update, create, filter, order by, annotation, or aggregate. When an
  7. expression outputs a boolean value, it may be used directly in filters. There
  8. are a number of built-in expressions (documented below) that can be used to
  9. help you write queries. Expressions can be combined, or in some cases nested,
  10. to form more complex computations.
  11. Supported arithmetic
  12. ====================
  13. Django supports negation, addition, subtraction, multiplication, division,
  14. modulo arithmetic, and the power operator on query expressions, using Python
  15. constants, variables, and even other expressions.
  16. .. _output-field:
  17. Output field
  18. ============
  19. Many of the expressions documented in this section support an optional
  20. ``output_field`` parameter. If given, Django will load the value into that
  21. field after retrieving it from the database.
  22. ``output_field`` takes a model field instance, like ``IntegerField()`` or
  23. ``BooleanField()``. Usually, the field doesn't need any arguments, like
  24. ``max_length``, since field arguments relate to data validation which will not
  25. be performed on the expression's output value.
  26. ``output_field`` is only required when Django is unable to automatically
  27. determine the result's field type, such as complex expressions that mix field
  28. types. For example, adding a ``DecimalField()`` and a ``FloatField()`` requires
  29. an output field, like ``output_field=FloatField()``.
  30. Some examples
  31. =============
  32. .. code-block:: pycon
  33. >>> from django.db.models import Count, F, Value
  34. >>> from django.db.models.functions import Length, Upper
  35. >>> from django.db.models.lookups import GreaterThan
  36. # Find companies that have more employees than chairs.
  37. >>> Company.objects.filter(num_employees__gt=F("num_chairs"))
  38. # Find companies that have at least twice as many employees
  39. # as chairs. Both the querysets below are equivalent.
  40. >>> Company.objects.filter(num_employees__gt=F("num_chairs") * 2)
  41. >>> Company.objects.filter(num_employees__gt=F("num_chairs") + F("num_chairs"))
  42. # How many chairs are needed for each company to seat all employees?
  43. >>> company = (
  44. ... Company.objects.filter(num_employees__gt=F("num_chairs"))
  45. ... .annotate(chairs_needed=F("num_employees") - F("num_chairs"))
  46. ... .first()
  47. ... )
  48. >>> company.num_employees
  49. 120
  50. >>> company.num_chairs
  51. 50
  52. >>> company.chairs_needed
  53. 70
  54. # Create a new company using expressions.
  55. >>> company = Company.objects.create(name="Google", ticker=Upper(Value("goog")))
  56. # Be sure to refresh it if you need to access the field.
  57. >>> company.refresh_from_db()
  58. >>> company.ticker
  59. 'GOOG'
  60. # Annotate models with an aggregated value. Both forms
  61. # below are equivalent.
  62. >>> Company.objects.annotate(num_products=Count("products"))
  63. >>> Company.objects.annotate(num_products=Count(F("products")))
  64. # Aggregates can contain complex computations also
  65. >>> Company.objects.annotate(num_offerings=Count(F("products") + F("services")))
  66. # Expressions can also be used in order_by(), either directly
  67. >>> Company.objects.order_by(Length("name").asc())
  68. >>> Company.objects.order_by(Length("name").desc())
  69. # or using the double underscore lookup syntax.
  70. >>> from django.db.models import CharField
  71. >>> from django.db.models.functions import Length
  72. >>> CharField.register_lookup(Length)
  73. >>> Company.objects.order_by("name__length")
  74. # Boolean expression can be used directly in filters.
  75. >>> from django.db.models import Exists, OuterRef
  76. >>> Company.objects.filter(
  77. ... Exists(Employee.objects.filter(company=OuterRef("pk"), salary__gt=10))
  78. ... )
  79. # Lookup expressions can also be used directly in filters
  80. >>> Company.objects.filter(GreaterThan(F("num_employees"), F("num_chairs")))
  81. # or annotations.
  82. >>> Company.objects.annotate(
  83. ... need_chairs=GreaterThan(F("num_employees"), F("num_chairs")),
  84. ... )
  85. Built-in Expressions
  86. ====================
  87. .. note::
  88. These expressions are defined in ``django.db.models.expressions`` and
  89. ``django.db.models.aggregates``, but for convenience they're available and
  90. usually imported from :mod:`django.db.models`.
  91. ``F()`` expressions
  92. -------------------
  93. .. class:: F
  94. An ``F()`` object represents the value of a model field, transformed value of a
  95. model field, or annotated column. It makes it possible to refer to model field
  96. values and perform database operations using them without actually having to
  97. pull them out of the database into Python memory.
  98. Instead, Django uses the ``F()`` object to generate an SQL expression that
  99. describes the required operation at the database level.
  100. Let's try this with an example. Normally, one might do something like this::
  101. # Tintin filed a news story!
  102. reporter = Reporters.objects.get(name="Tintin")
  103. reporter.stories_filed += 1
  104. reporter.save()
  105. Here, we have pulled the value of ``reporter.stories_filed`` from the database
  106. into memory and manipulated it using familiar Python operators, and then saved
  107. the object back to the database. But instead we could also have done::
  108. from django.db.models import F
  109. reporter = Reporters.objects.get(name="Tintin")
  110. reporter.stories_filed = F("stories_filed") + 1
  111. reporter.save()
  112. Although ``reporter.stories_filed = F('stories_filed') + 1`` looks like a
  113. normal Python assignment of value to an instance attribute, in fact it's an SQL
  114. construct describing an operation on the database.
  115. When Django encounters an instance of ``F()``, it overrides the standard Python
  116. operators to create an encapsulated SQL expression; in this case, one which
  117. instructs the database to increment the database field represented by
  118. ``reporter.stories_filed``.
  119. Whatever value is or was on ``reporter.stories_filed``, Python never gets to
  120. know about it - it is dealt with entirely by the database. All Python does,
  121. through Django's ``F()`` class, is create the SQL syntax to refer to the field
  122. and describe the operation.
  123. To access the new value saved this way, the object must be reloaded::
  124. reporter = Reporters.objects.get(pk=reporter.pk)
  125. # Or, more succinctly:
  126. reporter.refresh_from_db()
  127. As well as being used in operations on single instances as above, ``F()`` can
  128. be used with ``update()`` to perform bulk updates on a ``QuerySet``. This
  129. reduces the two queries we were using above - the ``get()`` and the
  130. :meth:`~Model.save()` - to just one::
  131. reporter = Reporters.objects.filter(name="Tintin")
  132. reporter.update(stories_filed=F("stories_filed") + 1)
  133. We can also use :meth:`~django.db.models.query.QuerySet.update()` to increment
  134. the field value on multiple objects - which could be very much faster than
  135. pulling them all into Python from the database, looping over them, incrementing
  136. the field value of each one, and saving each one back to the database::
  137. Reporter.objects.update(stories_filed=F("stories_filed") + 1)
  138. ``F()`` therefore can offer performance advantages by:
  139. * getting the database, rather than Python, to do work
  140. * reducing the number of queries some operations require
  141. .. _slicing-using-f:
  142. Slicing ``F()`` expressions
  143. ~~~~~~~~~~~~~~~~~~~~~~~~~~~
  144. For string-based fields, text-based fields, and
  145. :class:`~django.contrib.postgres.fields.ArrayField`, you can use Python's
  146. array-slicing syntax. The indices are 0-based and the ``step`` argument to
  147. ``slice`` is not supported. For example:
  148. .. code-block:: pycon
  149. >>> # Replacing a name with a substring of itself.
  150. >>> writer = Writers.objects.get(name="Priyansh")
  151. >>> writer.name = F("name")[1:5]
  152. >>> writer.save()
  153. >>> writer.refresh_from_db()
  154. >>> writer.name
  155. 'riya'
  156. .. _avoiding-race-conditions-using-f:
  157. Avoiding race conditions using ``F()``
  158. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  159. Another useful benefit of ``F()`` is that having the database - rather than
  160. Python - update a field's value avoids a *race condition*.
  161. If two Python threads execute the code in the first example above, one thread
  162. could retrieve, increment, and save a field's value after the other has
  163. retrieved it from the database. The value that the second thread saves will be
  164. based on the original value; the work of the first thread will be lost.
  165. If the database is responsible for updating the field, the process is more
  166. robust: it will only ever update the field based on the value of the field in
  167. the database when the :meth:`~Model.save()` or ``update()`` is executed, rather
  168. than based on its value when the instance was retrieved.
  169. ``F()`` assignments persist after ``Model.save()``
  170. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  171. ``F()`` objects assigned to model fields persist after saving the model
  172. instance and will be applied on each :meth:`~Model.save()`. For example::
  173. reporter = Reporters.objects.get(name="Tintin")
  174. reporter.stories_filed = F("stories_filed") + 1
  175. reporter.save()
  176. reporter.name = "Tintin Jr."
  177. reporter.save()
  178. ``stories_filed`` will be updated twice in this case. If it's initially ``1``,
  179. the final value will be ``3``. This persistence can be avoided by reloading the
  180. model object after saving it, for example, by using
  181. :meth:`~Model.refresh_from_db()`.
  182. Using ``F()`` in filters
  183. ~~~~~~~~~~~~~~~~~~~~~~~~
  184. ``F()`` is also very useful in ``QuerySet`` filters, where they make it
  185. possible to filter a set of objects against criteria based on their field
  186. values, rather than on Python values.
  187. This is documented in :ref:`using F() expressions in queries
  188. <using-f-expressions-in-filters>`.
  189. .. _using-f-with-annotations:
  190. Using ``F()`` with annotations
  191. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  192. ``F()`` can be used to create dynamic fields on your models by combining
  193. different fields with arithmetic::
  194. company = Company.objects.annotate(chairs_needed=F("num_employees") - F("num_chairs"))
  195. If the fields that you're combining are of different types you'll need to tell
  196. Django what kind of field will be returned. Most expressions support
  197. :ref:`output_field<output-field>` for this case, but since ``F()`` does not, you
  198. will need to wrap the expression with :class:`ExpressionWrapper`::
  199. from django.db.models import DateTimeField, ExpressionWrapper, F
  200. Ticket.objects.annotate(
  201. expires=ExpressionWrapper(
  202. F("active_at") + F("duration"), output_field=DateTimeField()
  203. )
  204. )
  205. When referencing relational fields such as ``ForeignKey``, ``F()`` returns the
  206. primary key value rather than a model instance:
  207. .. code-block:: pycon
  208. >>> car = Company.objects.annotate(built_by=F("manufacturer"))[0]
  209. >>> car.manufacturer
  210. <Manufacturer: Toyota>
  211. >>> car.built_by
  212. 3
  213. .. _using-f-to-sort-null-values:
  214. Using ``F()`` to sort null values
  215. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  216. Use ``F()`` and the ``nulls_first`` or ``nulls_last`` keyword argument to
  217. :meth:`.Expression.asc` or :meth:`~.Expression.desc` to control the ordering of
  218. a field's null values. By default, the ordering depends on your database.
  219. For example, to sort companies that haven't been contacted (``last_contacted``
  220. is null) after companies that have been contacted::
  221. from django.db.models import F
  222. Company.objects.order_by(F("last_contacted").desc(nulls_last=True))
  223. Using ``F()`` with logical operations
  224. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  225. ``F()`` expressions that output ``BooleanField`` can be logically negated with
  226. the inversion operator ``~F()``. For example, to swap the activation status of
  227. companies::
  228. from django.db.models import F
  229. Company.objects.update(is_active=~F("is_active"))
  230. .. _func-expressions:
  231. ``Func()`` expressions
  232. ----------------------
  233. ``Func()`` expressions are the base type of all expressions that involve
  234. database functions like ``COALESCE`` and ``LOWER``, or aggregates like ``SUM``.
  235. They can be used directly::
  236. from django.db.models import F, Func
  237. queryset.annotate(field_lower=Func(F("field"), function="LOWER"))
  238. or they can be used to build a library of database functions::
  239. class Lower(Func):
  240. function = "LOWER"
  241. queryset.annotate(field_lower=Lower("field"))
  242. But both cases will result in a queryset where each model is annotated with an
  243. extra attribute ``field_lower`` produced, roughly, from the following SQL:
  244. .. code-block:: sql
  245. SELECT
  246. ...
  247. LOWER("db_table"."field") as "field_lower"
  248. See :doc:`database-functions` for a list of built-in database functions.
  249. The ``Func`` API is as follows:
  250. .. class:: Func(*expressions, **extra)
  251. .. attribute:: function
  252. A class attribute describing the function that will be generated.
  253. Specifically, the ``function`` will be interpolated as the ``function``
  254. placeholder within :attr:`template`. Defaults to ``None``.
  255. .. attribute:: template
  256. A class attribute, as a format string, that describes the SQL that is
  257. generated for this function. Defaults to
  258. ``'%(function)s(%(expressions)s)'``.
  259. If you're constructing SQL like ``strftime('%W', 'date')`` and need a
  260. literal ``%`` character in the query, quadruple it (``%%%%``) in the
  261. ``template`` attribute because the string is interpolated twice: once
  262. during the template interpolation in ``as_sql()`` and once in the SQL
  263. interpolation with the query parameters in the database cursor.
  264. .. attribute:: arg_joiner
  265. A class attribute that denotes the character used to join the list of
  266. ``expressions`` together. Defaults to ``', '``.
  267. .. attribute:: arity
  268. A class attribute that denotes the number of arguments the function
  269. accepts. If this attribute is set and the function is called with a
  270. different number of expressions, ``TypeError`` will be raised. Defaults
  271. to ``None``.
  272. .. method:: as_sql(compiler, connection, function=None, template=None, arg_joiner=None, **extra_context)
  273. Generates the SQL fragment for the database function. Returns a tuple
  274. ``(sql, params)``, where ``sql`` is the SQL string, and ``params`` is
  275. the list or tuple of query parameters.
  276. The ``as_vendor()`` methods should use the ``function``, ``template``,
  277. ``arg_joiner``, and any other ``**extra_context`` parameters to
  278. customize the SQL as needed. For example:
  279. .. code-block:: python
  280. :caption: ``django/db/models/functions.py``
  281. class ConcatPair(Func):
  282. ...
  283. function = "CONCAT"
  284. ...
  285. def as_mysql(self, compiler, connection, **extra_context):
  286. return super().as_sql(
  287. compiler,
  288. connection,
  289. function="CONCAT_WS",
  290. template="%(function)s('', %(expressions)s)",
  291. **extra_context
  292. )
  293. To avoid an SQL injection vulnerability, ``extra_context`` :ref:`must
  294. not contain untrusted user input <avoiding-sql-injection-in-query-expressions>`
  295. as these values are interpolated into the SQL string rather than passed
  296. as query parameters, where the database driver would escape them.
  297. The ``*expressions`` argument is a list of positional expressions that the
  298. function will be applied to. The expressions will be converted to strings,
  299. joined together with ``arg_joiner``, and then interpolated into the ``template``
  300. as the ``expressions`` placeholder.
  301. Positional arguments can be expressions or Python values. Strings are
  302. assumed to be column references and will be wrapped in ``F()`` expressions
  303. while other values will be wrapped in ``Value()`` expressions.
  304. The ``**extra`` kwargs are ``key=value`` pairs that can be interpolated
  305. into the ``template`` attribute. To avoid an SQL injection vulnerability,
  306. ``extra`` :ref:`must not contain untrusted user input
  307. <avoiding-sql-injection-in-query-expressions>` as these values are interpolated
  308. into the SQL string rather than passed as query parameters, where the database
  309. driver would escape them.
  310. The ``function``, ``template``, and ``arg_joiner`` keywords can be used to
  311. replace the attributes of the same name without having to define your own
  312. class. :ref:`output_field<output-field>` can be used to define the expected
  313. return type.
  314. ``Aggregate()`` expressions
  315. ---------------------------
  316. An aggregate expression is a special case of a :ref:`Func() expression
  317. <func-expressions>` that informs the query that a ``GROUP BY`` clause
  318. is required. All of the :ref:`aggregate functions <aggregation-functions>`,
  319. like ``Sum()`` and ``Count()``, inherit from ``Aggregate()``.
  320. Since ``Aggregate``\s are expressions and wrap expressions, you can represent
  321. some complex computations::
  322. from django.db.models import Count
  323. Company.objects.annotate(
  324. managers_required=(Count("num_employees") / 4) + Count("num_managers")
  325. )
  326. The ``Aggregate`` API is as follows:
  327. .. class:: Aggregate(*expressions, output_field=None, distinct=False, filter=None, default=None, order_by=None, **extra)
  328. .. attribute:: template
  329. A class attribute, as a format string, that describes the SQL that is
  330. generated for this aggregate. Defaults to
  331. ``'%(function)s(%(distinct)s%(expressions)s)'``.
  332. .. attribute:: function
  333. A class attribute describing the aggregate function that will be
  334. generated. Specifically, the ``function`` will be interpolated as the
  335. ``function`` placeholder within :attr:`template`. Defaults to ``None``.
  336. .. attribute:: window_compatible
  337. Defaults to ``True`` since most aggregate functions can be used as the
  338. source expression in :class:`~django.db.models.expressions.Window`.
  339. .. attribute:: allow_distinct
  340. A class attribute determining whether or not this aggregate function
  341. allows passing a ``distinct`` keyword argument. If set to ``False``
  342. (default), ``TypeError`` is raised if ``distinct=True`` is passed.
  343. .. attribute:: allow_order_by
  344. .. versionadded:: 6.0
  345. A class attribute determining whether or not this aggregate function
  346. allows passing a ``order_by`` keyword argument. If set to ``False``
  347. (default), ``TypeError`` is raised if ``order_by`` is passed as a value
  348. other than ``None``.
  349. .. attribute:: empty_result_set_value
  350. Defaults to ``None`` since most aggregate functions result in ``NULL``
  351. when applied to an empty result set.
  352. The ``expressions`` positional arguments can include expressions, transforms of
  353. the model field, or the names of model fields. They will be converted to a
  354. string and used as the ``expressions`` placeholder within the ``template``.
  355. The ``distinct`` argument determines whether or not the aggregate function
  356. should be invoked for each distinct value of ``expressions`` (or set of
  357. values, for multiple ``expressions``). The argument is only supported on
  358. aggregates that have :attr:`~Aggregate.allow_distinct` set to ``True``.
  359. The ``filter`` argument takes a :class:`Q object <django.db.models.Q>` that's
  360. used to filter the rows that are aggregated. See :ref:`conditional-aggregation`
  361. and :ref:`filtering-on-annotations` for example usage.
  362. The ``order_by`` argument behaves similarly to the ``field_names`` input of the
  363. :meth:`~.QuerySet.order_by` function, accepting a field name (with an optional
  364. ``"-"`` prefix which indicates descending order) or an expression (or a tuple
  365. or list of strings and/or expressions) that specifies the ordering of the
  366. elements in the result.
  367. The ``default`` argument takes a value that will be passed along with the
  368. aggregate to :class:`~django.db.models.functions.Coalesce`. This is useful for
  369. specifying a value to be returned other than ``None`` when the queryset (or
  370. grouping) contains no entries.
  371. The ``**extra`` kwargs are ``key=value`` pairs that can be interpolated
  372. into the ``template`` attribute.
  373. .. versionchanged:: 6.0
  374. The ``order_by`` argument was added.
  375. Creating your own Aggregate Functions
  376. -------------------------------------
  377. You can create your own aggregate functions, too. At a minimum, you need to
  378. define ``function``, but you can also completely customize the SQL that is
  379. generated. Here's a brief example::
  380. from django.db.models import Aggregate
  381. class Sum(Aggregate):
  382. # Supports SUM(ALL field).
  383. function = "SUM"
  384. template = "%(function)s(%(all_values)s%(expressions)s)"
  385. allow_distinct = False
  386. arity = 1
  387. def __init__(self, expression, all_values=False, **extra):
  388. super().__init__(expression, all_values="ALL " if all_values else "", **extra)
  389. ``Value()`` expressions
  390. -----------------------
  391. .. class:: Value(value, output_field=None)
  392. A ``Value()`` object represents the smallest possible component of an
  393. expression: a simple value. When you need to represent the value of an integer,
  394. boolean, or string within an expression, you can wrap that value within a
  395. ``Value()``.
  396. You will rarely need to use ``Value()`` directly. When you write the expression
  397. ``F('field') + 1``, Django implicitly wraps the ``1`` in a ``Value()``,
  398. allowing simple values to be used in more complex expressions. You will need to
  399. use ``Value()`` when you want to pass a string to an expression. Most
  400. expressions interpret a string argument as the name of a field, like
  401. ``Lower('name')``.
  402. The ``value`` argument describes the value to be included in the expression,
  403. such as ``1``, ``True``, or ``None``. Django knows how to convert these Python
  404. values into their corresponding database type.
  405. If no :ref:`output_field<output-field>` is specified, it will be inferred from
  406. the type of the provided ``value`` for many common types. For example, passing
  407. an instance of :py:class:`datetime.datetime` as ``value`` defaults
  408. ``output_field`` to :class:`~django.db.models.DateTimeField`.
  409. ``ExpressionWrapper()`` expressions
  410. -----------------------------------
  411. .. class:: ExpressionWrapper(expression, output_field)
  412. ``ExpressionWrapper`` surrounds another expression and provides access to
  413. properties, such as :ref:`output_field<output-field>`, that may not be
  414. available on other expressions. ``ExpressionWrapper`` is necessary when using
  415. arithmetic on ``F()`` expressions with different types as described in
  416. :ref:`using-f-with-annotations`.
  417. Conditional expressions
  418. -----------------------
  419. Conditional expressions allow you to use :keyword:`if` ... :keyword:`elif` ...
  420. :keyword:`else` logic in queries. Django natively supports SQL ``CASE``
  421. expressions. For more details see :doc:`conditional-expressions`.
  422. ``Subquery()`` expressions
  423. --------------------------
  424. .. class:: Subquery(queryset, output_field=None)
  425. You can add an explicit subquery to a ``QuerySet`` using the ``Subquery``
  426. expression.
  427. For example, to annotate each post with the email address of the author of the
  428. newest comment on that post:
  429. .. code-block:: pycon
  430. >>> from django.db.models import OuterRef, Subquery
  431. >>> newest = Comment.objects.filter(post=OuterRef("pk")).order_by("-created_at")
  432. >>> Post.objects.annotate(newest_commenter_email=Subquery(newest.values("email")[:1]))
  433. On PostgreSQL, the SQL looks like:
  434. .. code-block:: sql
  435. SELECT "post"."id", (
  436. SELECT U0."email"
  437. FROM "comment" U0
  438. WHERE U0."post_id" = ("post"."id")
  439. ORDER BY U0."created_at" DESC LIMIT 1
  440. ) AS "newest_commenter_email" FROM "post"
  441. .. note::
  442. The examples in this section are designed to show how to force
  443. Django to execute a subquery. In some cases it may be possible to
  444. write an equivalent queryset that performs the same task more
  445. clearly or efficiently.
  446. Referencing columns from the outer queryset
  447. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  448. .. class:: OuterRef(field)
  449. Use ``OuterRef`` when a queryset in a ``Subquery`` needs to refer to a field
  450. from the outer query or its transform. It acts like an :class:`F` expression
  451. except that the check to see if it refers to a valid field isn't made until the
  452. outer queryset is resolved.
  453. Instances of ``OuterRef`` may be used in conjunction with nested instances
  454. of ``Subquery`` to refer to a containing queryset that isn't the immediate
  455. parent. For example, this queryset would need to be within a nested pair of
  456. ``Subquery`` instances to resolve correctly:
  457. .. code-block:: pycon
  458. >>> Book.objects.filter(author=OuterRef(OuterRef("pk")))
  459. Limiting a subquery to a single column
  460. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  461. There are times when a single column must be returned from a ``Subquery``, for
  462. instance, to use a ``Subquery`` as the target of an ``__in`` lookup. To return
  463. all comments for posts published within the last day:
  464. .. code-block:: pycon
  465. >>> from datetime import timedelta
  466. >>> from django.utils import timezone
  467. >>> one_day_ago = timezone.now() - timedelta(days=1)
  468. >>> posts = Post.objects.filter(published_at__gte=one_day_ago)
  469. >>> Comment.objects.filter(post__in=Subquery(posts.values("pk")))
  470. In this case, the subquery must use :meth:`~.QuerySet.values`
  471. to return only a single column: the primary key of the post.
  472. Limiting the subquery to a single row
  473. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  474. To prevent a subquery from returning multiple rows, a slice (``[:1]``) of the
  475. queryset is used:
  476. .. code-block:: pycon
  477. >>> subquery = Subquery(newest.values("email")[:1])
  478. >>> Post.objects.annotate(newest_commenter_email=subquery)
  479. In this case, the subquery must only return a single column *and* a single
  480. row: the email address of the most recently created comment.
  481. (Using :meth:`~.QuerySet.get` instead of a slice would fail because the
  482. ``OuterRef`` cannot be resolved until the queryset is used within a
  483. ``Subquery``.)
  484. ``Exists()`` subqueries
  485. ~~~~~~~~~~~~~~~~~~~~~~~
  486. .. class:: Exists(queryset)
  487. ``Exists`` is a ``Subquery`` subclass that uses an SQL ``EXISTS`` statement. In
  488. many cases it will perform better than a subquery since the database is able to
  489. stop evaluation of the subquery when a first matching row is found.
  490. For example, to annotate each post with whether or not it has a comment from
  491. within the last day:
  492. .. code-block:: pycon
  493. >>> from django.db.models import Exists, OuterRef
  494. >>> from datetime import timedelta
  495. >>> from django.utils import timezone
  496. >>> one_day_ago = timezone.now() - timedelta(days=1)
  497. >>> recent_comments = Comment.objects.filter(
  498. ... post=OuterRef("pk"),
  499. ... created_at__gte=one_day_ago,
  500. ... )
  501. >>> Post.objects.annotate(recent_comment=Exists(recent_comments))
  502. On PostgreSQL, the SQL looks like:
  503. .. code-block:: sql
  504. SELECT "post"."id", "post"."published_at", EXISTS(
  505. SELECT (1) as "a"
  506. FROM "comment" U0
  507. WHERE (
  508. U0."created_at" >= YYYY-MM-DD HH:MM:SS AND
  509. U0."post_id" = "post"."id"
  510. )
  511. LIMIT 1
  512. ) AS "recent_comment" FROM "post"
  513. It's unnecessary to force ``Exists`` to refer to a single column, since the
  514. columns are discarded and a boolean result is returned. Similarly, since
  515. ordering is unimportant within an SQL ``EXISTS`` subquery and would only
  516. degrade performance, it's automatically removed.
  517. You can query using ``NOT EXISTS`` with ``~Exists()``.
  518. Filtering on a ``Subquery()`` or ``Exists()`` expressions
  519. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  520. ``Subquery()`` that returns a boolean value and ``Exists()`` may be used as a
  521. ``condition`` in :class:`~django.db.models.expressions.When` expressions, or to
  522. directly filter a queryset:
  523. .. code-block:: pycon
  524. >>> recent_comments = Comment.objects.filter(...) # From above
  525. >>> Post.objects.filter(Exists(recent_comments))
  526. This will ensure that the subquery will not be added to the ``SELECT`` columns,
  527. which may result in a better performance.
  528. Using aggregates within a ``Subquery`` expression
  529. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  530. Aggregates may be used within a ``Subquery``, but they require a specific
  531. combination of :meth:`~.QuerySet.filter`, :meth:`~.QuerySet.values`, and
  532. :meth:`~.QuerySet.annotate` to get the subquery grouping correct.
  533. Assuming both models have a ``length`` field, to find posts where the post
  534. length is greater than the total length of all combined comments:
  535. .. code-block:: pycon
  536. >>> from django.db.models import OuterRef, Subquery, Sum
  537. >>> comments = Comment.objects.filter(post=OuterRef("pk")).order_by().values("post")
  538. >>> total_comments = comments.annotate(total=Sum("length")).values("total")
  539. >>> Post.objects.filter(length__gt=Subquery(total_comments))
  540. The initial ``filter(...)`` limits the subquery to the relevant parameters.
  541. ``order_by()`` removes the default :attr:`~django.db.models.Options.ordering`
  542. (if any) on the ``Comment`` model. ``values('post')`` aggregates comments by
  543. ``Post``. Finally, ``annotate(...)`` performs the aggregation. The order in
  544. which these queryset methods are applied is important. In this case, since the
  545. subquery must be limited to a single column, ``values('total')`` is required.
  546. This is the only way to perform an aggregation within a ``Subquery``, as
  547. using :meth:`~.QuerySet.aggregate` attempts to evaluate the queryset (and if
  548. there is an ``OuterRef``, this will not be possible to resolve).
  549. Raw SQL expressions
  550. -------------------
  551. .. currentmodule:: django.db.models.expressions
  552. .. class:: RawSQL(sql, params, output_field=None)
  553. Sometimes database expressions can't easily express a complex ``WHERE`` clause.
  554. In these edge cases, use the ``RawSQL`` expression. For example:
  555. .. code-block:: pycon
  556. >>> from django.db.models.expressions import RawSQL
  557. >>> queryset.annotate(val=RawSQL("select col from sometable where othercol = %s", (param,)))
  558. These extra lookups may not be portable to different database engines (because
  559. you're explicitly writing SQL code) and violate the DRY principle, so you
  560. should avoid them if possible.
  561. ``RawSQL`` expressions can also be used as the target of ``__in`` filters:
  562. .. code-block:: pycon
  563. >>> queryset.filter(id__in=RawSQL("select id from sometable where col = %s", (param,)))
  564. .. warning::
  565. To protect against `SQL injection attacks
  566. <https://en.wikipedia.org/wiki/SQL_injection>`_, you must escape any
  567. parameters that the user can control by using ``params``. ``params`` is a
  568. required argument to force you to acknowledge that you're not interpolating
  569. your SQL with user-provided data.
  570. You also must not quote placeholders in the SQL string. This example is
  571. vulnerable to SQL injection because of the quotes around ``%s``::
  572. RawSQL("select col from sometable where othercol = '%s'") # unsafe!
  573. You can read more about how Django's :ref:`SQL injection protection
  574. <sql-injection-protection>` works.
  575. Window functions
  576. ----------------
  577. Window functions provide a way to apply functions on partitions. Unlike a
  578. normal aggregation function which computes a final result for each set defined
  579. by the group by, window functions operate on :ref:`frames <window-frames>` and
  580. partitions, and compute the result for each row.
  581. You can specify multiple windows in the same query which in Django ORM would be
  582. equivalent to including multiple expressions in a :doc:`QuerySet.annotate()
  583. </topics/db/aggregation>` call. The ORM doesn't make use of named windows,
  584. instead they are part of the selected columns.
  585. .. class:: Window(expression, partition_by=None, order_by=None, frame=None, output_field=None)
  586. .. attribute:: template
  587. Defaults to ``%(expression)s OVER (%(window)s)``. If only the
  588. ``expression`` argument is provided, the window clause will be blank.
  589. The ``Window`` class is the main expression for an ``OVER`` clause.
  590. The ``expression`` argument is either a :ref:`window function
  591. <window-functions>`, an :ref:`aggregate function <aggregation-functions>`, or
  592. an expression that's compatible in a window clause.
  593. The ``partition_by`` argument accepts an expression or a sequence of
  594. expressions (column names should be wrapped in an ``F``-object) that control
  595. the partitioning of the rows. Partitioning narrows which rows are used to
  596. compute the result set.
  597. The :ref:`output_field<output-field>` is specified either as an argument or by
  598. the expression.
  599. The ``order_by`` argument accepts an expression on which you can call
  600. :meth:`~django.db.models.Expression.asc` and
  601. :meth:`~django.db.models.Expression.desc`, a string of a field name (with an
  602. optional ``"-"`` prefix which indicates descending order), or a tuple or list
  603. of strings and/or expressions. The ordering controls the order in which the
  604. expression is applied. For example, if you sum over the rows in a partition,
  605. the first result is the value of the first row, the second is the sum of first
  606. and second row.
  607. The ``frame`` parameter specifies which other rows that should be used in the
  608. computation. See :ref:`window-frames` for details.
  609. For example, to annotate each movie with the average rating for the movies by
  610. the same studio in the same genre and release year:
  611. .. code-block:: pycon
  612. >>> from django.db.models import Avg, F, Window
  613. >>> Movie.objects.annotate(
  614. ... avg_rating=Window(
  615. ... expression=Avg("rating"),
  616. ... partition_by=[F("studio"), F("genre")],
  617. ... order_by="released__year",
  618. ... ),
  619. ... )
  620. This allows you to check if a movie is rated better or worse than its peers.
  621. You may want to apply multiple expressions over the same window, i.e., the
  622. same partition and frame. For example, you could modify the previous example
  623. to also include the best and worst rating in each movie's group (same studio,
  624. genre, and release year) by using three window functions in the same query. The
  625. partition and ordering from the previous example is extracted into a dictionary
  626. to reduce repetition:
  627. .. code-block:: pycon
  628. >>> from django.db.models import Avg, F, Max, Min, Window
  629. >>> window = {
  630. ... "partition_by": [F("studio"), F("genre")],
  631. ... "order_by": "released__year",
  632. ... }
  633. >>> Movie.objects.annotate(
  634. ... avg_rating=Window(
  635. ... expression=Avg("rating"),
  636. ... **window,
  637. ... ),
  638. ... best=Window(
  639. ... expression=Max("rating"),
  640. ... **window,
  641. ... ),
  642. ... worst=Window(
  643. ... expression=Min("rating"),
  644. ... **window,
  645. ... ),
  646. ... )
  647. Filtering against window functions is supported as long as lookups are not
  648. disjunctive (not using ``OR`` or ``XOR`` as a connector) and against a queryset
  649. performing aggregation.
  650. For example, a query that relies on aggregation and has an ``OR``-ed filter
  651. against a window function and a field is not supported. Applying combined
  652. predicates post-aggregation could cause rows that would normally be excluded
  653. from groups to be included:
  654. .. code-block:: pycon
  655. >>> qs = Movie.objects.annotate(
  656. ... category_rank=Window(Rank(), partition_by="category", order_by="-rating"),
  657. ... scenes_count=Count("actors"),
  658. ... ).filter(Q(category_rank__lte=3) | Q(title__contains="Batman"))
  659. >>> list(qs)
  660. NotImplementedError: Heterogeneous disjunctive predicates against window functions
  661. are not implemented when performing conditional aggregation.
  662. Among Django's built-in database backends, MySQL, PostgreSQL, and Oracle
  663. support window expressions. Support for different window expression features
  664. varies among the different databases. For example, the options in
  665. :meth:`~django.db.models.Expression.asc` and
  666. :meth:`~django.db.models.Expression.desc` may not be supported. Consult the
  667. documentation for your database as needed.
  668. .. _window-frames:
  669. Frames
  670. ~~~~~~
  671. For a window frame, you can choose either a range-based sequence of rows or an
  672. ordinary sequence of rows.
  673. .. class:: ValueRange(start=None, end=None, exclusion=None)
  674. .. attribute:: frame_type
  675. This attribute is set to ``'RANGE'``.
  676. PostgreSQL has limited support for ``ValueRange`` and only supports use of
  677. the standard start and end points, such as ``CURRENT ROW`` and ``UNBOUNDED
  678. FOLLOWING``.
  679. .. class:: RowRange(start=None, end=None, exclusion=None)
  680. .. attribute:: frame_type
  681. This attribute is set to ``'ROWS'``.
  682. Both classes return SQL with the template:
  683. .. code-block:: sql
  684. %(frame_type)s BETWEEN %(start)s AND %(end)s
  685. .. class:: WindowFrameExclusion
  686. .. attribute:: CURRENT_ROW
  687. .. attribute:: GROUP
  688. .. attribute:: TIES
  689. .. attribute:: NO_OTHERS
  690. The ``exclusion`` argument allows excluding rows
  691. (:attr:`~WindowFrameExclusion.CURRENT_ROW`), groups
  692. (:attr:`~WindowFrameExclusion.GROUP`), and ties
  693. (:attr:`~WindowFrameExclusion.TIES`) from the window frames on supported
  694. databases:
  695. .. code-block:: sql
  696. %(frame_type)s BETWEEN %(start)s AND %(end)s EXCLUDE %(exclusion)s
  697. Frames narrow the rows that are used for computing the result. They shift from
  698. some start point to some specified end point. Frames can be used with and
  699. without partitions, but it's often a good idea to specify an ordering of the
  700. window to ensure a deterministic result. In a frame, a peer in a frame is a row
  701. with an equivalent value, or all rows if an ordering clause isn't present.
  702. The default starting point for a frame is ``UNBOUNDED PRECEDING`` which is the
  703. first row of the partition. The end point is always explicitly included in the
  704. SQL generated by the ORM and is by default ``UNBOUNDED FOLLOWING``. The default
  705. frame includes all rows from the partition to the last row in the set.
  706. The accepted values for the ``start`` and ``end`` arguments are ``None``, an
  707. integer, or zero. A negative integer for ``start`` results in ``N PRECEDING``,
  708. while ``None`` yields ``UNBOUNDED PRECEDING``. In ``ROWS`` mode, a positive
  709. integer can be used for ``start`` resulting in ``N FOLLOWING``. Positive
  710. integers are accepted for ``end`` and results in ``N FOLLOWING``. In ``ROWS``
  711. mode, a negative integer can be used for ``end`` resulting in ``N PRECEDING``.
  712. For both ``start`` and ``end``, zero will return ``CURRENT ROW``.
  713. There's a difference in what ``CURRENT ROW`` includes. When specified in
  714. ``ROWS`` mode, the frame starts or ends with the current row. When specified in
  715. ``RANGE`` mode, the frame starts or ends at the first or last peer according to
  716. the ordering clause. Thus, ``RANGE CURRENT ROW`` evaluates the expression for
  717. rows which have the same value specified by the ordering. Because the template
  718. includes both the ``start`` and ``end`` points, this may be expressed with::
  719. ValueRange(start=0, end=0)
  720. If a movie's "peers" are described as movies released by the same studio in the
  721. same genre in the same year, this ``RowRange`` example annotates each movie
  722. with the average rating of a movie's two prior and two following peers:
  723. .. code-block:: pycon
  724. >>> from django.db.models import Avg, F, RowRange, Window
  725. >>> Movie.objects.annotate(
  726. ... avg_rating=Window(
  727. ... expression=Avg("rating"),
  728. ... partition_by=[F("studio"), F("genre")],
  729. ... order_by="released__year",
  730. ... frame=RowRange(start=-2, end=2),
  731. ... ),
  732. ... )
  733. If the database supports it, you can specify the start and end points based on
  734. values of an expression in the partition. If the ``released`` field of the
  735. ``Movie`` model stores the release month of each movie, this ``ValueRange``
  736. example annotates each movie with the average rating of a movie's peers
  737. released between twelve months before and twelve months after each movie:
  738. .. code-block:: pycon
  739. >>> from django.db.models import Avg, F, ValueRange, Window
  740. >>> Movie.objects.annotate(
  741. ... avg_rating=Window(
  742. ... expression=Avg("rating"),
  743. ... partition_by=[F("studio"), F("genre")],
  744. ... order_by="released__year",
  745. ... frame=ValueRange(start=-12, end=12),
  746. ... ),
  747. ... )
  748. .. currentmodule:: django.db.models
  749. Technical Information
  750. =====================
  751. Below you'll find technical implementation details that may be useful to
  752. library authors. The technical API and examples below will help with
  753. creating generic query expressions that can extend the built-in functionality
  754. that Django provides.
  755. Expression API
  756. --------------
  757. Query expressions implement the :ref:`query expression API <query-expression>`,
  758. but also expose a number of extra methods and attributes listed below. All
  759. query expressions must inherit from ``Expression()`` or a relevant
  760. subclass.
  761. When a query expression wraps another expression, it is responsible for
  762. calling the appropriate methods on the wrapped expression.
  763. .. class:: Expression
  764. .. attribute:: allowed_default
  765. Tells Django that this expression can be used in
  766. :attr:`Field.db_default`. Defaults to ``False``.
  767. .. attribute:: constraint_validation_compatible
  768. Tells Django that this expression can be used during a constraint
  769. validation. Expressions with ``constraint_validation_compatible`` set
  770. to ``False`` must have only one source expression. Defaults to
  771. ``True``.
  772. .. attribute:: contains_aggregate
  773. Tells Django that this expression contains an aggregate and that a
  774. ``GROUP BY`` clause needs to be added to the query.
  775. .. attribute:: contains_over_clause
  776. Tells Django that this expression contains a
  777. :class:`~django.db.models.expressions.Window` expression. It's used,
  778. for example, to disallow window function expressions in queries that
  779. modify data.
  780. .. attribute:: filterable
  781. Tells Django that this expression can be referenced in
  782. :meth:`.QuerySet.filter`. Defaults to ``True``.
  783. .. attribute:: window_compatible
  784. Tells Django that this expression can be used as the source expression
  785. in :class:`~django.db.models.expressions.Window`. Defaults to
  786. ``False``.
  787. .. attribute:: empty_result_set_value
  788. Tells Django which value should be returned when the expression is used
  789. to apply a function over an empty result set. Defaults to
  790. :py:data:`NotImplemented` which forces the expression to be computed on
  791. the database.
  792. .. attribute:: set_returning
  793. .. versionadded:: 5.2
  794. Tells Django that this expression contains a set-returning function,
  795. enforcing subquery evaluation. It's used, for example, to allow some
  796. Postgres set-returning functions (e.g. ``JSONB_PATH_QUERY``,
  797. ``UNNEST``, etc.) to skip optimization and be properly evaluated when
  798. annotations spawn rows themselves. Defaults to ``False``.
  799. .. attribute:: allows_composite_expressions
  800. .. versionadded:: 5.2
  801. Tells Django that this expression allows composite expressions, for
  802. example, to support :ref:`composite primary keys
  803. <cpk-and-database-functions>`. Defaults to ``False``.
  804. .. method:: resolve_expression(query=None, allow_joins=True, reuse=None, summarize=False, for_save=False)
  805. Provides the chance to do any preprocessing or validation of
  806. the expression before it's added to the query. ``resolve_expression()``
  807. must also be called on any nested expressions. A ``copy()`` of ``self``
  808. should be returned with any necessary transformations.
  809. ``query`` is the backend query implementation.
  810. ``allow_joins`` is a boolean that allows or denies the use of
  811. joins in the query.
  812. ``reuse`` is a set of reusable joins for multi-join scenarios.
  813. ``summarize`` is a boolean that, when ``True``, signals that the
  814. query being computed is a terminal aggregate query.
  815. ``for_save`` is a boolean that, when ``True``, signals that the query
  816. being executed is performing a create or update.
  817. .. method:: get_source_expressions()
  818. Returns an ordered list of inner expressions. For example:
  819. .. code-block:: pycon
  820. >>> Sum(F("foo")).get_source_expressions()
  821. [F('foo')]
  822. .. method:: set_source_expressions(expressions)
  823. Takes a list of expressions and stores them such that
  824. ``get_source_expressions()`` can return them.
  825. .. method:: relabeled_clone(change_map)
  826. Returns a clone (copy) of ``self``, with any column aliases relabeled.
  827. Column aliases are renamed when subqueries are created.
  828. ``relabeled_clone()`` should also be called on any nested expressions
  829. and assigned to the clone.
  830. ``change_map`` is a dictionary mapping old aliases to new aliases.
  831. Example::
  832. def relabeled_clone(self, change_map):
  833. clone = copy.copy(self)
  834. clone.expression = self.expression.relabeled_clone(change_map)
  835. return clone
  836. .. method:: convert_value(value, expression, connection)
  837. A hook allowing the expression to coerce ``value`` into a more
  838. appropriate type.
  839. ``expression`` is the same as ``self``.
  840. .. method:: get_group_by_cols()
  841. Responsible for returning the list of columns references by
  842. this expression. ``get_group_by_cols()`` should be called on any
  843. nested expressions. ``F()`` objects, in particular, hold a reference
  844. to a column.
  845. .. method:: asc(nulls_first=None, nulls_last=None)
  846. Returns the expression ready to be sorted in ascending order.
  847. ``nulls_first`` and ``nulls_last`` define how null values are sorted.
  848. See :ref:`using-f-to-sort-null-values` for example usage.
  849. .. method:: desc(nulls_first=None, nulls_last=None)
  850. Returns the expression ready to be sorted in descending order.
  851. ``nulls_first`` and ``nulls_last`` define how null values are sorted.
  852. See :ref:`using-f-to-sort-null-values` for example usage.
  853. .. method:: reverse_ordering()
  854. Returns ``self`` with any modifications required to reverse the sort
  855. order within an ``order_by`` call. As an example, an expression
  856. implementing ``NULLS LAST`` would change its value to be
  857. ``NULLS FIRST``. Modifications are only required for expressions that
  858. implement sort order like ``OrderBy``. This method is called when
  859. :meth:`~django.db.models.query.QuerySet.reverse()` is called on a
  860. queryset.
  861. Writing your own Query Expressions
  862. ----------------------------------
  863. You can write your own query expression classes that use, and can integrate
  864. with, other query expressions. Let's step through an example by writing an
  865. implementation of the ``COALESCE`` SQL function, without using the built-in
  866. :ref:`Func() expressions <func-expressions>`.
  867. The ``COALESCE`` SQL function is defined as taking a list of columns or
  868. values. It will return the first column or value that isn't ``NULL``.
  869. We'll start by defining the template to be used for SQL generation and
  870. an ``__init__()`` method to set some attributes::
  871. import copy
  872. from django.db.models import Expression
  873. class Coalesce(Expression):
  874. template = "COALESCE( %(expressions)s )"
  875. def __init__(self, expressions, output_field):
  876. super().__init__(output_field=output_field)
  877. if len(expressions) < 2:
  878. raise ValueError("expressions must have at least 2 elements")
  879. for expression in expressions:
  880. if not hasattr(expression, "resolve_expression"):
  881. raise TypeError("%r is not an Expression" % expression)
  882. self.expressions = expressions
  883. We do some basic validation on the parameters, including requiring at least 2
  884. columns or values, and ensuring they are expressions. We are requiring
  885. :ref:`output_field<output-field>` here so that Django knows what kind of model
  886. field to assign the eventual result to.
  887. Now we implement the preprocessing and validation. Since we do not have
  888. any of our own validation at this point, we delegate to the nested
  889. expressions::
  890. def resolve_expression(
  891. self, query=None, allow_joins=True, reuse=None, summarize=False, for_save=False
  892. ):
  893. c = self.copy()
  894. c.is_summary = summarize
  895. for pos, expression in enumerate(self.expressions):
  896. c.expressions[pos] = expression.resolve_expression(
  897. query, allow_joins, reuse, summarize, for_save
  898. )
  899. return c
  900. Next, we write the method responsible for generating the SQL::
  901. def as_sql(self, compiler, connection, template=None):
  902. sql_expressions, sql_params = [], []
  903. for expression in self.expressions:
  904. sql, params = compiler.compile(expression)
  905. sql_expressions.append(sql)
  906. sql_params.extend(params)
  907. template = template or self.template
  908. data = {"expressions": ",".join(sql_expressions)}
  909. return template % data, sql_params
  910. def as_oracle(self, compiler, connection):
  911. """
  912. Example of vendor specific handling (Oracle in this case).
  913. Let's make the function name lowercase.
  914. """
  915. return self.as_sql(compiler, connection, template="coalesce( %(expressions)s )")
  916. ``as_sql()`` methods can support custom keyword arguments, allowing
  917. ``as_vendorname()`` methods to override data used to generate the SQL string.
  918. Using ``as_sql()`` keyword arguments for customization is preferable to
  919. mutating ``self`` within ``as_vendorname()`` methods as the latter can lead to
  920. errors when running on different database backends. If your class relies on
  921. class attributes to define data, consider allowing overrides in your
  922. ``as_sql()`` method.
  923. We generate the SQL for each of the ``expressions`` by using the
  924. ``compiler.compile()`` method, and join the result together with commas.
  925. Then the template is filled out with our data and the SQL and parameters
  926. are returned.
  927. We've also defined a custom implementation that is specific to the Oracle
  928. backend. The ``as_oracle()`` function will be called instead of ``as_sql()``
  929. if the Oracle backend is in use.
  930. Finally, we implement the rest of the methods that allow our query expression
  931. to play nice with other query expressions::
  932. def get_source_expressions(self):
  933. return self.expressions
  934. def set_source_expressions(self, expressions):
  935. self.expressions = expressions
  936. Let's see how it works:
  937. .. code-block:: pycon
  938. >>> from django.db.models import F, Value, CharField
  939. >>> qs = Company.objects.annotate(
  940. ... tagline=Coalesce(
  941. ... [F("motto"), F("ticker_name"), F("description"), Value("No Tagline")],
  942. ... output_field=CharField(),
  943. ... )
  944. ... )
  945. >>> for c in qs:
  946. ... print("%s: %s" % (c.name, c.tagline))
  947. ...
  948. Google: Do No Evil
  949. Apple: AAPL
  950. Yahoo: Internet Company
  951. Django Software Foundation: No Tagline
  952. .. _avoiding-sql-injection-in-query-expressions:
  953. Avoiding SQL injection
  954. ~~~~~~~~~~~~~~~~~~~~~~
  955. Since a ``Func``'s keyword arguments for ``__init__()`` (``**extra``) and
  956. ``as_sql()`` (``**extra_context``) are interpolated into the SQL string rather
  957. than passed as query parameters (where the database driver would escape them),
  958. they must not contain untrusted user input.
  959. For example, if ``substring`` is user-provided, this function is vulnerable to
  960. SQL injection::
  961. from django.db.models import Func
  962. class Position(Func):
  963. function = "POSITION"
  964. template = "%(function)s('%(substring)s' in %(expressions)s)"
  965. def __init__(self, expression, substring):
  966. # substring=substring is an SQL injection vulnerability!
  967. super().__init__(expression, substring=substring)
  968. This function generates an SQL string without any parameters. Since
  969. ``substring`` is passed to ``super().__init__()`` as a keyword argument, it's
  970. interpolated into the SQL string before the query is sent to the database.
  971. Here's a corrected rewrite::
  972. class Position(Func):
  973. function = "POSITION"
  974. arg_joiner = " IN "
  975. def __init__(self, expression, substring):
  976. super().__init__(substring, expression)
  977. With ``substring`` instead passed as a positional argument, it'll be passed as
  978. a parameter in the database query.
  979. Adding support in third-party database backends
  980. -----------------------------------------------
  981. If you're using a database backend that uses a different SQL syntax for a
  982. certain function, you can add support for it by monkey patching a new method
  983. onto the function's class.
  984. Let's say we're writing a backend for Microsoft's SQL Server which uses the SQL
  985. ``LEN`` instead of ``LENGTH`` for the :class:`~functions.Length` function.
  986. We'll monkey patch a new method called ``as_sqlserver()`` onto the ``Length``
  987. class::
  988. from django.db.models.functions import Length
  989. def sqlserver_length(self, compiler, connection):
  990. return self.as_sql(compiler, connection, function="LEN")
  991. Length.as_sqlserver = sqlserver_length
  992. You can also customize the SQL using the ``template`` parameter of ``as_sql()``.
  993. We use ``as_sqlserver()`` because ``django.db.connection.vendor`` returns
  994. ``sqlserver`` for the backend.
  995. Third-party backends can register their functions in the top level
  996. ``__init__.py`` file of the backend package or in a top level ``expressions.py``
  997. file (or package) that is imported from the top level ``__init__.py``.
  998. For user projects wishing to patch the backend that they're using, this code
  999. should live in an :meth:`AppConfig.ready()<django.apps.AppConfig.ready>` method.