Move % addition to lookups, refactor postgres lookups.

These refactorings making overriding some text based lookup names on
other fields (specifically `contains`) much cleaner. It also removes a
bunch of duplication in the contrib.postgres lookups.

2 files changed, 22 insertions, 92 deletions

diff --git a/django/contrib/postgres/fields/array.py b/django/contrib/postgres/fields/array.py
index 65f3dc6f6a..318afabd2c 100644
--- a/django/contrib/postgres/fields/array.py
+++ b/django/contrib/postgres/fields/array.py
@@ -1,9 +1,10 @@
 import json
 
+from django.contrib.postgres import lookups
 from django.contrib.postgres.forms import SimpleArrayField
 from django.contrib.postgres.validators import ArrayMaxLengthValidator
 from django.core import checks, exceptions
-from django.db.models import Field, Lookup, Transform, IntegerField
+from django.db.models import Field, Transform, IntegerField
 from django.utils import six
 from django.utils.translation import string_concat, ugettext_lazy as _
 
@@ -74,12 +75,6 @@ class ArrayField(Field):
             return [self.base_field.get_prep_value(i) for i in value]
         return value
 
-    def get_db_prep_lookup(self, lookup_type, value, connection, prepared=False):
-        if lookup_type == 'contains':
-            return [self.get_prep_value(value)]
-        return super(ArrayField, self).get_db_prep_lookup(lookup_type, value,
-                connection, prepared=False)
-
     def deconstruct(self):
         name, path, args, kwargs = super(ArrayField, self).deconstruct()
         if path == 'django.contrib.postgres.fields.array.ArrayField':
@@ -156,46 +151,21 @@ class ArrayField(Field):
 
 
 @ArrayField.register_lookup
-class ArrayContainsLookup(Lookup):
-    lookup_name = 'contains'
-
-    def as_sql(self, compiler, connection):
-        lhs, lhs_params = self.process_lhs(compiler, connection)
-        rhs, rhs_params = self.process_rhs(compiler, connection)
-        params = lhs_params + rhs_params
-        type_cast = self.lhs.output_field.db_type(connection)
-        return '%s @> %s::%s' % (lhs, rhs, type_cast), params
-
-
-@ArrayField.register_lookup
-class ArrayContainedByLookup(Lookup):
-    lookup_name = 'contained_by'
-
-    def as_sql(self, compiler, connection):
-        lhs, lhs_params = self.process_lhs(compiler, connection)
-        rhs, rhs_params = self.process_rhs(compiler, connection)
-        params = lhs_params + rhs_params
-        return '%s <@ %s' % (lhs, rhs), params
-
+class ArrayContains(lookups.DataContains):
+    def as_sql(self, qn, connection):
+        sql, params = super(ArrayContains, self).as_sql(qn, connection)
+        sql += '::%s' % self.lhs.output_field.db_type(connection)
+        return sql, params
 
-@ArrayField.register_lookup
-class ArrayOverlapLookup(Lookup):
-    lookup_name = 'overlap'
 
-    def as_sql(self, compiler, connection):
-        lhs, lhs_params = self.process_lhs(compiler, connection)
-        rhs, rhs_params = self.process_rhs(compiler, connection)
-        params = lhs_params + rhs_params
-        return '%s && %s' % (lhs, rhs), params
+ArrayField.register_lookup(lookups.ContainedBy)
+ArrayField.register_lookup(lookups.Overlap)
 
 
 @ArrayField.register_lookup
 class ArrayLenTransform(Transform):
     lookup_name = 'len'
-
-    @property
-    def output_field(self):
-        return IntegerField()
+    output_field = IntegerField()
 
     def as_sql(self, compiler, connection):
         lhs, params = compiler.compile(self.lhs)
diff --git a/django/contrib/postgres/fields/hstore.py b/django/contrib/postgres/fields/hstore.py
index be998488fb..1524368ecf 100644
--- a/django/contrib/postgres/fields/hstore.py
+++ b/django/contrib/postgres/fields/hstore.py
@@ -1,9 +1,9 @@
 import json
 
-from django.contrib.postgres import forms
+from django.contrib.postgres import forms, lookups
 from django.contrib.postgres.fields.array import ArrayField
 from django.core import exceptions
-from django.db.models import Field, Lookup, Transform, TextField
+from django.db.models import Field, Transform, TextField
 from django.utils import six
 from django.utils.translation import ugettext_lazy as _
 
@@ -21,12 +21,6 @@ class HStoreField(Field):
     def db_type(self, connection):
         return 'hstore'
 
-    def get_db_prep_lookup(self, lookup_type, value, connection, prepared=False):
-        if lookup_type == 'contains':
-            return [self.get_prep_value(value)]
-        return super(HStoreField, self).get_db_prep_lookup(lookup_type, value,
-                connection, prepared=False)
-
     def get_transform(self, name):
         transform = super(HStoreField, self).get_transform(name)
         if transform:
@@ -60,48 +54,20 @@ class HStoreField(Field):
         return super(HStoreField, self).formfield(**defaults)
 
 
-@HStoreField.register_lookup
-class HStoreContainsLookup(Lookup):
-    lookup_name = 'contains'
-
-    def as_sql(self, compiler, connection):
-        lhs, lhs_params = self.process_lhs(compiler, connection)
-        rhs, rhs_params = self.process_rhs(compiler, connection)
-        params = lhs_params + rhs_params
-        return '%s @> %s' % (lhs, rhs), params
-
-
-@HStoreField.register_lookup
-class HStoreContainedByLookup(Lookup):
-    lookup_name = 'contained_by'
-
-    def as_sql(self, compiler, connection):
-        lhs, lhs_params = self.process_lhs(compiler, connection)
-        rhs, rhs_params = self.process_rhs(compiler, connection)
-        params = lhs_params + rhs_params
-        return '%s <@ %s' % (lhs, rhs), params
+HStoreField.register_lookup(lookups.DataContains)
+HStoreField.register_lookup(lookups.ContainedBy)
 
 
 @HStoreField.register_lookup
-class HasKeyLookup(Lookup):
+class HasKeyLookup(lookups.PostgresSimpleLookup):
     lookup_name = 'has_key'
-
-    def as_sql(self, compiler, connection):
-        lhs, lhs_params = self.process_lhs(compiler, connection)
-        rhs, rhs_params = self.process_rhs(compiler, connection)
-        params = lhs_params + rhs_params
-        return '%s ? %s' % (lhs, rhs), params
+    operator = '?'
 
 
 @HStoreField.register_lookup
-class HasKeysLookup(Lookup):
+class HasKeysLookup(lookups.PostgresSimpleLookup):
     lookup_name = 'has_keys'
-
-    def as_sql(self, compiler, connection):
-        lhs, lhs_params = self.process_lhs(compiler, connection)
-        rhs, rhs_params = self.process_rhs(compiler, connection)
-        params = lhs_params + rhs_params
-        return '%s ?& %s' % (lhs, rhs), params
+    operator = '?&'
 
 
 class KeyTransform(Transform):
@@ -126,20 +92,14 @@ class KeyTransformFactory(object):
 
 
 @HStoreField.register_lookup
-class KeysTransform(Transform):
+class KeysTransform(lookups.FunctionTransform):
     lookup_name = 'keys'
+    function = 'akeys'
     output_field = ArrayField(TextField())
 
-    def as_sql(self, compiler, connection):
-        lhs, params = compiler.compile(self.lhs)
-        return 'akeys(%s)' % lhs, params
-
 
 @HStoreField.register_lookup
-class ValuesTransform(Transform):
+class ValuesTransform(lookups.FunctionTransform):
     lookup_name = 'values'
+    function = 'avals'
     output_field = ArrayField(TextField())
-
-    def as_sql(self, compiler, connection):
-        lhs, params = compiler.compile(self.lhs)
-        return 'avals(%s)' % lhs, params