unchained/mission-control
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add conditional creation of skills marketplace and gateway installed skills tables with index checks

Browse Source
This commit is contained in:
Abhimanyu Saharan
2026-02-13 23:39:32 +05:30
committed by Abhimanyu Saharan
parent e7b5df0bce
commit 88565f4d69
1 changed files with 106 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

177
backend/migrations/versions/c9d7e9b6a4f2_add_skills_marketplace_tables.py
View File

@@ -19,81 +19,116 @@ branch_labels = None
depends_on = None depends_on = None
def _has_table(table_name: str) -> bool:
return sa.inspect(op.get_bind()).has_table(table_name)
def _has_index(table_name: str, index_name: str) -> bool:
if not _has_table(table_name):
return False
indexes = sa.inspect(op.get_bind()).get_indexes(table_name)
return any(index["name"] == index_name for index in indexes)
def upgrade() -> None: def upgrade() -> None:
op.create_table( if not _has_table("marketplace_skills"):
"marketplace_skills", op.create_table(
sa.Column("id", sa.Uuid(), nullable=False), "marketplace_skills",
sa.Column("organization_id", sa.Uuid(), nullable=False), sa.Column("id", sa.Uuid(), nullable=False),
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False), sa.Column("organization_id", sa.Uuid(), nullable=False),
sa.Column("description", sqlmodel.sql.sqltypes.AutoString(), nullable=True), sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column("source_url", sqlmodel.sql.sqltypes.AutoString(), nullable=False), sa.Column("description", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
sa.Column("created_at", sa.DateTime(), nullable=False), sa.Column("source_url", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False), sa.Column("created_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint( sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(
["organization_id"],
["organizations.id"],
),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint(
"organization_id",
"source_url",
name="uq_marketplace_skills_org_source_url",
),
)
marketplace_org_idx = op.f("ix_marketplace_skills_organization_id")
if not _has_index("marketplace_skills", marketplace_org_idx):
op.create_index(
marketplace_org_idx,
"marketplace_skills",
["organization_id"], ["organization_id"],
["organizations.id"], unique=False,
), )
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint( if not _has_table("gateway_installed_skills"):
"organization_id", op.create_table(
"source_url", "gateway_installed_skills",
name="uq_marketplace_skills_org_source_url", sa.Column("id", sa.Uuid(), nullable=False),
), sa.Column("gateway_id", sa.Uuid(), nullable=False),
) sa.Column("skill_id", sa.Uuid(), nullable=False),
op.create_index( sa.Column("created_at", sa.DateTime(), nullable=False),
op.f("ix_marketplace_skills_organization_id"), sa.Column("updated_at", sa.DateTime(), nullable=False),
"marketplace_skills", sa.ForeignKeyConstraint(
["organization_id"], ["gateway_id"],
unique=False, ["gateways.id"],
) ),
op.create_table( sa.ForeignKeyConstraint(
"gateway_installed_skills", ["skill_id"],
sa.Column("id", sa.Uuid(), nullable=False), ["marketplace_skills.id"],
sa.Column("gateway_id", sa.Uuid(), nullable=False), ),
sa.Column("skill_id", sa.Uuid(), nullable=False), sa.PrimaryKeyConstraint("id"),
sa.Column("created_at", sa.DateTime(), nullable=False), sa.UniqueConstraint(
sa.Column("updated_at", sa.DateTime(), nullable=False), "gateway_id",
sa.ForeignKeyConstraint( "skill_id",
name="uq_gateway_installed_skills_gateway_id_skill_id",
),
)
gateway_id_idx = op.f("ix_gateway_installed_skills_gateway_id")
if not _has_index("gateway_installed_skills", gateway_id_idx):
op.create_index(
gateway_id_idx,
"gateway_installed_skills",
["gateway_id"], ["gateway_id"],
["gateways.id"], unique=False,
), )
sa.ForeignKeyConstraint(
gateway_skill_idx = op.f("ix_gateway_installed_skills_skill_id")
if not _has_index("gateway_installed_skills", gateway_skill_idx):
op.create_index(
gateway_skill_idx,
"gateway_installed_skills",
["skill_id"], ["skill_id"],
["marketplace_skills.id"], unique=False,
), )
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint(
"gateway_id",
"skill_id",
name="uq_gateway_installed_skills_gateway_id_skill_id",
),
)
op.create_index(
op.f("ix_gateway_installed_skills_gateway_id"),
"gateway_installed_skills",
["gateway_id"],
unique=False,
)
op.create_index(
op.f("ix_gateway_installed_skills_skill_id"),
"gateway_installed_skills",
["skill_id"],
unique=False,
)
def downgrade() -> None: def downgrade() -> None:
op.drop_index( gateway_skill_idx = op.f("ix_gateway_installed_skills_skill_id")
op.f("ix_gateway_installed_skills_skill_id"), if _has_index("gateway_installed_skills", gateway_skill_idx):
table_name="gateway_installed_skills", op.drop_index(
) gateway_skill_idx,
op.drop_index( table_name="gateway_installed_skills",
op.f("ix_gateway_installed_skills_gateway_id"), )
table_name="gateway_installed_skills",
) gateway_id_idx = op.f("ix_gateway_installed_skills_gateway_id")
op.drop_table("gateway_installed_skills") if _has_index("gateway_installed_skills", gateway_id_idx):
op.drop_index( op.drop_index(
op.f("ix_marketplace_skills_organization_id"), gateway_id_idx,
table_name="marketplace_skills", table_name="gateway_installed_skills",
) )
op.drop_table("marketplace_skills")
if _has_table("gateway_installed_skills"):
op.drop_table("gateway_installed_skills")
marketplace_org_idx = op.f("ix_marketplace_skills_organization_id")
if _has_index("marketplace_skills", marketplace_org_idx):
op.drop_index(
marketplace_org_idx,
table_name="marketplace_skills",
)
if _has_table("marketplace_skills"):
op.drop_table("marketplace_skills")