Stardream
135 lines
6.0 KiB
Python
135 lines
6.0 KiB
Python
"""initial_schema
|
|
|
|
Revision ID: 86cca6d7adcb
|
|
Revises:
|
|
Create Date: 2026-05-27 16:31:52.870967
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision: str = '86cca6d7adcb'
|
|
down_revision: Union[str, None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
'users',
|
|
sa.Column('id', sa.BigInteger(), nullable=False),
|
|
sa.Column('username', sa.String(255), nullable=True),
|
|
sa.Column('first_name', sa.String(255), nullable=False),
|
|
sa.Column('last_name', sa.String(255), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
|
|
op.create_table(
|
|
'lotteries',
|
|
sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('creator_id', sa.BigInteger(), nullable=False),
|
|
sa.Column('title', sa.String(255), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('status', sa.String(20), nullable=False),
|
|
sa.Column('target_chat_id', sa.BigInteger(), nullable=True),
|
|
sa.Column('target_chat_title', sa.String(255), nullable=True),
|
|
sa.Column('require_membership', sa.Boolean(), nullable=False),
|
|
sa.Column('required_chat_ids', sa.JSON(), nullable=True),
|
|
sa.Column('required_chat_titles', sa.JSON(), nullable=True),
|
|
sa.Column('required_chat_types', sa.JSON(), nullable=True),
|
|
sa.Column('required_chat_usernames', sa.JSON(), nullable=True),
|
|
sa.Column('announcement_targets', sa.JSON(), nullable=True),
|
|
sa.Column('max_participants', sa.Integer(), nullable=True),
|
|
sa.Column('announcement_message_id', sa.BigInteger(), nullable=True),
|
|
sa.Column('contact_info', sa.Text(), nullable=True),
|
|
sa.Column('draw_at', sa.DateTime(), nullable=True),
|
|
sa.Column('auto_draw_count', sa.Integer(), nullable=True),
|
|
sa.Column('allow_multiple_wins', sa.Boolean(), server_default=sa.text('false'), nullable=False),
|
|
sa.Column('invite_link_chat_ids', sa.JSON(), server_default=sa.text("'[]'"), nullable=False),
|
|
sa.Column('creator_timezone', sa.String(64), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.Column('drawn_at', sa.DateTime(), nullable=True),
|
|
sa.Column('draw_trigger', sa.String(20), nullable=True),
|
|
sa.Column('required_chat_passphrases', sa.JSON(), nullable=True),
|
|
sa.ForeignKeyConstraint(['creator_id'], ['users.id']),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
|
|
op.create_table(
|
|
'prizes',
|
|
sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('lottery_id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('name', sa.String(255), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('quantity', sa.Integer(), nullable=False),
|
|
sa.Column('sort_order', sa.Integer(), nullable=False),
|
|
sa.ForeignKeyConstraint(['lottery_id'], ['lotteries.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
|
|
op.create_table(
|
|
'participants',
|
|
sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('lottery_id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('user_id', sa.BigInteger(), nullable=False),
|
|
sa.Column('joined_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['lottery_id'], ['lotteries.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id']),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('lottery_id', 'user_id'),
|
|
)
|
|
|
|
op.create_table(
|
|
'winners',
|
|
sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('lottery_id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('user_id', sa.BigInteger(), nullable=False),
|
|
sa.Column('prize_id', postgresql.UUID(as_uuid=True), nullable=True),
|
|
sa.Column('prize_name', sa.String(255), nullable=False),
|
|
sa.Column('drawn_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['lottery_id'], ['lotteries.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['prize_id'], ['prizes.id'], ondelete='SET NULL'),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id']),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
|
|
op.create_table(
|
|
'managed_chats',
|
|
sa.Column('id', sa.BigInteger(), nullable=False),
|
|
sa.Column('title', sa.String(255), nullable=False),
|
|
sa.Column('username', sa.String(255), nullable=True),
|
|
sa.Column('chat_type', sa.String(50), nullable=False),
|
|
sa.Column('invite_link', sa.String(512), nullable=True),
|
|
sa.Column('last_message_id', sa.BigInteger(), nullable=True),
|
|
sa.Column('is_active', sa.Boolean(), nullable=False),
|
|
sa.Column('added_at', sa.DateTime(), nullable=False),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
|
|
op.create_table(
|
|
'passphrase_verifications',
|
|
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column('lottery_id', postgresql.UUID(as_uuid=True), nullable=False),
|
|
sa.Column('chat_id', sa.BigInteger(), nullable=False),
|
|
sa.Column('user_id', sa.BigInteger(), nullable=False),
|
|
sa.Column('verified_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['lottery_id'], ['lotteries.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id']),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('lottery_id', 'chat_id', 'user_id'),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table('passphrase_verifications')
|
|
op.drop_table('managed_chats')
|
|
op.drop_table('winners')
|
|
op.drop_table('participants')
|
|
op.drop_table('prizes')
|
|
op.drop_table('lotteries')
|
|
op.drop_table('users')
|