a

Backend/schemas/friendship.py

@@ -0,0 +1,59 @@
+from pydantic import BaseModel, field_validator
+from datetime import datetime
+import enum
+
+
+class FriendshipStatus(str, enum.Enum):
+    PENDING = "pending"
+    ACCEPTED = "accepted"
+    REJECTED = "rejected"
+
+
+class FriendshipRequest(BaseModel):
+    friend_username: str
+
+    @field_validator("friend_username")
+    @classmethod
+    def validate_friend_username(cls, v):
+        if len(v.strip()) < 1:
+            raise ValueError("Friend username cannot be empty")
+        return v.strip()
+
+
+class FriendshipResponse(BaseModel):
+    id: int
+    user_id: int
+    friend_id: int
+    friend_username: str
+    status: FriendshipStatus
+    created_at: datetime
+
+
+class FriendshipUpdate(BaseModel):
+    status: FriendshipStatus
+
+
+class Friendship:
+    def __init__(
+        self,
+        id: int,
+        user_id: int,
+        friend_id: int,
+        status: str,
+        created_at: datetime,
+    ):
+        self.id = id
+        self.user_id = user_id
+        self.friend_id = friend_id
+        self.status = status
+        self.created_at = created_at
+
+    def to_response(self, friend_username: str) -> FriendshipResponse:
+        return FriendshipResponse(
+            id=self.id,
+            user_id=self.user_id,
+            friend_id=self.friend_id,
+            friend_username=friend_username,
+            status=FriendshipStatus(self.status),
+            created_at=self.created_at,
+        )