schemas.py

from datetime import datetime
from typing import Optional, List, Dict
from pydantic import BaseModel, Field, HttpUrl

#기본 요청
class RequestBase(BaseModel):
    job_id: str
    schedule_id: str
    schedule_his_id: Optional[int] = None

#기본 응답
class ResponseBase(BaseModel):
    job_id: str
    schedule_id: str
    status: str


#네이버 키워드 추출
class RequestNaverSearch(RequestBase):
    tag: str
    category: str
    startDate :datetime
    endDate :datetime

class ResponseNaverSearch(ResponseBase):
    category: str
    keyword: str
    total_keyword: dict[int, str]


#키워드 사다구몰 검증
class RequestSadaguValidate(RequestBase):
    tag: str
    category: str

class ResponseSadaguValidate(ResponseBase):
    keyword: str

#사다구몰 크롤링
class RequestSadaguCrawl(RequestBase):
    tag: str = Field(..., description="크롤링 태그 (예: 'detail')")
    product_url: HttpUrl = Field(..., description="크롤링할 상품의 URL")
    use_selenium: bool = Field(default=True, description="Selenium 사용 여부")
    include_images: bool = Field(default=False, description="이미지 정보 포함 여부")

class ResponseSadaguCrawl(ResponseBase):
    tag: str
    product_url: str
    use_selenium: bool
    include_images: bool
    product_detail: Optional[dict] = None
    crawled_at: Optional[str] = None

#블로그 생성
class RequestBlogCreate(RequestBase):
    tag: str
    category: str

class ResponseBlogCreate(ResponseBase):
    pass

#블로그 배포
class RequestBlogPublish(RequestBase):
    tag: str
    category: str

class ResponseBlogPublish(ResponseBase):
    pass