a06ea946b1
- "backend/models.py" - "backend/auth.py" - "backend/schemas.py" - "backend/requirements.txt" - "alembic/versions/016_add_users_and_invite_codes.py" GSD-Task: S02/T01
50 lines
2.2 KiB
Python
50 lines
2.2 KiB
Python
"""Add users and invite_codes tables for creator authentication.
|
|
|
|
Revision ID: 016_add_users_and_invite_codes
|
|
Revises: 015_add_creator_profile
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
|
|
revision = "016_add_users_and_invite_codes"
|
|
down_revision = "015_add_creator_profile"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Create user_role enum type
|
|
user_role_enum = sa.Enum("creator", "admin", name="user_role", create_constraint=True)
|
|
user_role_enum.create(op.get_bind(), checkfirst=True)
|
|
|
|
# Create users table
|
|
op.create_table(
|
|
"users",
|
|
sa.Column("id", UUID(as_uuid=True), primary_key=True, server_default=sa.func.gen_random_uuid()),
|
|
sa.Column("email", sa.String(255), nullable=False, unique=True),
|
|
sa.Column("hashed_password", sa.String(255), nullable=False),
|
|
sa.Column("display_name", sa.String(255), nullable=False),
|
|
sa.Column("role", user_role_enum, nullable=False, server_default="creator"),
|
|
sa.Column("creator_id", UUID(as_uuid=True), sa.ForeignKey("creators.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("is_active", sa.Boolean(), nullable=False, server_default="true"),
|
|
sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
|
|
sa.Column("updated_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
|
|
)
|
|
|
|
# Create invite_codes table
|
|
op.create_table(
|
|
"invite_codes",
|
|
sa.Column("id", UUID(as_uuid=True), primary_key=True, server_default=sa.func.gen_random_uuid()),
|
|
sa.Column("code", sa.String(100), nullable=False, unique=True),
|
|
sa.Column("uses_remaining", sa.Integer(), nullable=False, server_default="1"),
|
|
sa.Column("created_by", UUID(as_uuid=True), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("expires_at", sa.DateTime(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("invite_codes")
|
|
op.drop_table("users")
|
|
sa.Enum(name="user_role").drop(op.get_bind(), checkfirst=True)
|