Egor
78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
"""Add subscription_events table"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.engine.reflection import Inspector
|
|
|
|
|
|
revision: str = "c2f9c3b5f5c4"
|
|
down_revision: Union[str, None] = "9f0f2d5a1c7b"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
TABLE_NAME = "subscription_events"
|
|
|
|
|
|
def _table_exists(inspector: Inspector) -> bool:
|
|
return TABLE_NAME in inspector.get_table_names()
|
|
|
|
|
|
def upgrade() -> None:
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
|
|
if _table_exists(inspector):
|
|
return
|
|
|
|
op.create_table(
|
|
TABLE_NAME,
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("event_type", sa.String(length=50), nullable=False),
|
|
sa.Column(
|
|
"user_id",
|
|
sa.Integer(),
|
|
sa.ForeignKey("users.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"subscription_id",
|
|
sa.Integer(),
|
|
sa.ForeignKey("subscriptions.id", ondelete="SET NULL"),
|
|
nullable=True,
|
|
),
|
|
sa.Column(
|
|
"transaction_id",
|
|
sa.Integer(),
|
|
sa.ForeignKey("transactions.id", ondelete="SET NULL"),
|
|
nullable=True,
|
|
),
|
|
sa.Column("amount_kopeks", sa.Integer(), nullable=True),
|
|
sa.Column("currency", sa.String(length=16), nullable=True),
|
|
sa.Column("message", sa.Text(), nullable=True),
|
|
sa.Column(
|
|
"occurred_at", sa.DateTime(), nullable=False, server_default=sa.func.now()
|
|
),
|
|
sa.Column("extra", sa.JSON(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
|
|
)
|
|
|
|
op.create_index(
|
|
"ix_subscription_events_event_type", TABLE_NAME, ["event_type"]
|
|
)
|
|
op.create_index("ix_subscription_events_user_id", TABLE_NAME, ["user_id"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
|
|
if not _table_exists(inspector):
|
|
return
|
|
|
|
op.drop_index("ix_subscription_events_user_id", table_name=TABLE_NAME)
|
|
op.drop_index("ix_subscription_events_event_type", table_name=TABLE_NAME)
|
|
op.drop_table(TABLE_NAME)
|