import datetime
from typing import Optional, Union

from pydantic import BaseModel


class IDData(BaseModel):
    id: int


class ShareItem(BaseModel):
    expire_value: int
    expire_style: str = "day"
    filename: str


class DeleteItem(BaseModel):
    filename: str


class LoginData(BaseModel):
    password: str


class UpdateFileData(BaseModel):
    id: int
    code: Optional[str] = None
    prefix: Optional[str] = None
    suffix: Optional[str] = None
    expired_at: Optional[Union[datetime.datetime, str]] = None
    expired_count: Optional[int] = None