76 lines
3.3 KiB
Python
76 lines
3.3 KiB
Python
"""add_uploads_and_achievements
|
|
|
|
Revision ID: 0003_uploads_achievements
|
|
Revises: 0002_groups_oauth
|
|
Create Date: 2026-02-03 21:00:00
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "0003_uploads_achievements"
|
|
down_revision = "0002_groups_oauth"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Create uploads table
|
|
op.create_table(
|
|
"uploads",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("filename", sa.String(length=255), nullable=False),
|
|
sa.Column("content_type", sa.String(length=100), nullable=False),
|
|
sa.Column("file_path", sa.String(length=512), nullable=False),
|
|
sa.Column("size_bytes", sa.Integer(), nullable=False),
|
|
sa.Column("uploaded_by", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index(op.f("ix_uploads_uploaded_by"), "uploads", ["uploaded_by"])
|
|
|
|
# Add upload_id to modules table
|
|
op.add_column("modules", sa.Column("upload_id", sa.Integer(), sa.ForeignKey("uploads.id", ondelete="SET NULL"), nullable=True))
|
|
op.create_index(op.f("ix_modules_upload_id"), "modules", ["upload_id"])
|
|
|
|
# Create achievements table
|
|
op.create_table(
|
|
"achievements",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("code", sa.String(length=100), nullable=False, unique=True),
|
|
sa.Column("name", sa.String(length=255), nullable=False),
|
|
sa.Column("description", sa.Text(), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index(op.f("ix_achievements_code"), "achievements", ["code"], unique=True)
|
|
|
|
# Create user_achievements table
|
|
op.create_table(
|
|
"user_achievements",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("achievement_id", sa.Integer(), sa.ForeignKey("achievements.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("earned_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.UniqueConstraint("user_id", "achievement_id", name="uq_user_achievement"),
|
|
)
|
|
op.create_index(op.f("ix_user_achievements_user_id"), "user_achievements", ["user_id"])
|
|
op.create_index(op.f("ix_user_achievements_achievement_id"), "user_achievements", ["achievement_id"])
|
|
|
|
# Create login_events table for tracking logins
|
|
op.create_table(
|
|
"login_events",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("logged_in_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index(op.f("ix_login_events_user_id"), "login_events", ["user_id"])
|
|
op.create_index(op.f("ix_login_events_logged_in_at"), "login_events", ["logged_in_at"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("login_events")
|
|
op.drop_table("user_achievements")
|
|
op.drop_table("achievements")
|
|
op.drop_index(op.f("ix_modules_upload_id"), "modules")
|
|
op.drop_column("modules", "upload_id")
|
|
op.drop_table("uploads")
|