diff options
| author | Marc Tamlyn <marc.tamlyn@gmail.com> | 2014-10-09 17:04:50 +0100 |
|---|---|---|
| committer | Marc Tamlyn <marc.tamlyn@gmail.com> | 2014-10-09 18:44:58 +0100 |
| commit | 92a17eaae081a213171b044858d6fc29df2df733 (patch) | |
| tree | eae95057cfb26745d2bd99dba63811aae6d5db8e /tests/custom_lookups | |
| parent | d6a87eefd87250e68457488cfd62fb2cc8211b24 (diff) | |
Fixed #23627 -- Allowed register_lookup to work as a decorator.
Diffstat (limited to 'tests/custom_lookups')
| -rw-r--r-- | tests/custom_lookups/tests.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/custom_lookups/tests.py b/tests/custom_lookups/tests.py index c89b268f6e..a965e5a4e8 100644 --- a/tests/custom_lookups/tests.py +++ b/tests/custom_lookups/tests.py @@ -50,6 +50,7 @@ class YearTransform(models.Transform): return models.IntegerField() +@YearTransform.register_lookup class YearExact(models.lookups.Lookup): lookup_name = 'exact' @@ -66,9 +67,9 @@ class YearExact(models.lookups.Lookup): return ("%(lhs)s >= (%(rhs)s || '-01-01')::date " "AND %(lhs)s <= (%(rhs)s || '-12-31')::date" % {'lhs': lhs_sql, 'rhs': rhs_sql}, params) -YearTransform.register_lookup(YearExact) +@YearTransform.register_lookup class YearLte(models.lookups.LessThanOrEqual): """ The purpose of this lookup is to efficiently compare the year of the field. @@ -86,7 +87,6 @@ class YearLte(models.lookups.LessThanOrEqual): # WHERE somecol <= '2013-12-31') # but also make it work if the rhs_sql is field reference. return "%s <= (%s || '-12-31')::date" % (lhs_sql, rhs_sql), params -YearTransform.register_lookup(YearLte) class SQLFunc(models.Lookup): |