<filename>nerddiary/user/user.py
""" User model """

from __future__ import annotations

from datetime import tzinfo

import pytz
from pydantic import BaseModel, PrivateAttr, validator
from pydantic.fields import Field

from ..poll.poll import Poll
from ..primitive.timezone import TimeZone
from ..report.report import Report

from typing import Dict, List, Optional


class User(BaseModel):
    id: str = Field(description="This user id", regex=r"^\w{1,64}$")
    username: str | None = Field(default=None, description="Optional user name")
    lang_code: str = Field(
        default="en", min_length=2, max_length=2, description="User preferred language (2 letter code)"
    )
    timezone: TimeZone = pytz.timezone("US/Eastern")  # type: ignore
    polls: Optional[List[Poll]] = Field(min_items=1)
    reports: Optional[List[Report]] = Field(min_items=1)

    _polls_dict: Dict[str, Poll] = PrivateAttr(default={})
    """ Dictionary of polls for workflow convinience
    """

    class Config:
        title = "User Configuration"
        extra = "forbid"
        json_encoders = {tzinfo: lambda t: str(t)}

    def __init__(self, **data) -> None:
        super().__init__(**data)

        # convert_reminder_times_to_local_if_set
        if self.polls:
            # Create help mappings for workflow processing
            self._polls_dict = {}

            for poll in self.polls:
                self._polls_dict |= {poll.poll_name: poll}

                if poll.reminder_time:
                    poll.reminder_time = poll.reminder_time.replace(tzinfo=self.timezone)

    @validator("polls")
    def poll_names_must_be_unique(cls, v: List[Poll]):
        if v:
            poll_names = [p.poll_name for p in v]
            poll_names_set = set(poll_names)
            if len(poll_names_set) != len(poll_names):
                raise ValueError("Poll names must be unique")
        return v

    @validator("polls")
    def poll_commands_must_be_unique(cls, v: List[Poll]):
        if v:
            poll_commands = [p.command for p in v if p.command is not None]
            poll_commands_set = set(poll_commands)
            if len(poll_commands_set) != len(poll_commands):
                raise ValueError("Poll commands must be unique")
        return v

<filename>nerddiary/user/user.py
""" User model """

from __future__ import annotations

from datetime import tzinfo

import pytz
from pydantic import BaseModel, PrivateAttr, validator
from pydantic.fields import Field

from ..poll.poll import Poll
from ..primitive.timezone import TimeZone
from ..report.report import Report

from typing import Dict, List, Optional


class User(BaseModel):
    id: str = Field(description="This user id", regex=r"^\w{1,64}$")
    username: str | None = Field(default=None, description="Optional user name")
    lang_code: str = Field(
        default="en", min_length=2, max_length=2, description="User preferred language (2 letter code)"
    )
    timezone: TimeZone = pytz.timezone("US/Eastern")  # type: ignore
    polls: Optional[List[Poll]] = Field(min_items=1)
    reports: Optional[List[Report]] = Field(min_items=1)

    _polls_dict: Dict[str, Poll] = PrivateAttr(default={})
    """ Dictionary of polls for workflow convinience
    """

    class Config:
        title = "User Configuration"
        extra = "forbid"
        json_encoders = {tzinfo: lambda t: str(t)}

    def __init__(self, **data) -> None:
        super().__init__(**data)

        # convert_reminder_times_to_local_if_set
        if self.polls:
            # Create help mappings for workflow processing
            self._polls_dict = {}

            for poll in self.polls:
                self._polls_dict |= {poll.poll_name: poll}

                if poll.reminder_time:
                    poll.reminder_time = poll.reminder_time.replace(tzinfo=self.timezone)

    @validator("polls")
    def poll_names_must_be_unique(cls, v: List[Poll]):
        if v:
            poll_names = [p.poll_name for p in v]
            poll_names_set = set(poll_names)
            if len(poll_names_set) != len(poll_names):
                raise ValueError("Poll names must be unique")
        return v

    @validator("polls")
    def poll_commands_must_be_unique(cls, v: List[Poll]):
        if v:
            poll_commands = [p.command for p in v if p.command is not None]
            poll_commands_set = set(poll_commands)
            if len(poll_commands_set) != len(poll_commands):
                raise ValueError("Poll commands must be unique")
        return v

User model # type: ignore Dictionary of polls for workflow convinience # convert_reminder_times_to_local_if_set # Create help mappings for workflow processing