⚗ cms migration

Author: hardbyte

alembic/versions/156d8781d7b8_add_vertexai_orgin.py

@@ -5,13 +5,14 @@
 Create Date: 2024-06-09 18:29:19.197616
 
 """
-from alembic import op
+
 import sqlalchemy as sa
 
+from alembic import op
 
 # revision identifiers, used by Alembic.
-revision = '156d8781d7b8'
-down_revision = '7dd85b891761'
+revision = "156d8781d7b8"
+down_revision = "7dd85b891761"
 branch_labels = None
 depends_on = None
 old_values = """'HUMAN', 'GPT4', 'PREDICTED_NIELSEN', 'NIELSEN_CBMC', 'NIELSEN_BIC', 'NIELSEN_THEMA', 'NIELSEN_IA', 'NIELSEN_RA', 'CLUSTER_RELEVANCE', 'CLUSTER_ZAINAB', 'OTHER'"""

alembic/versions/281723ba07be_add_cms_content_table.py

@@ -23,6 +23,15 @@ def upgrade():
         "JOKE", "QUESTION", "FACT", "QUOTE", name="enum_cms_content_type"
     )
 
+    cms_status_enum = sa.Enum(
+        "DRAFT",
+        "PENDING_REVIEW",
+        "APPROVED",
+        "PUBLISHED",
+        "ARCHIVED",
+        name="enum_cms_content_status",
+    )
+
     op.create_table(
         "cms_content",
         sa.Column(
@@ -33,7 +42,23 @@ def upgrade():
             cms_types_enum,
             nullable=False,
         ),
-        sa.Column("content", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column(
+            "status", cms_status_enum, server_default=sa.text("'DRAFT'"), nullable=False
+        ),
+        sa.Column("version", sa.Integer(), server_default=sa.text("1"), nullable=False),
+        sa.Column("content", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
+        sa.Column(
+            "info",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::json"),
+            nullable=False,
+        ),
+        sa.Column(
+            "tags",
+            postgresql.ARRAY(sa.String()),
+            server_default=sa.text("'{}'::text[]"),
+            nullable=False,
+        ),
         sa.Column(
             "created_at",
             sa.DateTime(),
@@ -46,14 +71,18 @@ def upgrade():
             server_default=sa.text("CURRENT_TIMESTAMP"),
             nullable=False,
         ),
-        sa.Column("user_id", sa.UUID(), nullable=True),
+        sa.Column("created_by", sa.UUID(), nullable=True),
         sa.ForeignKeyConstraint(
-            ["user_id"], ["users.id"], name="fk_content_user", ondelete="CASCADE"
+            ["created_by"], ["users.id"], name="fk_content_user", ondelete="SET NULL"
         ),
         sa.PrimaryKeyConstraint("id"),
     )
-    op.create_index(op.f("ix_cms_content_id"), "cms_content", ["id"], unique=True)
+
     op.create_index(op.f("ix_cms_content_type"), "cms_content", ["type"], unique=False)
+    op.create_index(
+        op.f("ix_cms_content_status"), "cms_content", ["status"], unique=False
+    )
+    op.create_index(op.f("ix_cms_content_tags"), "cms_content", ["tags"], unique=False)
 
 
 def downgrade():

alembic/versions/35112b0ae03e_make_username_non_nullable_for_readers.py

@@ -8,7 +8,6 @@
 
 # revision identifiers, used by Alembic.
 
-
 revision = "35112b0ae03e"
 down_revision = "77c90a741ba7"
 branch_labels = None

alembic/versions/a65ff088f9ae_create_country_tables.py

@@ -1,7 +1,7 @@
 """Create original tables
 
 Revision ID: a65ff088f9ae
-Revises: 
+Revises:
 Create Date: 2021-12-27 10:15:54.848632
 
 """