Initial release

2026-06-02 10:58:24 +10:00
commit eace48732d
88 changed files with 11140 additions and 0 deletions
@@ -0,0 +1,134 @@
+"""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')
@@ -0,0 +1,25 @@
+"""convert_status_to_varchar
+
+Revision ID: a1b2c3d4e5f6
+Revises: 86cca6d7adcb
+Create Date: 2026-06-02 12:00:00.000000
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+
+revision: str = 'a1b2c3d4e5f6'
+down_revision: Union[str, None] = '86cca6d7adcb'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.execute("ALTER TABLE lotteries ALTER COLUMN status TYPE VARCHAR(20) USING status::text")
+    op.execute("DROP TYPE IF EXISTS lotterystatus")
+
+
+def downgrade() -> None:
+    op.execute("CREATE TYPE lotterystatus AS ENUM ('draft', 'active', 'drawing', 'finished', 'cancelled')")
+    op.execute("ALTER TABLE lotteries ALTER COLUMN status TYPE lotterystatus USING status::lotterystatus")