Agent Zero
66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
"""Pydantic schemas for Activity endpoints."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, model_validator
|
|
|
|
from app.models.activity import ActivityType
|
|
|
|
|
|
class ActivityBase(BaseModel):
|
|
type: ActivityType
|
|
subject: str = Field(..., min_length=1, max_length=255)
|
|
body: str | None = None
|
|
due_date: datetime | None = None
|
|
account_id: int | None = None
|
|
contact_id: int | None = None
|
|
deal_id: int | None = None
|
|
|
|
@model_validator(mode="after")
|
|
def _check_at_least_one_parent(self) -> ActivityBase:
|
|
if self.account_id is None and self.contact_id is None and self.deal_id is None:
|
|
raise ValueError("At least one of account_id, contact_id, deal_id must be set")
|
|
return self
|
|
|
|
|
|
class ActivityCreate(ActivityBase):
|
|
pass
|
|
|
|
|
|
class ActivityUpdate(BaseModel):
|
|
type: ActivityType | None = None
|
|
subject: str | None = Field(None, min_length=1, max_length=255)
|
|
body: str | None = None
|
|
due_date: datetime | None = None
|
|
account_id: int | None = None
|
|
contact_id: int | None = None
|
|
deal_id: int | None = None
|
|
|
|
|
|
class ActivityOut(ActivityBase):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
org_id: int
|
|
owner_id: int
|
|
completed_at: datetime | None = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
deleted_at: datetime | None = None
|
|
|
|
|
|
class ActivityCompleteRequest(BaseModel):
|
|
"""Body for PATCH /api/v1/activities/{id}/complete."""
|
|
|
|
outcome: str | None = Field(None, max_length=2048)
|
|
|
|
|
|
__all__ = [
|
|
"ActivityCompleteRequest",
|
|
"ActivityCreate",
|
|
"ActivityOut",
|
|
"ActivityUpdate",
|
|
]
|