from dataclasses import dataclass
from typing import Union

from bson import ObjectId

from modules.database import col_users


@dataclass
class PyroUser:
    """Dataclass of DB entry of a user"""

    __slots__ = ("_id", "id", "locale")

    _id: ObjectId
    id: int
    locale: Union[str, None]

    async def update_locale(self, locale: str) -> None:
        """Change user's locale stored in the database

        ### Args:
            * locale (`str`): New locale to be set
        """
        await col_users.update_one({"_id": self._id}, {"$set": {"locale": locale}})