# Code generated by Lark OpenAPI.

from typing import Any, Optional, Union, Dict, List, Set, IO, Callable, Type
from lark_oapi.core.construct import init
from .attendance_rule import AttendanceRule
from .attendance_rule_description import AttendanceRuleDescription
from .attendance_rule_description import AttendanceRuleDescription


class UserAttendanceRule(object):
    _types = {
        "user_id": str,
        "attendance_rule": AttendanceRule,
        "attendance_rule_descriptions": List[AttendanceRuleDescription],
        "shift_descriptions": List[AttendanceRuleDescription],
    }

    def __init__(self, d=None):
        self.user_id: Optional[str] = None
        self.attendance_rule: Optional[AttendanceRule] = None
        self.attendance_rule_descriptions: Optional[List[AttendanceRuleDescription]] = None
        self.shift_descriptions: Optional[List[AttendanceRuleDescription]] = None
        init(self, d, self._types)

    @staticmethod
    def builder() -> "UserAttendanceRuleBuilder":
        return UserAttendanceRuleBuilder()


class UserAttendanceRuleBuilder(object):
    def __init__(self) -> None:
        self._user_attendance_rule = UserAttendanceRule()

    def user_id(self, user_id: str) -> "UserAttendanceRuleBuilder":
        self._user_attendance_rule.user_id = user_id
        return self

    def attendance_rule(self, attendance_rule: AttendanceRule) -> "UserAttendanceRuleBuilder":
        self._user_attendance_rule.attendance_rule = attendance_rule
        return self

    def attendance_rule_descriptions(self, attendance_rule_descriptions: List[
        AttendanceRuleDescription]) -> "UserAttendanceRuleBuilder":
        self._user_attendance_rule.attendance_rule_descriptions = attendance_rule_descriptions
        return self

    def shift_descriptions(self, shift_descriptions: List[AttendanceRuleDescription]) -> "UserAttendanceRuleBuilder":
        self._user_attendance_rule.shift_descriptions = shift_descriptions
        return self

    def build(self) -> "UserAttendanceRule":
        return self._user_attendance_rule