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

1 files changed, 4 insertions, 12 deletions

diff --git a/django/contrib/postgres/fields/array.py b/django/contrib/postgres/fields/array.py
index dea9c0440d..1a969d7c0e 100644
--- a/django/contrib/postgres/fields/array.py
+++ b/django/contrib/postgres/fields/array.py
@@ -160,6 +160,7 @@ class ArrayField(Field):
         return super(ArrayField, self).formfield(**defaults)
 
 
+@ArrayField.register_lookup
 class ArrayContainsLookup(Lookup):
     lookup_name = 'contains'
 
@@ -171,9 +172,7 @@ class ArrayContainsLookup(Lookup):
         return '%s @> %s::%s' % (lhs, rhs, type_cast), params
 
 
-ArrayField.register_lookup(ArrayContainsLookup)
-
-
+@ArrayField.register_lookup
 class ArrayContainedByLookup(Lookup):
     lookup_name = 'contained_by'
 
@@ -184,9 +183,7 @@ class ArrayContainedByLookup(Lookup):
         return '%s <@ %s' % (lhs, rhs), params
 
 
-ArrayField.register_lookup(ArrayContainedByLookup)
-
-
+@ArrayField.register_lookup
 class ArrayOverlapLookup(Lookup):
     lookup_name = 'overlap'
 
@@ -197,9 +194,7 @@ class ArrayOverlapLookup(Lookup):
         return '%s && %s' % (lhs, rhs), params
 
 
-ArrayField.register_lookup(ArrayOverlapLookup)
-
-
+@ArrayField.register_lookup
 class ArrayLenTransform(Transform):
     lookup_name = 'len'
 
@@ -212,9 +207,6 @@ class ArrayLenTransform(Transform):
         return 'array_length(%s, 1)' % lhs, params
 
 
-ArrayField.register_lookup(ArrayLenTransform)
-
-
 class IndexTransform(Transform):
 
     def __init__(self, index, base_field, *args, **kwargs):