Update migration

Author: hardbyte

alembic/versions/281723ba07be_add_cms_content_table.py

@@ -19,14 +19,18 @@
 
 
 def upgrade():
+    cms_types_enum = sa.Enum(
+        "JOKE", "QUESTION", "FACT", "QUOTE", name="enum_cms_content_type"
+    )
+
     op.create_table(
         "cms_content",
         sa.Column(
             "id", sa.UUID(), server_default=sa.text("gen_random_uuid()"), nullable=False
         ),
         sa.Column(
             "type",
-            sa.Enum("JOKE", "QUESTION", "FACT", "QUOTE", name="enum_cms_content_type"),
+            cms_types_enum,
             nullable=False,
         ),
         sa.Column("content", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
@@ -56,3 +60,7 @@ def downgrade():
     op.drop_index(op.f("ix_cms_content_type"), table_name="cms_content")
     op.drop_index(op.f("ix_cms_content_id"), table_name="cms_content")
     op.drop_table("cms_content")
+
+    op.execute("DROP TYPE enum_cms_content_type")
+    genresource = sa.Enum(name="enum_cms_content_type")
+    genresource.drop(op.get_bind(), checkfirst=True)