203671a7bf
Add the experience service with sites through analytics/AI hooks, migrations through 0011, and phase docs/manifests marking the track complete. Co-authored-by: Cursor <cursoragent@cursor.com>
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
"""Filtering, search and sorting specifications — Phase 11.10 AI hooks."""
|
|
from dataclasses import dataclass
|
|
from uuid import UUID
|
|
|
|
from sqlalchemy import asc, desc, or_
|
|
|
|
from app.models.ai_hooks import (
|
|
ExperienceAiContentDispatch,
|
|
ExperienceAiContentHookRegistration,
|
|
)
|
|
|
|
HOOK_REGISTRATION_SORT = frozenset(
|
|
{"created_at", "updated_at", "code", "name", "status", "task_kind"}
|
|
)
|
|
DISPATCH_SORT = frozenset({"created_at", "updated_at", "status", "task"})
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class AiHooksListSpec:
|
|
workspace_id: UUID | None = None
|
|
site_id: UUID | None = None
|
|
page_id: UUID | None = None
|
|
registration_id: UUID | None = None
|
|
status: str | None = None
|
|
task_kind: str | None = None
|
|
q: str | None = None
|
|
sort_by: str = "created_at"
|
|
sort_dir: str = "desc"
|
|
model = None
|
|
sort_fields = frozenset()
|
|
|
|
def filter_clauses(self):
|
|
m = self.model
|
|
clauses = []
|
|
for field in (
|
|
"workspace_id",
|
|
"site_id",
|
|
"page_id",
|
|
"registration_id",
|
|
"status",
|
|
"task_kind",
|
|
):
|
|
value = getattr(self, field)
|
|
if value is not None and hasattr(m, field):
|
|
clauses.append(getattr(m, field) == value)
|
|
if self.q:
|
|
term = f"%{self.q.strip()}%"
|
|
searchable = [
|
|
getattr(m, name).ilike(term)
|
|
for name in ("code", "name", "task", "context_ref")
|
|
if hasattr(m, name)
|
|
]
|
|
if searchable:
|
|
clauses.append(or_(*searchable))
|
|
return clauses
|
|
|
|
def apply(self, stmt):
|
|
for clause in self.filter_clauses():
|
|
stmt = stmt.where(clause)
|
|
direction = desc if self.sort_dir.lower() == "desc" else asc
|
|
sort_col = getattr(self.model, self.sort_by, self.model.created_at)
|
|
if self.sort_by not in self.sort_fields:
|
|
sort_col = self.model.created_at
|
|
return stmt.order_by(direction(sort_col))
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class AiContentHookRegistrationListSpec(AiHooksListSpec):
|
|
model = ExperienceAiContentHookRegistration
|
|
sort_fields = HOOK_REGISTRATION_SORT
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class AiContentDispatchListSpec(AiHooksListSpec):
|
|
model = ExperienceAiContentDispatch
|
|
sort_fields = DISPATCH_SORT
|