Cheng Zhou
51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
"""create users table with review status"""
|
|
from __future__ import annotations
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "20240501_000001"
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
user_role_enum = sa.Enum("ADMIN", "PM", "CRA", "PV", "IMP", name="user_role")
|
|
user_status_enum = sa.Enum("PENDING", "ACTIVE", "REJECTED", "DISABLED", name="user_status")
|
|
|
|
|
|
def upgrade() -> None:
|
|
bind = op.get_bind()
|
|
user_role_enum.create(bind, checkfirst=True)
|
|
user_status_enum.create(bind, checkfirst=True)
|
|
op.create_table(
|
|
"users",
|
|
sa.Column("id", postgresql.UUID(as_uuid=True), primary_key=True, nullable=False),
|
|
sa.Column("email", sa.String(length=255), nullable=False),
|
|
sa.Column("password_hash", sa.String(length=255), nullable=False),
|
|
sa.Column("full_name", sa.String(length=255), nullable=False),
|
|
sa.Column("role", user_role_enum, nullable=False),
|
|
sa.Column("department", sa.String(length=255), nullable=False),
|
|
sa.Column("status", user_status_enum, nullable=False, server_default="PENDING"),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
nullable=False,
|
|
server_default=sa.text("now()"),
|
|
),
|
|
sa.Column("approved_by", postgresql.UUID(as_uuid=True), sa.ForeignKey("users.id"), nullable=True),
|
|
sa.Column("approved_at", sa.DateTime(timezone=True), nullable=True),
|
|
)
|
|
op.create_index("ix_users_email", "users", ["email"], unique=True)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_users_email", table_name="users")
|
|
op.drop_table("users")
|
|
bind = op.get_bind()
|
|
user_role_enum.drop(bind, checkfirst=True)
|
|
user_status_enum.drop(bind, checkfirst=True)
|