from datetime import datetime
from typing import Optional

from beanie import Document
from .constants import MediaType, DataFormat


class DocumentDoc(Document):
    document_id: Optional[str]
    file_name: str
    created_by: str
    create_time: datetime
    updated_by: str
    update_time: datetime
    location: Optional[str]
    version_number: int
    media_type: MediaType
    data_format: DataFormat
    is_deleted: Optional[bool] = False

    class Settings:
        name = "document"