feat: add language column, AI Generated category, and mine/shared filters
- Add language column (powershell/bash/python) to script_templates model and schemas - Seed 'AI Generated' script category via migration 063 - Add mine and shared query params to list_templates endpoint Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,55 @@
|
|||||||
|
"""add_language_to_script_templates_and_ai_generated_category
|
||||||
|
|
||||||
|
Revision ID: 063
|
||||||
|
Revises: 062
|
||||||
|
Create Date: 2026-03-21 21:13:32.239533
|
||||||
|
|
||||||
|
"""
|
||||||
|
from typing import Sequence, Union
|
||||||
|
|
||||||
|
from alembic import op
|
||||||
|
import sqlalchemy as sa
|
||||||
|
|
||||||
|
|
||||||
|
# revision identifiers, used by Alembic.
|
||||||
|
revision: str = '063'
|
||||||
|
down_revision: Union[str, None] = '062'
|
||||||
|
branch_labels: Union[str, Sequence[str], None] = None
|
||||||
|
depends_on: Union[str, Sequence[str], None] = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade() -> None:
|
||||||
|
# Add language column to script_templates
|
||||||
|
op.add_column(
|
||||||
|
'script_templates',
|
||||||
|
sa.Column(
|
||||||
|
'language',
|
||||||
|
sa.String(length=30),
|
||||||
|
nullable=True,
|
||||||
|
comment='Script language: powershell, bash, python',
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
# Seed "AI Generated" category
|
||||||
|
op.execute(
|
||||||
|
sa.text("""
|
||||||
|
INSERT INTO script_categories (id, name, slug, description, icon, sort_order, is_active, created_at, updated_at)
|
||||||
|
VALUES (
|
||||||
|
'a0000000-0000-0000-0000-000000000001'::uuid,
|
||||||
|
'AI Generated',
|
||||||
|
'ai-generated',
|
||||||
|
'Scripts generated by the AI Script Builder',
|
||||||
|
'sparkles',
|
||||||
|
100,
|
||||||
|
true,
|
||||||
|
NOW(),
|
||||||
|
NOW()
|
||||||
|
)
|
||||||
|
ON CONFLICT (slug) DO NOTHING
|
||||||
|
""")
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade() -> None:
|
||||||
|
op.execute(sa.text("DELETE FROM script_categories WHERE slug = 'ai-generated'"))
|
||||||
|
op.drop_column('script_templates', 'language')
|
||||||
@@ -76,6 +76,8 @@ async def list_templates(
|
|||||||
search: Optional[str] = Query(None),
|
search: Optional[str] = Query(None),
|
||||||
tags: Optional[str] = Query(None, description="Comma-separated tags"),
|
tags: Optional[str] = Query(None, description="Comma-separated tags"),
|
||||||
managed: Optional[bool] = Query(None, description="If true, return only templates this user can edit"),
|
managed: Optional[bool] = Query(None, description="If true, return only templates this user can edit"),
|
||||||
|
mine: bool = Query(False, description="If true, return only templates created by the current user"),
|
||||||
|
shared: bool = Query(False, description="If true, return only templates shared with the user's team"),
|
||||||
) -> list[ScriptTemplateListItem]:
|
) -> list[ScriptTemplateListItem]:
|
||||||
query = (
|
query = (
|
||||||
select(ScriptTemplate)
|
select(ScriptTemplate)
|
||||||
@@ -116,6 +118,12 @@ async def list_templates(
|
|||||||
# engineers see only their own
|
# engineers see only their own
|
||||||
query = query.where(ScriptTemplate.created_by == current_user.id)
|
query = query.where(ScriptTemplate.created_by == current_user.id)
|
||||||
|
|
||||||
|
if mine:
|
||||||
|
query = query.where(ScriptTemplate.created_by == current_user.id)
|
||||||
|
|
||||||
|
if shared:
|
||||||
|
query = query.where(ScriptTemplate.team_id == current_user.team_id)
|
||||||
|
|
||||||
result = await db.execute(query.order_by(ScriptTemplate.name))
|
result = await db.execute(query.order_by(ScriptTemplate.name))
|
||||||
templates = result.scalars().all()
|
templates = result.scalars().all()
|
||||||
|
|
||||||
|
|||||||
@@ -52,6 +52,10 @@ class ScriptTemplate(Base):
|
|||||||
description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||||
use_case: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
use_case: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||||
script_body: Mapped[str] = mapped_column(Text, nullable=False)
|
script_body: Mapped[str] = mapped_column(Text, nullable=False)
|
||||||
|
language: Mapped[Optional[str]] = mapped_column(
|
||||||
|
String(30), nullable=True, default="powershell",
|
||||||
|
comment="Script language: powershell, bash, python",
|
||||||
|
)
|
||||||
parameters_schema: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)
|
parameters_schema: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)
|
||||||
default_values: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)
|
default_values: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)
|
||||||
validation_rules: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)
|
validation_rules: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)
|
||||||
|
|||||||
@@ -62,6 +62,7 @@ class ScriptTemplateCreate(BaseModel):
|
|||||||
estimated_runtime: Optional[str] = None
|
estimated_runtime: Optional[str] = None
|
||||||
requires_elevation: bool = False
|
requires_elevation: bool = False
|
||||||
requires_modules: list[str] = Field(default_factory=list)
|
requires_modules: list[str] = Field(default_factory=list)
|
||||||
|
language: str | None = None
|
||||||
|
|
||||||
class ScriptTemplateUpdate(BaseModel):
|
class ScriptTemplateUpdate(BaseModel):
|
||||||
name: Optional[str] = Field(None, min_length=1, max_length=200)
|
name: Optional[str] = Field(None, min_length=1, max_length=200)
|
||||||
@@ -93,6 +94,7 @@ class ScriptTemplateListItem(BaseModel):
|
|||||||
requires_modules: list[str]
|
requires_modules: list[str]
|
||||||
is_verified: bool
|
is_verified: bool
|
||||||
usage_count: int
|
usage_count: int
|
||||||
|
language: str | None = None
|
||||||
|
|
||||||
class Config:
|
class Config:
|
||||||
from_attributes = True
|
from_attributes = True
|
||||||
|
|||||||
Reference in New Issue
Block a user