57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
"""Add durable Cloud governance policies."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision = "0004_cloud_governance"
|
|
down_revision = "0003_cloud_user_authentication"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
inspector = sa.inspect(op.get_bind())
|
|
tables = set(inspector.get_table_names())
|
|
if "cloud_user_submission_policies" not in tables:
|
|
op.create_table(
|
|
"cloud_user_submission_policies",
|
|
sa.Column(
|
|
"user_id",
|
|
sa.String(),
|
|
sa.ForeignKey("cloud_users.id", ondelete="CASCADE"),
|
|
primary_key=True,
|
|
),
|
|
sa.Column("revision", sa.Integer(), nullable=False),
|
|
sa.Column("submission_enabled", sa.Integer(), nullable=False),
|
|
sa.Column("allowed_host_ids_json", sa.Text(), nullable=True),
|
|
sa.Column("allowed_device_targets_json", sa.Text(), nullable=True),
|
|
sa.Column("updated_at", sa.String(), nullable=False),
|
|
)
|
|
if "cloud_host_governance_policies" not in tables:
|
|
op.create_table(
|
|
"cloud_host_governance_policies",
|
|
sa.Column(
|
|
"host_id",
|
|
sa.String(),
|
|
sa.ForeignKey("host_registrations.host_id", ondelete="CASCADE"),
|
|
primary_key=True,
|
|
),
|
|
sa.Column("revision", sa.Integer(), nullable=False),
|
|
sa.Column("self_submission_enabled", sa.Integer(), nullable=False),
|
|
sa.Column("max_active_tasks", sa.Integer(), nullable=True),
|
|
sa.Column("daily_token_budget", sa.Integer(), nullable=True),
|
|
sa.Column("updated_at", sa.String(), nullable=False),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
inspector = sa.inspect(op.get_bind())
|
|
tables = set(inspector.get_table_names())
|
|
if "cloud_host_governance_policies" in tables:
|
|
op.drop_table("cloud_host_governance_policies")
|
|
if "cloud_user_submission_policies" in tables:
|
|
op.drop_table("cloud_user_submission_policies")
|