dograh/api/alembic/versions/37d0a90fccba_add_api_keys_model.py
Abhishek Kumar 4f2a629340 Initial Commit 🚀 🚀
2025-09-09 14:37:32 +05:30

60 lines
2.2 KiB
Python

"""Add API Keys model
Revision ID: 37d0a90fccba
Revises: c425d3445750
Create Date: 2025-09-06 14:19:35.300761
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "37d0a90fccba"
down_revision: Union[str, None] = "c425d3445750"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.create_table(
"api_keys",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("organization_id", sa.Integer(), nullable=False),
sa.Column("name", sa.String(), nullable=False),
sa.Column("key_hash", sa.String(), nullable=False),
sa.Column("key_prefix", sa.String(), nullable=False),
sa.Column("is_active", sa.Boolean(), nullable=False),
sa.Column("created_by", sa.Integer(), nullable=True),
sa.Column("last_used_at", sa.DateTime(timezone=True), nullable=True),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=True),
sa.Column("archived_at", sa.DateTime(timezone=True), nullable=True),
sa.ForeignKeyConstraint(
["created_by"],
["users.id"],
),
sa.ForeignKeyConstraint(
["organization_id"], ["organizations.id"], ondelete="CASCADE"
),
sa.PrimaryKeyConstraint("id"),
)
op.create_index("ix_api_keys_active", "api_keys", ["is_active"], unique=False)
op.create_index(op.f("ix_api_keys_id"), "api_keys", ["id"], unique=False)
op.create_index("ix_api_keys_key_hash", "api_keys", ["key_hash"], unique=True)
op.create_index(
"ix_api_keys_organization_id", "api_keys", ["organization_id"], unique=False
)
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_index("ix_api_keys_organization_id", table_name="api_keys")
op.drop_index("ix_api_keys_key_hash", table_name="api_keys")
op.drop_index(op.f("ix_api_keys_id"), table_name="api_keys")
op.drop_index("ix_api_keys_active", table_name="api_keys")
op.drop_table("api_keys")
# ### end Alembic commands ###