fix(sqla_factory): fixed handling SQLA column constraints (litestar-o…

…rg#634)
nisemenov · Jan 31, 2025 · fcc9976 · fcc9976
1 parent 9332fa8
commit fcc9976
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/polyfactory/factories/sqlalchemy_factory.py b/polyfactory/factories/sqlalchemy_factory.py
@@ -175,6 +175,8 @@ def _get_type_from_type_engine(cls, type_engine: TypeEngine) -> type:
             for sqlalchemy_field, constraint_field in constraint_fields.items():
                 if (value := getattr(type_engine, sqlalchemy_field, None)) is not None:
                     constraints[constraint_field] = value  # type: ignore[literal-required]
+                if sqlalchemy_field == "precision":
+                    constraints.setdefault("decimal_places", 0)
         if constraints:
             annotation = Annotated[annotation, Frozendict(constraints)]  # type: ignore[assignment]