mirror of
https://github.com/khoaliber/LetterFeed.git
synced 2026-03-02 21:19:13 +00:00
64 lines
1.3 KiB
Python
64 lines
1.3 KiB
Python
from typing import List
|
|
|
|
from pydantic import BaseModel, ConfigDict, EmailStr, field_validator
|
|
|
|
from app.core.slug import sanitize_slug
|
|
|
|
|
|
class SenderBase(BaseModel):
|
|
"""Base schema for a sender."""
|
|
|
|
email: EmailStr
|
|
|
|
|
|
class SenderCreate(SenderBase):
|
|
"""Schema for creating a new sender."""
|
|
|
|
pass
|
|
|
|
|
|
class Sender(SenderBase):
|
|
"""Schema for retrieving a sender with its ID and newsletter ID."""
|
|
|
|
id: str
|
|
newsletter_id: str
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
class NewsletterBase(BaseModel):
|
|
"""Base schema for a newsletter."""
|
|
|
|
name: str
|
|
slug: str | None = None
|
|
move_to_folder: str | None = None
|
|
extract_content: bool = False
|
|
|
|
@field_validator("slug")
|
|
def sanitize_slug_field(cls, v: str | None) -> str | None:
|
|
"""Sanitize slug."""
|
|
return sanitize_slug(v)
|
|
|
|
|
|
class NewsletterCreate(NewsletterBase):
|
|
"""Schema for creating a new newsletter."""
|
|
|
|
sender_emails: List[EmailStr]
|
|
|
|
|
|
class NewsletterUpdate(NewsletterBase):
|
|
"""Schema for updating an existing newsletter."""
|
|
|
|
sender_emails: List[EmailStr]
|
|
|
|
|
|
class Newsletter(NewsletterBase):
|
|
"""Schema for retrieving a newsletter with its ID, active status, senders, and entries count."""
|
|
|
|
id: str
|
|
is_active: bool
|
|
senders: List[Sender] = []
|
|
entries_count: int
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|