[field] Use cast in where representation to handle user-defined domains.

collorg · Nov 22, 2024 · 188e06b · 188e06b
1 parent 2942b9e
commit 188e06b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/half_orm/field.py b/half_orm/field.py
@@ -83,10 +83,13 @@ def _where_repr(self, query, ho_id):
         where_repr = ''
         comp_str = '%s'
         comp = self._comp()
+        cast = ''
+        if self.__value != NULL:
+            cast = f'::{self.__sql_type}'
         if not self.unaccent:
-            where_repr = f"{self.__praf(query, ho_id)} {comp} {comp_str}"
+            where_repr = f"{self.__praf(query, ho_id)} {comp} {comp_str}{cast}"
         else:
-            where_repr = f"unaccent({self.__praf(query, ho_id)}) {comp} unaccent({comp_str})"
+            where_repr = f"unaccent({self.__praf(query, ho_id)}) {comp} unaccent({comp_str}{cast})"
         return where_repr
 
     @property

diff --git a/test/field/misc_test.py b/test/field/misc_test.py
@@ -123,7 +123,7 @@ def test_comps(self):
         self.post.content.set({'bonjour', 'au revoir'})
         self.assertIsInstance(self.post.content.value, tuple)
         self.post.content.unaccent = True
-        self.assertEqual(self.post.content._where_repr('', id(self.post)), 'unaccent("content") = unaccent(%s)')
+        self.assertEqual(self.post.content._where_repr('', id(self.post)), 'unaccent("content") = unaccent(%s::text)')
 
     def test_py_type(self):
         self.assertEqual(str(self.comment.tags.py_type), 'typing.List[str]')