from datetime import datetime
from typing import Optional

from pydantic import BaseModel


class SMessage(BaseModel):
    message: Optional[str] = None
    image_url: Optional[str] = None
    chat_id: int
    user_id: int
    username: str
    created_at: datetime
    avatar_image: str

    class Config:
        from_attributes = True


class SLastMessages:
    def __init__(
            self,
            messages_loaded: int,
            messages_to_get: int,
    ):
        self.messages_loaded = messages_loaded
        self.messages_to_get = messages_to_get


class SMessageSchema(BaseModel):
    message: str