2025 SSF Public

Backend/schemas/avatar.py

@@ -0,0 +1,106 @@
+from pydantic import BaseModel
+import enum
+
+
+class AvatarType(str, enum.Enum):
+    MALE = "남성"
+    FEMALE = "여성"
+
+
+class TopClothesType(str, enum.Enum):
+    SCHOOL_CLOTHES = "교복"
+    SCHOOL_CLOTHES_2 = "교복 조끼"
+    ANA_CLOTHES = "AnA 동잠"
+    SUSPENDERS_CLOTHES_1 = "멜빵 바지"
+    SUSPENDERS_CLOTHES_2 = "멜빵 치마"
+    RAINBOW_CLOTHES = "무지개 맨투맨"
+    SANTA_CLOTHES = "산타"
+
+
+class BottomClothesType(str, enum.Enum):
+    SCHOOL_CLOTHES = "교복 바지"
+    SCHOOL_CLOTHES_2 = "교복 치마"
+    SCHOOL_CLOTHES_3 = "교복 조끼 바지"
+    SCHOOL_CLOTHES_4 = "교복 조끼 치마"
+    SANTA_CLOTHES = "산타 바지"
+    JEANS = "청바지"
+
+
+avatar_path_ = {"남성": "public/avatar/남자.png", "여성": "public/avatar/여자.png"}
+top_clothe_path_ = {
+    "교복": "public/avatar/교복상의.png",
+    "교복 조끼": "public/avatar/교복조끼상의.png",
+    "AnA 동잠": "public/avatar/동잠상의.png",
+    "멜빵 바지": "public/avatar/멜빵바지상의.png",
+    "멜빵 치마": "public/avatar/멜빵치마상의.png",
+    "무지개 맨투맨": "public/avatar/무지개맨투맨상의.png",
+    "산타": "public/avatar/산타상의.png",
+}
+bottom_clothe_path_ = {
+    "교복 바지": "public/avatar/교복하의남.png",
+    "교복 치마": "public/avatar/교복하의여.png",
+    "교복 조끼 바지": "public/avatar/교복조끼하의남.png",
+    "교복 조끼 치마": "public/avatar/교복조끼하의여.png",
+    "산타 바지": "public/avatar/산타하의.png",
+    "청바지": "public/avatar/청바지하의.png",
+}
+
+
+class AvatarTypeResponse(BaseModel):
+    name: str
+    path: str
+
+
+class AvatarUpdate(BaseModel):
+    avatar_type: AvatarType
+    top_clothe_type: TopClothesType
+    bottom_clothe_type: BottomClothesType
+
+
+class AvatarResponse(BaseModel):
+    id: int
+    user_id: int
+    avatar_type: AvatarTypeResponse
+    top_clothe_type: AvatarTypeResponse
+    bottom_clothe_type: AvatarTypeResponse
+
+
+class AvatarOptions(BaseModel):
+    avatar_types: list[str]
+    top_clothe_types: list[str]
+    bottom_clothe_types: list[str]
+
+
+
+class Avatar:
+    def __init__(
+        self,
+        id: int,
+        user_id: int,
+        avatar_type: AvatarType,
+        top_clothe_type: TopClothesType,
+        bottom_clothe_type: BottomClothesType,
+    ):
+        self.id = id
+        self.user_id = user_id
+        self.avatar_type = avatar_type
+        self.top_clothe_type = top_clothe_type
+        self.bottom_clothe_type = bottom_clothe_type
+
+    def to_response(self) -> AvatarResponse:
+        return AvatarResponse(
+            id=self.id,
+            user_id=self.user_id,
+            avatar_type=AvatarTypeResponse(
+                name=self.avatar_type,
+                path=avatar_path_[self.avatar_type],
+            ),
+            top_clothe_type=AvatarTypeResponse(
+                name=self.top_clothe_type,
+                path=top_clothe_path_[self.top_clothe_type],
+            ),
+            bottom_clothe_type=AvatarTypeResponse(
+                name=self.bottom_clothe_type,
+                path=bottom_clothe_path_[self.bottom_clothe_type],
+            ),
+        )