Files
headquarter/apps/api/alembic/versions/0001_initial_schema.py
T

107 lines
4.5 KiB
Python

"""initial schema
Revision ID: 0001_initial_schema
Revises:
Create Date: 2026-05-17 00:00:00.000000
"""
import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql
revision = "0001_initial_schema"
down_revision = None
branch_labels = None
depends_on = None
TABLE_NAMES = [
"users",
"ssh_keys",
"projects",
"git_repositories",
"user_configs",
]
def upgrade() -> None:
op.create_table(
"users",
sa.Column("email", sa.String(length=255), nullable=False),
sa.Column("name", sa.String(length=255), nullable=False),
sa.Column("authentik_id", sa.String(length=255), nullable=False),
sa.Column("avatar_url", sa.String(length=1024), nullable=True),
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
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.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("authentik_id"),
sa.UniqueConstraint("email"),
)
op.create_index(op.f("ix_users_authentik_id"), "users", ["authentik_id"], unique=True)
op.create_index(op.f("ix_users_email"), "users", ["email"], unique=True)
op.create_table(
"ssh_keys",
sa.Column("name", sa.String(length=255), nullable=False),
sa.Column("public_key", sa.Text(), nullable=False),
sa.Column("private_key_encrypted", sa.Text(), nullable=False),
sa.Column("user_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("project_id", postgresql.UUID(as_uuid=True), nullable=True),
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
sa.PrimaryKeyConstraint("id"),
sa.ForeignKeyConstraint(["user_id"], ["users.id"]),
)
op.create_table(
"projects",
sa.Column("name", sa.String(length=255), nullable=False),
sa.Column("description", sa.Text(), nullable=True),
sa.Column("owner_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("default_ssh_key_id", postgresql.UUID(as_uuid=True), nullable=True),
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
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.PrimaryKeyConstraint("id"),
sa.ForeignKeyConstraint(["default_ssh_key_id"], ["ssh_keys.id"]),
sa.ForeignKeyConstraint(["owner_id"], ["users.id"]),
)
op.create_table(
"git_repositories",
sa.Column("name", sa.String(length=255), nullable=False),
sa.Column("path", sa.String(length=1024), nullable=False),
sa.Column("project_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("owner_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("is_mirror", sa.Boolean(), nullable=False),
sa.Column("remote_url", sa.String(length=1024), nullable=True),
sa.Column("last_push", sa.DateTime(timezone=True), nullable=True),
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
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.PrimaryKeyConstraint("id"),
sa.ForeignKeyConstraint(["owner_id"], ["users.id"]),
sa.ForeignKeyConstraint(["project_id"], ["projects.id"]),
)
op.create_table(
"user_configs",
sa.Column("user_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.Column("config", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
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.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("user_id"),
sa.ForeignKeyConstraint(["user_id"], ["users.id"]),
)
def downgrade() -> None:
op.drop_table("user_configs")
op.drop_table("git_repositories")
op.drop_table("projects")
op.drop_table("ssh_keys")
op.drop_index(op.f("ix_users_email"), table_name="users")
op.drop_index(op.f("ix_users_authentik_id"), table_name="users")
op.drop_table("users")