src/ctp_slack_bot/models/google_drive.py

from datetime import datetime
from pydantic import BaseModel, ConfigDict
from typing import Self


class GoogleDriveMetadata(BaseModel):
    """Represents Google Drive file or folder metadata."""

    id: str
    name: str
    modified_time: datetime
    mime_type: str
    folder_path: str

    model_config = ConfigDict(frozen=True)

    @classmethod
    def from_folder_path_and_dict(cls, folder_path: str, dict: dict) -> Self:
        id = dict["id"]
        name = dict["name"]
        modified_time = datetime.fromisoformat(dict["modifiedTime"])
        mime_type = dict["mimeType"]
        return GoogleDriveMetadata(id=id, name=name, modified_time=modified_time, mime_type=mime_type, folder_path=folder_path)