Add user_id as unique constraint.

moonstreamdb-v3/moonstreamdbv3/alembic_indexes/versions/2c5133943a38_add_subscription_id.py

@@ -0,0 +1,49 @@
+"""Add subscription id
+
+Revision ID: 2c5133943a38
+Revises: e02c90ea67bb
+Create Date: 2024-07-08 17:09:24.623322
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "2c5133943a38"
+down_revision: Union[str, None] = "e02c90ea67bb"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column("abi_jobs", sa.Column("subscription_id", sa.UUID(), nullable=True))
+    op.drop_constraint("uq_abi_jobs", "abi_jobs", type_="unique")
+    op.create_unique_constraint(
+        "uq_abi_jobs",
+        "abi_jobs",
+        ["chain", "address", "abi_selector", "customer_id", "user_id"],
+    )
+    op.create_index(
+        op.f("ix_abi_jobs_subscription_id"),
+        "abi_jobs",
+        ["subscription_id"],
+        unique=False,
+    )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index(op.f("ix_abi_jobs_subscription_id"), table_name="abi_jobs")
+    op.drop_constraint("uq_abi_jobs", "abi_jobs", type_="unique")
+    op.create_unique_constraint(
+        "uq_abi_jobs", "abi_jobs", ["chain", "address", "abi_selector", "customer_id"]
+    )
+    op.drop_column("abi_jobs", "subscription_id")
+    # ### end Alembic commands ###