Fix query building edgecases, to support Wagtail on workers-dbms

G4brym · Nov 10, 2024 · 103941a · 103941a
1 parent c7ba34c
commit 103941a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/django_dbms/base.py b/django_dbms/base.py
@@ -27,11 +27,20 @@ class DatabaseFeatures(SQLiteDatabaseFeatures):
 
 class DatabaseOperations(SQLiteDatabaseOperations):
     pragma_foreign_keys = None
+    cast_data_types = {
+        "DateField": "TEXT",
+        "DateTimeField": "TEXT",
+        "PositiveIntegerField": "INTEGER",
+    }
 
     def _quote_columns(self, sql):
         """
         Ensure column names are properly quoted and aliased to avoid collisions.
+        Also handles CAST statements properly for NULL values.
         """
+        # First, fix any problematic CAST statements with None
+        sql = sql.replace('None(', '(')
+
         parsed = sqlparse.parse(sql)
         if not parsed:
             return sql  # Unable to parse, return original SQL

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "django-cf"
-version = "0.0.16"
+version = "0.0.17"
 authors = [
   { name="Gabriel Massadas" },
 ]