Fix unique key case

adsharma · Jan 18, 2025 · ec46455 · ec46455
1 parent 7d1ec47
commit ec46455
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/fquery/sqlmodel.py b/fquery/sqlmodel.py
@@ -43,7 +43,7 @@
 
 
 def unique():
-    pass
+    return field(default=None, metadata={"SQL": {"unique": True}})
 
 
 def foreignkey(name):
@@ -85,11 +85,12 @@ def sqlmodel(self) -> SQLModel:
         return self.__sqlmodel__(**attrs)
 
     def get_field_def(cls, field) -> Union[Field, Relationship]:
-        if field.default == unique:
-            return Field(unique=True)
         sql_meta = field.metadata.get("SQL", {})
         has_foreign_key = bool(sql_meta.get("foreign_key", None))
         has_relationship = bool(sql_meta.get("relationship", None))
+        has_unique_constraint = sql_meta.get("unique", False)
+        if has_unique_constraint:
+            return Field(unique=True)
 
         if not sql_meta or not (has_foreign_key or has_relationship):
             return Field(