a

Backend/schemas/diary.py

@@ -0,0 +1,109 @@
+from pydantic import BaseModel, field_validator
+from datetime import datetime
+from typing import Optional, List
+from fastapi import Form
+
+
+class DiaryCreate(BaseModel):
+    title: str
+    content: str
+    category: str
+
+    @classmethod
+    def as_form(
+        cls,
+        title: str = Form(...),
+        content: str = Form(...),
+        category: str = Form(...),
+    ) -> "DiaryCreate":
+        return cls(title=title, content=content, category=category)
+
+    @field_validator("title")
+    @classmethod
+    def validate_title(cls, v):
+        if len(v.strip()) < 1:
+            raise ValueError("Title cannot be empty")
+        if len(v) > 100:
+            raise ValueError("Title must be less than 100 characters")
+        return v.strip()
+
+    @field_validator("content")
+    @classmethod
+    def validate_content(cls, v):
+        if len(v.strip()) < 1:
+            raise ValueError("Content cannot be empty")
+        if len(v) > 5000:
+            raise ValueError("Content must be less than 5000 characters")
+        return v.strip()
+
+
+class DiaryUpdate(BaseModel):
+    title: Optional[str] = None
+    content: Optional[str] = None
+    category: Optional[str] = None
+
+    @classmethod
+    def as_form(
+        cls,
+        title: Optional[str] = Form(None),
+        content: Optional[str] = Form(None),
+        category: Optional[str] = Form(None),
+    ) -> "DiaryUpdate":
+        return cls(title=title, content=content, category=category)
+
+
+class DiaryResponse(BaseModel):
+    id: int
+    user_id: int
+    title: str
+    content: str
+    images: List[str]
+    category: str
+    created_at: datetime
+    is_submitted: bool
+    email_sent: bool
+
+
+class Diary:
+    def __init__(
+        self,
+        id: int,
+        user_id: int,
+        title: str,
+        content: str,
+        images: str,  # JSON string
+        category: str,
+        created_at: datetime,
+        is_submitted: bool = False,
+        email_sent: bool = False,
+    ):
+        self.id = id
+        self.user_id = user_id
+        self.title = title
+        self.content = content
+        self.images = images
+        self.category = category
+        self.created_at = created_at
+        self.is_submitted = is_submitted
+        self.email_sent = email_sent
+
+    @property
+    def image_list(self) -> List[str]:
+        return (
+            [img.strip() for img in self.images.split(",") if img.strip()]
+            if self.images
+            else []
+        )
+
+    def to_response(self) -> DiaryResponse:
+        return DiaryResponse(
+            id=self.id,
+            user_id=self.user_id,
+            title=self.title,
+            content=self.content,
+            images=self.image_list,
+            category=self.category,
+            created_at=self.created_at,
+            is_submitted=self.is_submitted,
+            email_sent=self.email_sent,
+        )