q792602257/agentic-mobile-control
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(cloud-store): verify schema migrations

Browse Source
This commit is contained in:
q792602257
2026-07-12 16:59:08 +08:00
parent 24d9dbf38b
commit 7f5f740e2a
5 changed files with 224 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/cloud-platform/cloud/migrations/versions/0001_cloud_repository.py
+24 -13
View File
@@ -12,13 +12,13 @@ branch_labels = None
depends_on = None
LEASE_COLUMNS = (
sa.Column("attempt_count", sa.Integer(), nullable=False, server_default="0"),
sa.Column("lease_id", sa.String(), nullable=True),
sa.Column("lease_expires_at", sa.String(), nullable=True),
sa.Column("failure_reason", sa.Text(), nullable=True),
sa.Column("result_json", sa.Text(), nullable=True),
sa.Column("updated_at", sa.String(), nullable=True),
LEASE_COLUMN_NAMES = (
"attempt_count",
"lease_id",
"lease_expires_at",
"failure_reason",
"result_json",
"updated_at",
)
@@ -32,9 +32,9 @@ def upgrade() -> None:
column["name"] for column in inspector.get_columns("scheduled_tasks")
}
with op.batch_alter_table("scheduled_tasks") as batch:
for column in LEASE_COLUMNS:
for column in _lease_columns():
if column.name not in existing_columns:
batch.add_column(column.copy())
batch.add_column(column)
inspector = sa.inspect(op.get_bind())
if "task_attempts" not in inspector.get_table_names():
@@ -52,9 +52,9 @@ def downgrade() -> None:
column["name"] for column in inspector.get_columns("scheduled_tasks")
}
with op.batch_alter_table("scheduled_tasks") as batch:
for column in reversed(LEASE_COLUMNS):
if column.name in existing_columns:
batch.drop_column(column.name)
for column_name in reversed(LEASE_COLUMN_NAMES):
if column_name in existing_columns:
batch.drop_column(column_name)
def _create_base_tables() -> None:
@@ -82,7 +82,7 @@ def _create_base_tables() -> None:
sa.Column("status", sa.String(), nullable=False),
sa.Column("assigned_device_id", sa.String(), nullable=True),
sa.Column("assigned_host_id", sa.String(), nullable=True),
*[column.copy() for column in LEASE_COLUMNS],
*_lease_columns(),
sa.Column("created_at", sa.String(), nullable=False),
)
op.create_table(
@@ -110,3 +110,14 @@ def _create_attempts_table() -> None:
sa.Column("failure_reason", sa.Text(), nullable=True),
sa.Column("result_json", sa.Text(), nullable=True),
)
def _lease_columns() -> tuple[sa.Column, ...]:
return (
sa.Column("attempt_count", sa.Integer(), nullable=False, server_default="0"),
sa.Column("lease_id", sa.String(), nullable=True),
sa.Column("lease_expires_at", sa.String(), nullable=True),
sa.Column("failure_reason", sa.Text(), nullable=True),
sa.Column("result_json", sa.Text(), nullable=True),
sa.Column("updated_at", sa.String(), nullable=True),
)