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:
Michael Chihlas
2026-03-21 17:17:41 -04:00
parent 25d7191575
commit 628761473f
4 changed files with 69 additions and 0 deletions

View File

@@ -76,6 +76,8 @@ async def list_templates(
search: Optional[str] = Query(None),
tags: Optional[str] = Query(None, description="Comma-separated tags"),
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]:
query = (
select(ScriptTemplate)
@@ -116,6 +118,12 @@ async def list_templates(
# engineers see only their own
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))
templates = result.scalars().all()

View File

@@ -52,6 +52,10 @@ class ScriptTemplate(Base):
description: 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)
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)
default_values: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)
validation_rules: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)

View File

@@ -62,6 +62,7 @@ class ScriptTemplateCreate(BaseModel):
estimated_runtime: Optional[str] = None
requires_elevation: bool = False
requires_modules: list[str] = Field(default_factory=list)
language: str | None = None
class ScriptTemplateUpdate(BaseModel):
name: Optional[str] = Field(None, min_length=1, max_length=200)
@@ -93,6 +94,7 @@ class ScriptTemplateListItem(BaseModel):
requires_modules: list[str]
is_verified: bool
usage_count: int
language: str | None = None
class Config:
from_attributes = True