# 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 .user import User
from .completion_condition import CompletionCondition


class TrainingRegistration(object):
    _types = {
        "training_id": str,
        "session_id": str,
        "learner": User,
        "enroll_at": int,
        "enroll_type": int,
        "enroll_sub_type": int,
        "sign_in_state": int,
        "learning_state": int,
        "completion_condition": CompletionCondition,
        "meeting_duration": int,
        "playback_duration": int,
        "finished_at": int,
    }

    def __init__(self, d=None):
        self.training_id: Optional[str] = None
        self.session_id: Optional[str] = None
        self.learner: Optional[User] = None
        self.enroll_at: Optional[int] = None
        self.enroll_type: Optional[int] = None
        self.enroll_sub_type: Optional[int] = None
        self.sign_in_state: Optional[int] = None
        self.learning_state: Optional[int] = None
        self.completion_condition: Optional[CompletionCondition] = None
        self.meeting_duration: Optional[int] = None
        self.playback_duration: Optional[int] = None
        self.finished_at: Optional[int] = None
        init(self, d, self._types)

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


class TrainingRegistrationBuilder(object):
    def __init__(self) -> None:
        self._training_registration = TrainingRegistration()

    def training_id(self, training_id: str) -> "TrainingRegistrationBuilder":
        self._training_registration.training_id = training_id
        return self

    def session_id(self, session_id: str) -> "TrainingRegistrationBuilder":
        self._training_registration.session_id = session_id
        return self

    def learner(self, learner: User) -> "TrainingRegistrationBuilder":
        self._training_registration.learner = learner
        return self

    def enroll_at(self, enroll_at: int) -> "TrainingRegistrationBuilder":
        self._training_registration.enroll_at = enroll_at
        return self

    def enroll_type(self, enroll_type: int) -> "TrainingRegistrationBuilder":
        self._training_registration.enroll_type = enroll_type
        return self

    def enroll_sub_type(self, enroll_sub_type: int) -> "TrainingRegistrationBuilder":
        self._training_registration.enroll_sub_type = enroll_sub_type
        return self

    def sign_in_state(self, sign_in_state: int) -> "TrainingRegistrationBuilder":
        self._training_registration.sign_in_state = sign_in_state
        return self

    def learning_state(self, learning_state: int) -> "TrainingRegistrationBuilder":
        self._training_registration.learning_state = learning_state
        return self

    def completion_condition(self, completion_condition: CompletionCondition) -> "TrainingRegistrationBuilder":
        self._training_registration.completion_condition = completion_condition
        return self

    def meeting_duration(self, meeting_duration: int) -> "TrainingRegistrationBuilder":
        self._training_registration.meeting_duration = meeting_duration
        return self

    def playback_duration(self, playback_duration: int) -> "TrainingRegistrationBuilder":
        self._training_registration.playback_duration = playback_duration
        return self

    def finished_at(self, finished_at: int) -> "TrainingRegistrationBuilder":
        self._training_registration.finished_at = finished_at
        return self

    def build(self) -> "TrainingRegistration":
        return self._training_registration