Another thing needs an enum.

Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
Cliff Hill
2024-01-25 14:25:00 -05:00
parent 6a8f7ffeb3
commit ced386b0c5
2 changed files with 37 additions and 10 deletions

View File

@@ -78,7 +78,8 @@ def enum_column(enum_type, **kwargs):
return db.Column(
db.Enum(
*[i.value for i in enum_type],
name=_enum_column_names[enum_type]
name=_enum_column_names[enum_type],
values_callable=(lambda x: [i.value for i in x]),
),
**kwargs,
)
@@ -1449,7 +1450,13 @@ class NotificationAllTimeView(db.Model):
sent_at = db.Column(db.DateTime)
sent_by = db.Column(db.String)
updated_at = db.Column(db.DateTime)
status = db.Column("notification_status", db.Text)
status = enum_column(
NotificationStatus,
name="notification_status",
nullable=True,
default=NotificationStatus.CREATED,
key="status",
)
reference = db.Column(db.String)
client_reference = db.Column(db.String)
international = db.Column(db.Boolean)