mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-08-04 10:59:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Fixed #23627 -- Allowed register_lookup to work as a decorator.

Browse source
This commit is contained in:
Marc Tamlyn 2014-10-09 17:04:50 +01:00
parent d6a87eefd8
commit 92a17eaae0
5 changed files with 22 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/custom_lookups/tests.py
View file

@ -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):