implemet unique index for sqlite

tortoise · Jun 8, 2024 · 9c8e553 · 9c8e553
1 parent faf0475
commit 9c8e553
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tortoise/contrib/sqlite/indexes.py b/tortoise/contrib/sqlite/indexes.py
@@ -0,0 +1,14 @@
+from typing import Sequence
+from tortoise.indexes import UniqueIndexABC
+
+
+class UniqueIndex(UniqueIndexABC[Sequence[str]]):
+    def nulls(self, null_fields: Sequence[str]):
+        conditions = tuple(
+            map(
+                lambda field_name: f"{field_name} is not null",
+                null_fields or [],
+            )
+        )
+        result = " and ".join(conditions)
+        return f"where {result}" if result else result