moonstream/moonstreamdb-v3/moonstreamdbv3/alembic/versions/1d53afc1eff4_add_b3.py

308 wiersze
11 KiB
Python

"""add-b3
Revision ID: 1d53afc1eff4
Revises: db9559f9566c
Create Date: 2024-09-11 14:19:12.762655
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision: str = "1d53afc1eff4"
down_revision: Union[str, None] = "db9559f9566c"
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.create_table(
"b3_labels",
sa.Column("id", sa.UUID(), nullable=False),
sa.Column("label", sa.VARCHAR(length=256), nullable=False),
sa.Column("transaction_hash", sa.VARCHAR(length=128), nullable=False),
sa.Column("log_index", sa.Integer(), nullable=True),
sa.Column("block_number", sa.BigInteger(), nullable=False),
sa.Column("block_hash", sa.VARCHAR(length=256), nullable=False),
sa.Column("block_timestamp", sa.BigInteger(), nullable=False),
sa.Column("caller_address", sa.LargeBinary(), nullable=True),
sa.Column("origin_address", sa.LargeBinary(), nullable=True),
sa.Column("address", sa.LargeBinary(), nullable=False),
sa.Column("label_name", sa.Text(), nullable=True),
sa.Column("label_type", sa.VARCHAR(length=64), nullable=True),
sa.Column("label_data", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
sa.Column(
"created_at",
sa.DateTime(timezone=True),
server_default=sa.text("TIMEZONE('utc', statement_timestamp())"),
nullable=False,
),
sa.PrimaryKeyConstraint("id", name=op.f("pk_b3_labels")),
sa.UniqueConstraint("id", name=op.f("uq_b3_labels_id")),
)
op.create_index(
"ix_b3_labels_addr_block_num",
"b3_labels",
["address", "block_number"],
unique=False,
)
op.create_index(
"ix_b3_labels_addr_block_ts",
"b3_labels",
["address", "block_timestamp"],
unique=False,
)
op.create_index(
op.f("ix_b3_labels_address"), "b3_labels", ["address"], unique=False
)
op.create_index(
op.f("ix_b3_labels_block_number"), "b3_labels", ["block_number"], unique=False
)
op.create_index(
op.f("ix_b3_labels_caller_address"),
"b3_labels",
["caller_address"],
unique=False,
)
op.create_index(op.f("ix_b3_labels_label"), "b3_labels", ["label"], unique=False)
op.create_index(
op.f("ix_b3_labels_label_name"), "b3_labels", ["label_name"], unique=False
)
op.create_index(
op.f("ix_b3_labels_label_type"), "b3_labels", ["label_type"], unique=False
)
op.create_index(
op.f("ix_b3_labels_origin_address"),
"b3_labels",
["origin_address"],
unique=False,
)
op.create_index(
op.f("ix_b3_labels_transaction_hash"),
"b3_labels",
["transaction_hash"],
unique=False,
)
op.create_index(
"uk_b3_labels_tx_hash_log_idx_evt",
"b3_labels",
["transaction_hash", "log_index"],
unique=True,
postgresql_where=sa.text("label='seer' and label_type='event'"),
)
op.create_index(
"uk_b3_labels_tx_hash_log_idx_evt_raw",
"b3_labels",
["transaction_hash", "log_index"],
unique=True,
postgresql_where=sa.text("label='seer-raw' and label_type='event'"),
)
op.create_index(
"uk_b3_labels_tx_hash_tx_call",
"b3_labels",
["transaction_hash"],
unique=True,
postgresql_where=sa.text("label='seer' and label_type='tx_call'"),
)
op.create_index(
"uk_b3_labels_tx_hash_tx_call_raw",
"b3_labels",
["transaction_hash"],
unique=True,
postgresql_where=sa.text("label='seer-raw' and label_type='tx_call'"),
)
op.create_table(
"b3_sepolia_labels",
sa.Column("id", sa.UUID(), nullable=False),
sa.Column("label", sa.VARCHAR(length=256), nullable=False),
sa.Column("transaction_hash", sa.VARCHAR(length=128), nullable=False),
sa.Column("log_index", sa.Integer(), nullable=True),
sa.Column("block_number", sa.BigInteger(), nullable=False),
sa.Column("block_hash", sa.VARCHAR(length=256), nullable=False),
sa.Column("block_timestamp", sa.BigInteger(), nullable=False),
sa.Column("caller_address", sa.LargeBinary(), nullable=True),
sa.Column("origin_address", sa.LargeBinary(), nullable=True),
sa.Column("address", sa.LargeBinary(), nullable=False),
sa.Column("label_name", sa.Text(), nullable=True),
sa.Column("label_type", sa.VARCHAR(length=64), nullable=True),
sa.Column("label_data", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
sa.Column(
"created_at",
sa.DateTime(timezone=True),
server_default=sa.text("TIMEZONE('utc', statement_timestamp())"),
nullable=False,
),
sa.PrimaryKeyConstraint("id", name=op.f("pk_b3_sepolia_labels")),
sa.UniqueConstraint("id", name=op.f("uq_b3_sepolia_labels_id")),
)
op.create_index(
"ix_b3_sepolia_labels_addr_block_num",
"b3_sepolia_labels",
["address", "block_number"],
unique=False,
)
op.create_index(
"ix_b3_sepolia_labels_addr_block_ts",
"b3_sepolia_labels",
["address", "block_timestamp"],
unique=False,
)
op.create_index(
op.f("ix_b3_sepolia_labels_address"),
"b3_sepolia_labels",
["address"],
unique=False,
)
op.create_index(
op.f("ix_b3_sepolia_labels_block_number"),
"b3_sepolia_labels",
["block_number"],
unique=False,
)
op.create_index(
op.f("ix_b3_sepolia_labels_caller_address"),
"b3_sepolia_labels",
["caller_address"],
unique=False,
)
op.create_index(
op.f("ix_b3_sepolia_labels_label"), "b3_sepolia_labels", ["label"], unique=False
)
op.create_index(
op.f("ix_b3_sepolia_labels_label_name"),
"b3_sepolia_labels",
["label_name"],
unique=False,
)
op.create_index(
op.f("ix_b3_sepolia_labels_label_type"),
"b3_sepolia_labels",
["label_type"],
unique=False,
)
op.create_index(
op.f("ix_b3_sepolia_labels_origin_address"),
"b3_sepolia_labels",
["origin_address"],
unique=False,
)
op.create_index(
op.f("ix_b3_sepolia_labels_transaction_hash"),
"b3_sepolia_labels",
["transaction_hash"],
unique=False,
)
op.create_index(
"uk_b3_sepolia_labels_tx_hash_log_idx_evt",
"b3_sepolia_labels",
["transaction_hash", "log_index"],
unique=True,
postgresql_where=sa.text("label='seer' and label_type='event'"),
)
op.create_index(
"uk_b3_sepolia_labels_tx_hash_log_idx_evt_raw",
"b3_sepolia_labels",
["transaction_hash", "log_index"],
unique=True,
postgresql_where=sa.text("label='seer-raw' and label_type='event'"),
)
op.create_index(
"uk_b3_sepolia_labels_tx_hash_tx_call",
"b3_sepolia_labels",
["transaction_hash"],
unique=True,
postgresql_where=sa.text("label='seer' and label_type='tx_call'"),
)
op.create_index(
"uk_b3_sepolia_labels_tx_hash_tx_call_raw",
"b3_sepolia_labels",
["transaction_hash"],
unique=True,
postgresql_where=sa.text("label='seer-raw' and label_type='tx_call'"),
)
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_index(
"uk_b3_sepolia_labels_tx_hash_tx_call_raw",
table_name="b3_sepolia_labels",
postgresql_where=sa.text("label='seer-raw' and label_type='tx_call'"),
)
op.drop_index(
"uk_b3_sepolia_labels_tx_hash_tx_call",
table_name="b3_sepolia_labels",
postgresql_where=sa.text("label='seer' and label_type='tx_call'"),
)
op.drop_index(
"uk_b3_sepolia_labels_tx_hash_log_idx_evt_raw",
table_name="b3_sepolia_labels",
postgresql_where=sa.text("label='seer-raw' and label_type='event'"),
)
op.drop_index(
"uk_b3_sepolia_labels_tx_hash_log_idx_evt",
table_name="b3_sepolia_labels",
postgresql_where=sa.text("label='seer' and label_type='event'"),
)
op.drop_index(
op.f("ix_b3_sepolia_labels_transaction_hash"), table_name="b3_sepolia_labels"
)
op.drop_index(
op.f("ix_b3_sepolia_labels_origin_address"), table_name="b3_sepolia_labels"
)
op.drop_index(
op.f("ix_b3_sepolia_labels_label_type"), table_name="b3_sepolia_labels"
)
op.drop_index(
op.f("ix_b3_sepolia_labels_label_name"), table_name="b3_sepolia_labels"
)
op.drop_index(op.f("ix_b3_sepolia_labels_label"), table_name="b3_sepolia_labels")
op.drop_index(
op.f("ix_b3_sepolia_labels_caller_address"), table_name="b3_sepolia_labels"
)
op.drop_index(
op.f("ix_b3_sepolia_labels_block_number"), table_name="b3_sepolia_labels"
)
op.drop_index(op.f("ix_b3_sepolia_labels_address"), table_name="b3_sepolia_labels")
op.drop_index("ix_b3_sepolia_labels_addr_block_ts", table_name="b3_sepolia_labels")
op.drop_index("ix_b3_sepolia_labels_addr_block_num", table_name="b3_sepolia_labels")
op.drop_table("b3_sepolia_labels")
op.drop_index(
"uk_b3_labels_tx_hash_tx_call_raw",
table_name="b3_labels",
postgresql_where=sa.text("label='seer-raw' and label_type='tx_call'"),
)
op.drop_index(
"uk_b3_labels_tx_hash_tx_call",
table_name="b3_labels",
postgresql_where=sa.text("label='seer' and label_type='tx_call'"),
)
op.drop_index(
"uk_b3_labels_tx_hash_log_idx_evt_raw",
table_name="b3_labels",
postgresql_where=sa.text("label='seer-raw' and label_type='event'"),
)
op.drop_index(
"uk_b3_labels_tx_hash_log_idx_evt",
table_name="b3_labels",
postgresql_where=sa.text("label='seer' and label_type='event'"),
)
op.drop_index(op.f("ix_b3_labels_transaction_hash"), table_name="b3_labels")
op.drop_index(op.f("ix_b3_labels_origin_address"), table_name="b3_labels")
op.drop_index(op.f("ix_b3_labels_label_type"), table_name="b3_labels")
op.drop_index(op.f("ix_b3_labels_label_name"), table_name="b3_labels")
op.drop_index(op.f("ix_b3_labels_label"), table_name="b3_labels")
op.drop_index(op.f("ix_b3_labels_caller_address"), table_name="b3_labels")
op.drop_index(op.f("ix_b3_labels_block_number"), table_name="b3_labels")
op.drop_index(op.f("ix_b3_labels_address"), table_name="b3_labels")
op.drop_index("ix_b3_labels_addr_block_ts", table_name="b3_labels")
op.drop_index("ix_b3_labels_addr_block_num", table_name="b3_labels")
op.drop_table("b3_labels")
# ### end Alembic commands ###