from datetime import datetime

from pydantic import Field

from trauma.api.message.dto import Author
from trauma.core.database import MongoBaseModel


class MessageModel(MongoBaseModel):
    chatId: str
    author: Author
    text: str
    datetimeInserted: datetime = Field(default_factory=datetime.now)
    datetimeUpdated: datetime = Field(default_factory=datetime.now)