99 lines
4.5 KiB
Python
99 lines
4.5 KiB
Python
"""init schema
|
|
|
|
Revision ID: 0001_init
|
|
Revises:
|
|
Create Date: 2026-02-12
|
|
"""
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision = "0001_init"
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"users",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("email", sa.String(length=255), nullable=False, unique=True),
|
|
sa.Column("password_hash", sa.String(length=255), nullable=False),
|
|
sa.Column("role", sa.String(length=20), nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
)
|
|
op.create_index("ix_users_email", "users", ["email"], unique=True)
|
|
|
|
op.create_table(
|
|
"targets",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("name", sa.String(length=120), nullable=False, unique=True),
|
|
sa.Column("host", sa.String(length=255), nullable=False),
|
|
sa.Column("port", sa.Integer(), nullable=False, server_default="5432"),
|
|
sa.Column("dbname", sa.String(length=120), nullable=False),
|
|
sa.Column("username", sa.String(length=120), nullable=False),
|
|
sa.Column("encrypted_password", sa.Text(), nullable=False),
|
|
sa.Column("sslmode", sa.String(length=20), nullable=False, server_default="prefer"),
|
|
sa.Column("tags", sa.JSON(), nullable=False, server_default="{}"),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
)
|
|
op.create_index("ix_targets_name", "targets", ["name"], unique=True)
|
|
|
|
op.create_table(
|
|
"metrics",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("target_id", sa.Integer(), sa.ForeignKey("targets.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("ts", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("metric_name", sa.String(length=120), nullable=False),
|
|
sa.Column("value", sa.Float(), nullable=False),
|
|
sa.Column("labels", sa.JSON(), nullable=False, server_default="{}"),
|
|
)
|
|
op.create_index("ix_metrics_target_id", "metrics", ["target_id"])
|
|
op.create_index("ix_metrics_ts", "metrics", ["ts"])
|
|
op.create_index("ix_metrics_metric_name", "metrics", ["metric_name"])
|
|
|
|
op.create_table(
|
|
"query_stats",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("target_id", sa.Integer(), sa.ForeignKey("targets.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("ts", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("queryid", sa.String(length=100), nullable=False),
|
|
sa.Column("calls", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("total_time", sa.Float(), nullable=False, server_default="0"),
|
|
sa.Column("mean_time", sa.Float(), nullable=False, server_default="0"),
|
|
sa.Column("rows", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("query_text", sa.Text(), nullable=True),
|
|
)
|
|
op.create_index("ix_query_stats_target_id", "query_stats", ["target_id"])
|
|
op.create_index("ix_query_stats_ts", "query_stats", ["ts"])
|
|
|
|
op.create_table(
|
|
"audit_logs",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("ts", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id"), nullable=True),
|
|
sa.Column("action", sa.String(length=120), nullable=False),
|
|
sa.Column("payload", sa.JSON(), nullable=False, server_default="{}"),
|
|
)
|
|
op.create_index("ix_audit_logs_ts", "audit_logs", ["ts"])
|
|
op.create_index("ix_audit_logs_user_id", "audit_logs", ["user_id"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_audit_logs_user_id", table_name="audit_logs")
|
|
op.drop_index("ix_audit_logs_ts", table_name="audit_logs")
|
|
op.drop_table("audit_logs")
|
|
op.drop_index("ix_query_stats_ts", table_name="query_stats")
|
|
op.drop_index("ix_query_stats_target_id", table_name="query_stats")
|
|
op.drop_table("query_stats")
|
|
op.drop_index("ix_metrics_metric_name", table_name="metrics")
|
|
op.drop_index("ix_metrics_ts", table_name="metrics")
|
|
op.drop_index("ix_metrics_target_id", table_name="metrics")
|
|
op.drop_table("metrics")
|
|
op.drop_index("ix_targets_name", table_name="targets")
|
|
op.drop_table("targets")
|
|
op.drop_index("ix_users_email", table_name="users")
|
|
op.drop_table("users")
|