Two more column types for check_constraint

hasgeek · Jan 3, 2024 · 12c22f7 · 12c22f7
1 parent f811ab8
commit 12c22f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/funnel/models/account.py b/funnel/models/account.py
@@ -232,7 +232,7 @@ class Account(UuidMixin, BaseMixin[int, 'Account'], Model):
     _state: Mapped[int] = sa_orm.mapped_column(
         'state',
         sa.SmallInteger,
-        StateManager.check_constraint('state', ACCOUNT_STATE),
+        StateManager.check_constraint('state', ACCOUNT_STATE, sa.SmallInteger),
         nullable=False,
         default=ACCOUNT_STATE.ACTIVE,
     )
@@ -246,7 +246,7 @@ class Account(UuidMixin, BaseMixin[int, 'Account'], Model):
     _profile_state: Mapped[int] = sa_orm.mapped_column(
         'profile_state',
         sa.SmallInteger,
-        StateManager.check_constraint('profile_state', PROFILE_STATE),
+        StateManager.check_constraint('profile_state', PROFILE_STATE, sa.SmallInteger),
         nullable=False,
         default=PROFILE_STATE.AUTO,
     )