from typing import Union

from attrs import define
from bson import ObjectId

from modules.database import col_users


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

    _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
        """
        col_users.update_one({"_id": self._id}, {"$set": {"locale": locale}})