99af0164e6
Backend: - Add PostgreSQL service to docker-compose with health checks - Add SQLAlchemy async models for all entities (Agency, User, Campaign, Proof, ProofVersion, FlaggedItem, ResolvedItem, ErrorItem) - Add Alembic migration framework with initial schema migration - Add repository layer for CRUD operations - Add REST API endpoints for campaigns, proofs, and audit items - Add file storage service for proof uploads - Update WebSocket handler to optionally persist analysis results Frontend: - Add apiService.ts for REST API communication - Update geminiService.ts to support database persistence options Deployment: - Update deploy.sh to handle database migrations (6-step process) - Update Dockerfile to include alembic configuration - Add PostgreSQL environment variables to .env templates 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
165 lines
3.6 KiB
Python
165 lines
3.6 KiB
Python
"""Pydantic schemas for API request/response validation."""
|
|
import uuid
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
# Campaign schemas
|
|
class CampaignCreate(BaseModel):
|
|
name: str
|
|
workfront_id: Optional[str] = None
|
|
client_lead: Optional[str] = None
|
|
agency_lead: Optional[str] = None
|
|
brand_guidelines: Optional[str] = None
|
|
|
|
|
|
class CampaignUpdate(BaseModel):
|
|
name: Optional[str] = None
|
|
workfront_id: Optional[str] = None
|
|
client_lead: Optional[str] = None
|
|
agency_lead: Optional[str] = None
|
|
brand_guidelines: Optional[str] = None
|
|
status: Optional[str] = None
|
|
|
|
|
|
class CampaignResponse(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
workfront_id: Optional[str]
|
|
client_lead: Optional[str]
|
|
agency_lead: Optional[str]
|
|
brand_guidelines: Optional[str]
|
|
status: str
|
|
agency: Optional[str]
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
proofs: int = 0
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# Proof schemas
|
|
class ProofCreate(BaseModel):
|
|
proof_name: str
|
|
channel: Optional[str] = None
|
|
sub_channel: Optional[str] = None
|
|
proof_type: Optional[str] = None
|
|
|
|
|
|
class ProofVersionResponse(BaseModel):
|
|
id: uuid.UUID
|
|
version: int
|
|
file_storage_key: Optional[str]
|
|
thumbnail_url: Optional[str]
|
|
agent_review: Optional[dict]
|
|
overall_status: Optional[str]
|
|
workfront_id: Optional[str]
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ProofResponse(BaseModel):
|
|
id: uuid.UUID
|
|
proof_name: str
|
|
channel: Optional[str]
|
|
sub_channel: Optional[str]
|
|
proof_type: Optional[str]
|
|
workfront_id: Optional[str]
|
|
created_at: datetime
|
|
versions: list[ProofVersionResponse] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# Audit schemas
|
|
class FlaggedItemCreate(BaseModel):
|
|
proof_version_id: uuid.UUID
|
|
agent_flagged: str
|
|
comments: Optional[str] = None
|
|
|
|
|
|
class FlaggedItemResponse(BaseModel):
|
|
id: uuid.UUID
|
|
proof_version_id: uuid.UUID
|
|
agent_flagged: str
|
|
comments: Optional[str]
|
|
submitter_name: Optional[str]
|
|
submitter_agency: Optional[str]
|
|
campaign_name: Optional[str]
|
|
proof_name: Optional[str]
|
|
version: Optional[int]
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ResolvedItemCreate(BaseModel):
|
|
proof_version_id: uuid.UUID
|
|
agent: str
|
|
issue: Optional[str] = None
|
|
resolution: Optional[str] = None
|
|
|
|
|
|
class ResolvedItemResponse(BaseModel):
|
|
id: uuid.UUID
|
|
proof_version_id: uuid.UUID
|
|
agent: str
|
|
issue: Optional[str]
|
|
resolution: Optional[str]
|
|
submitter_name: Optional[str]
|
|
submitter_agency: Optional[str]
|
|
campaign_name: Optional[str]
|
|
proof_name: Optional[str]
|
|
version: Optional[int]
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ErrorItemResponse(BaseModel):
|
|
id: uuid.UUID
|
|
proof_version_id: uuid.UUID
|
|
error_summary: Optional[str]
|
|
campaign_name: Optional[str]
|
|
proof_name: Optional[str]
|
|
version: Optional[int]
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# Analytics schemas
|
|
class AnalyticsResponse(BaseModel):
|
|
total_reviews: int
|
|
passed: int
|
|
failed: int
|
|
errors: int
|
|
legal_review: int
|
|
|
|
|
|
# Dropdown options schemas
|
|
class DropdownOptionsResponse(BaseModel):
|
|
campaigns: list[str]
|
|
channels: dict[str, dict[str, list[str]]]
|
|
|
|
|
|
# User schemas
|
|
class UserResponse(BaseModel):
|
|
id: uuid.UUID
|
|
email: str
|
|
name: str
|
|
role: str
|
|
agency: Optional[str]
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|