Olympus/backend/hermes/alembic/versions/3f4c7b8d1e2a_extend_users_enterprise_fields.py

65 lines
1.7 KiB
Python

"""extend users enterprise fields
Revision ID: 3f4c7b8d1e2a
Revises: 0a247aadcfda
Create Date: 2026-07-02 12:00:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
revision: str = "3f4c7b8d1e2a"
down_revision: Union[str, Sequence[str], None] = "0a247aadcfda"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.add_column(
"users",
sa.Column("first_name", sa.String(length=100), server_default="", nullable=False),
)
op.add_column(
"users",
sa.Column("last_name", sa.String(length=100), server_default="", nullable=False),
)
op.add_column(
"users",
sa.Column("role", sa.String(length=50), server_default="user", nullable=False),
)
op.add_column(
"users",
sa.Column("is_active", sa.Boolean(), server_default=sa.true(), nullable=False),
)
op.add_column(
"users",
sa.Column(
"created_at",
sa.DateTime(timezone=True),
server_default=sa.func.now(),
nullable=False,
),
)
op.add_column(
"users",
sa.Column(
"updated_at",
sa.DateTime(timezone=True),
server_default=sa.func.now(),
nullable=False,
),
)
op.create_index(op.f("ix_users_email"), "users", ["email"], unique=True)
def downgrade() -> None:
op.drop_index(op.f("ix_users_email"), table_name="users")
op.drop_column("users", "updated_at")
op.drop_column("users", "created_at")
op.drop_column("users", "is_active")
op.drop_column("users", "role")
op.drop_column("users", "last_name")
op.drop_column("users", "first_name")