Merge pull request #62 from Aristotle-Metadata-Enterprises/json-support

Add lookup support

garnett/lookups.py

@@ -125,15 +125,36 @@ def process_rhs(self, compiler, connection):
 
 
 @TranslatedKeyTransform.register_lookup
-class KeyTransformExact(json.KeyTransformExact):
+class KeyTransformExact(json.KeyTransformTextLookupMixin, lookups.Exact):
     def process_lhs(self, compiler, connection):
         self.lhs = Cast(self.lhs, CharField())
         return super().process_lhs(compiler, connection)
 
     def process_rhs(self, compiler, connection):
-        self.rhs = Cast(self.lhs, CharField())
+        return super().process_rhs(compiler, connection)
+
+
+@TranslatedKeyTransform.register_lookup
+class KeyTransformGreaterThan(json.KeyTransformTextLookupMixin, lookups.GreaterThan):
+    def process_lhs(self, compiler, connection):
+        self.lhs = Cast(self.lhs, CharField())
+        return super().process_lhs(compiler, connection)
+
+    def process_rhs(self, compiler, connection):
+        return super().process_rhs(compiler, connection)
+
+
+@TranslatedKeyTransform.register_lookup
+class KeyTransformGreaterThanOrEqual(
+    json.KeyTransformTextLookupMixin, lookups.GreaterThanOrEqual
+):
+    def process_lhs(self, compiler, connection):
+        self.lhs = Cast(self.lhs, CharField())
         return super().process_lhs(compiler, connection)
 
+    def process_rhs(self, compiler, connection):
+        return super().process_rhs(compiler, connection)
+
 
 @TranslatedField.register_lookup
 class BaseLanguageStartsWith(CurrentLanguageMixin, json.KeyTransformStartsWith):
@@ -189,6 +210,54 @@ def process_rhs(self, compiler, connection):
         return super().process_rhs(compiler, connection)
 
 
+@TranslatedField.register_lookup
+class BaseLanguageGreaterThan(
+    CurrentLanguageMixin, json.KeyTransformTextLookupMixin, lookups.GreaterThan
+):
+    def process_lhs(self, compiler, connection):
+        self.lhs = Cast(self.lhs, CharField())
+        return super().process_lhs(compiler, connection)
+
+    def process_rhs(self, compiler, connection):
+        return super().process_rhs(compiler, connection)
+
+
+@TranslatedField.register_lookup
+class BaseLanguageGreaterThanOrEqual(
+    CurrentLanguageMixin, json.KeyTransformTextLookupMixin, lookups.GreaterThanOrEqual
+):
+    def process_lhs(self, compiler, connection):
+        self.lhs = Cast(self.lhs, CharField())
+        return super().process_lhs(compiler, connection)
+
+    def process_rhs(self, compiler, connection):
+        return super().process_rhs(compiler, connection)
+
+
+@TranslatedField.register_lookup
+class BaseLanguageLessThan(
+    CurrentLanguageMixin, json.KeyTransformTextLookupMixin, lookups.LessThan
+):
+    def process_lhs(self, compiler, connection):
+        self.lhs = Cast(self.lhs, CharField())
+        return super().process_lhs(compiler, connection)
+
+    def process_rhs(self, compiler, connection):
+        return super().process_rhs(compiler, connection)
+
+
+@TranslatedField.register_lookup
+class BaseLanguageLessThanOrEqual(
+    CurrentLanguageMixin, json.KeyTransformTextLookupMixin, lookups.LessThanOrEqual
+):
+    def process_lhs(self, compiler, connection):
+        self.lhs = Cast(self.lhs, CharField())
+        return super().process_lhs(compiler, connection)
+
+    def process_rhs(self, compiler, connection):
+        return super().process_rhs(compiler, connection)
+
+
 @TranslatedField.register_lookup
 class BaseLanguageSearch(
     CurrentLanguageMixin, json.KeyTransformTextLookupMixin, SearchLookup

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "django-garnett"
-version = "0.4.1"
+version = "0.4.2"
 description = "Simple translatable Django fields"
 authors = ["Aristotle Metadata Enterprises"]
 license = "BSD-3-Clause"