From 92a17eaae081a213171b044858d6fc29df2df733 Mon Sep 17 00:00:00 2001 From: Marc Tamlyn Date: Thu, 9 Oct 2014 17:04:50 +0100 Subject: Fixed #23627 -- Allowed register_lookup to work as a decorator. --- tests/custom_lookups/tests.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tests') 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): -- cgit v1.3