72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
"""create user token tables
|
|
|
|
Revision ID: 001
|
|
Revises:
|
|
Create Date: 2023-10-10
|
|
|
|
"""
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.sql import func
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "001"
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Create users table
|
|
op.create_table(
|
|
"users",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("email", sa.String(), nullable=False),
|
|
sa.Column("username", sa.String(), nullable=False),
|
|
sa.Column("hashed_password", sa.String(), nullable=False),
|
|
sa.Column("full_name", sa.String(), nullable=True),
|
|
sa.Column("is_active", sa.Boolean(), default=True),
|
|
sa.Column("is_superuser", sa.Boolean(), default=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=func.now()),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=func.now(),
|
|
onupdate=func.now(),
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
|
|
op.create_index(op.f("ix_users_id"), "users", ["id"], unique=False)
|
|
op.create_index(op.f("ix_users_email"), "users", ["email"], unique=True)
|
|
op.create_index(op.f("ix_users_username"), "users", ["username"], unique=True)
|
|
|
|
# Create tokens table
|
|
op.create_table(
|
|
"tokens",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("user_id", sa.Integer(), nullable=False),
|
|
sa.Column("token", sa.String(), nullable=False),
|
|
sa.Column("expires_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("is_revoked", sa.Boolean(), default=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=func.now()),
|
|
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
|
|
op.create_index(op.f("ix_tokens_id"), "tokens", ["id"], unique=False)
|
|
op.create_index(op.f("ix_tokens_token"), "tokens", ["token"], unique=True)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index(op.f("ix_tokens_token"), table_name="tokens")
|
|
op.drop_index(op.f("ix_tokens_id"), table_name="tokens")
|
|
op.drop_table("tokens")
|
|
|
|
op.drop_index(op.f("ix_users_username"), table_name="users")
|
|
op.drop_index(op.f("ix_users_email"), table_name="users")
|
|
op.drop_index(op.f("ix_users_id"), table_name="users")
|
|
op.drop_table("users")
|