92 lines
2.3 KiB
Python
92 lines
2.3 KiB
Python
from dataclasses import dataclass, field
|
|
from datetime import datetime
|
|
from typing import Dict, List
|
|
|
|
from fastapi import Request
|
|
|
|
|
|
@dataclass
|
|
class StoreData:
|
|
id: int
|
|
created_at: datetime
|
|
store_name: str
|
|
store_category: str | None = None
|
|
store_region: str | None = None
|
|
store_address: str | None = None
|
|
store_phone_number: str | None = None
|
|
store_info: str | None = None
|
|
|
|
|
|
@dataclass
|
|
class AttributeData:
|
|
id: int
|
|
attr_category: str
|
|
attr_value: str
|
|
created_at: datetime
|
|
|
|
|
|
@dataclass
|
|
class SongSampleData:
|
|
id: int
|
|
ai: str
|
|
ai_model: str
|
|
sample_song: str
|
|
season: str | None = None
|
|
num_of_people: int | None = None
|
|
people_category: str | None = None
|
|
genre: str | None = None
|
|
|
|
|
|
@dataclass
|
|
class PromptTemplateData:
|
|
id: int
|
|
prompt: str
|
|
description: str | None = None
|
|
|
|
|
|
@dataclass
|
|
class SongFormData:
|
|
store_name: str
|
|
store_id: str
|
|
prompts: str
|
|
attributes: Dict[str, str] = field(default_factory=dict)
|
|
attributes_str: str = ""
|
|
lyrics_ids: List[int] = field(default_factory=list)
|
|
llm_model: str = "gpt-4o"
|
|
|
|
@classmethod
|
|
async def from_form(cls, request: Request):
|
|
"""Request의 form 데이터로부터 dataclass 인스턴스 생성"""
|
|
form_data = await request.form()
|
|
|
|
# 고정 필드명들
|
|
fixed_keys = {"store_info_name", "store_id", "llm_model", "prompts"}
|
|
|
|
# lyrics-{id} 형태의 모든 키를 찾아서 ID 추출
|
|
lyrics_ids = []
|
|
attributes = {}
|
|
|
|
for key, value in form_data.items():
|
|
if key.startswith("lyrics-"):
|
|
lyrics_id = key.split("-")[1]
|
|
lyrics_ids.append(int(lyrics_id))
|
|
elif key not in fixed_keys:
|
|
attributes[key] = value
|
|
|
|
# attributes를 문자열로 변환
|
|
attributes_str = (
|
|
"\r\n\r\n".join([f"{key} : {value}" for key, value in attributes.items()])
|
|
if attributes
|
|
else ""
|
|
)
|
|
|
|
return cls(
|
|
store_name=form_data.get("store_info_name", ""),
|
|
store_id=form_data.get("store_id", ""),
|
|
attributes=attributes,
|
|
attributes_str=attributes_str,
|
|
lyrics_ids=lyrics_ids,
|
|
llm_model=form_data.get("llm_model", "gpt-4o"),
|
|
prompts=form_data.get("prompts", ""),
|
|
)
|