homelab_automation/alembic/versions/0004_add_users_table.py

47 lines
1.8 KiB
Python

"""Add users table for authentication
Revision ID: 0004_add_users
Revises: 0003_add_notification_type
Create Date: 2025-12-09
"""
from __future__ import annotations
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision = "0004_add_users"
down_revision = "0003_add_notification_type"
branch_labels = None
depends_on = None
def upgrade() -> None:
op.create_table(
"users",
sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
sa.Column("username", sa.String(50), nullable=False, unique=True),
sa.Column("email", sa.String(255), nullable=True, unique=True),
sa.Column("hashed_password", sa.String(255), nullable=False),
sa.Column("role", sa.String(20), nullable=False, server_default=sa.text("'admin'")),
sa.Column("is_active", sa.Boolean(), nullable=False, server_default=sa.text("1")),
sa.Column("is_superuser", sa.Boolean(), nullable=False, server_default=sa.text("0")),
sa.Column("display_name", sa.String(100), nullable=True),
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
sa.Column("last_login", sa.DateTime(timezone=True), nullable=True),
sa.Column("password_changed_at", sa.DateTime(timezone=True), nullable=True),
sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
)
# Create index on username for fast lookups
op.create_index("idx_users_username", "users", ["username"])
op.create_index("idx_users_email", "users", ["email"])
def downgrade() -> None:
op.drop_index("idx_users_email", table_name="users")
op.drop_index("idx_users_username", table_name="users")
op.drop_table("users")