32 lines
786 B
Python
32 lines
786 B
Python
from datetime import datetime, UTC
|
|
from enum import Enum
|
|
from typing import Optional, Any
|
|
|
|
from pydantic import BaseModel, computed_field
|
|
|
|
|
|
class AssetType(str, Enum):
|
|
IMAGE = 'image'
|
|
PROMPT = 'prompt'
|
|
|
|
|
|
class Asset(BaseModel):
|
|
id: Optional[str] = None
|
|
name: str
|
|
type: AssetType
|
|
linked_char_id: Optional[str] = None
|
|
data: Optional[bytes] = None
|
|
tg_doc_file_id: str
|
|
tg_photo_file_id: Optional[str] = None
|
|
created_at: datetime = datetime.now(UTC)
|
|
|
|
# --- CALCULATED FIELD ---
|
|
@computed_field
|
|
def link(self) -> str:
|
|
"""
|
|
Это поле автоматически вычислится и попадет в model_dump() / .json()
|
|
"""
|
|
if self.id:
|
|
return f"/api/assets/{self.id}"
|
|
return ""
|