Add changes.

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

@@ -1,49 +0,0 @@
-"""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 ###

moonstreamdb-v3/moonstreamdbv3/models_indexes.py

@@ -571,7 +571,6 @@ class AbiJobs(Base):
             "address",
             "abi_selector",
             "customer_id",
-            "user_id",
             name="uq_abi_jobs",
         ),
     )
@@ -580,7 +579,6 @@ class AbiJobs(Base):
     address = Column(LargeBinary, nullable=False, index=True)
     user_id = Column(UUID(as_uuid=True), nullable=False, index=True)
     customer_id = Column(UUID(as_uuid=True), nullable=True, index=True)
-    subscription_id = Column(UUID(as_uuid=True), nullable=True, index=True)
     abi_selector = Column(VARCHAR(256), nullable=False, index=True)
     chain = Column(VARCHAR(256), nullable=False, index=True)
     abi_name = Column(VARCHAR(256), nullable=False, index=True)
@@ -595,3 +593,20 @@ class AbiJobs(Base):
     updated_at = Column(
         DateTime(timezone=True), server_default=utcnow(), nullable=False
     )
+
+
+class AbiSubscriptions(Base):
+    __tablename__ = "abi_subscriptions"
+
+    __table_args__ = (PrimaryKeyConstraint("abi_job_id", "subscription_id"),)
+
+    abi_job_id = Column(
+        UUID(as_uuid=True),
+        ForeignKey("abi_jobs.id"),
+        nullable=False,
+        index=True,
+    )
+    subscription_id = Column(UUID(as_uuid=True), nullable=False, index=True)
+    created_at = Column(
+        DateTime(timezone=True), server_default=utcnow(), nullable=False
+    )