darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-11 10:28:55 -04:00
Code Issues Packages Projects Releases Wiki Activity

Adding another set of columns to the migration.

Browse Source 
Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
Cliff Hill
2024-02-28 12:44:18 -05:00
parent b1a53c76ee
commit 544bf35c06
2 changed files with 51 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/models.py
+3 -3
View File
@@ -2138,9 +2138,9 @@ class FactNotificationStatus(db.Model):
nullable=False, nullable=False,
) )
job_id = db.Column(UUID(as_uuid=True), primary_key=True, index=True, nullable=False) job_id = db.Column(UUID(as_uuid=True), primary_key=True, index=True, nullable=False)
notification_type = db.Column(db.Text, primary_key=True, nullable=False) notification_type = enum_column(NotificationType, primary_key=True, nullable=False)
key_type = db.Column(db.Text, primary_key=True, nullable=False) key_type = enum_column(KeyType, primary_key=True, nullable=False)
notification_status = db.Column(db.Text, primary_key=True, nullable=False) notification_status = enum_column(NotificationStatus, primary_key=True, nullable=False)
notification_count = db.Column(db.Integer(), nullable=False) notification_count = db.Column(db.Integer(), nullable=False)
created_at = db.Column( created_at = db.Column(
db.DateTime, db.DateTime,
migrations/versions/0410_enums_for_everything.py
+48
View File
@@ -560,6 +560,30 @@ def upgrade():
existing_nullable=False, existing_nullable=False,
postgresql_using=enum_using("code_type", CodeType), postgresql_using=enum_using("code_type", CodeType),
) )
op.alter_column(
"ft_notification_status",
"notification_type",
existing_type=sa.TEXT(),
type_=enum_type(NotificationType),
existing_nullable=False,
postgresql_using=enum_using("notification_type", NotificationType),
)
op.alter_column(
"ft_notification_status",
"key_type",
existing_type=sa.TEXT(),
type_=enum_type(KeyType),
existing_nullable=False,
postgresql_using=enum_using("key_type", KeyType),
)
op.alter_column(
"ft_notification_status",
"notification_type",
existing_type=sa.TEXT(),
type_=enum_type(NotificationStatus),
existing_nullable=False,
postgresql_using=enum_using("notification_status", NotificationStatus),
)
# Drop old enum types. # Drop old enum types.
enum_drop( enum_drop(
@@ -587,6 +611,30 @@ def downgrade():
enum_create(values=["email", "sms"], name="verify_code_types") enum_create(values=["email", "sms"], name="verify_code_types")
# Alter columns back # Alter columns back
op.alter_column(
"ft_notification_status",
"notification_type",
existing_type=enum_type(NotificationStatus),
type_=sa.TEXT(),
existing_nullable=False,
postgresql_using=enum_using("notification_status", NotificationStatus),
)
op.alter_column(
"ft_notification_status",
"key_type",
erxisting_type=enum_type(KeyType),
type_=sa.TEXT(),
existing_nullable=False,
postgresql_using=enum_using("key_type", KeyType),
)
op.alter_column(
"ft_notification_status",
"notification_type",
existing_type=enum_type(NotificationType),
type_=sa.TEXT(),
existing_nullable=False,
postgresql_using=enum_using("notification_type", NotificationType),
)
op.alter_column( op.alter_column(
"verify_codes", "verify_codes",
"code_type", "code_type",