# 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 .enum import Enum
from .enum import Enum
from .custom_field_data import CustomFieldData
from .enum import Enum
from .enum import Enum
from .enum import Enum
from .assessment import Assessment
from .enum import Enum
from .enum import Enum


class ProbationInfo(object):
    _types = {
        "employment_id": str,
        "probation_id": str,
        "probation_start_date": str,
        "probation_expected_end_date": str,
        "actual_probation_end_date": str,
        "initiating_time": str,
        "submission_type": Enum,
        "initiator_id": str,
        "probation_status": Enum,
        "self_review": str,
        "notes": str,
        "process_id": str,
        "converted_via_bpm": bool,
        "custom_fields": List[CustomFieldData],
        "final_assessment_status": Enum,
        "final_assessment_result": Enum,
        "final_assessment_score": float,
        "final_assessment_grade": Enum,
        "final_assessment_comment": str,
        "final_assessment_detail": str,
        "assessments": List[Assessment],
        "probation_extend_expected_end_date": str,
        "extended_probation_period_duration": int,
        "extended_probation_period_unit": Enum,
        "probation_outcome": Enum,
    }

    def __init__(self, d=None):
        self.employment_id: Optional[str] = None
        self.probation_id: Optional[str] = None
        self.probation_start_date: Optional[str] = None
        self.probation_expected_end_date: Optional[str] = None
        self.actual_probation_end_date: Optional[str] = None
        self.initiating_time: Optional[str] = None
        self.submission_type: Optional[Enum] = None
        self.initiator_id: Optional[str] = None
        self.probation_status: Optional[Enum] = None
        self.self_review: Optional[str] = None
        self.notes: Optional[str] = None
        self.process_id: Optional[str] = None
        self.converted_via_bpm: Optional[bool] = None
        self.custom_fields: Optional[List[CustomFieldData]] = None
        self.final_assessment_status: Optional[Enum] = None
        self.final_assessment_result: Optional[Enum] = None
        self.final_assessment_score: Optional[float] = None
        self.final_assessment_grade: Optional[Enum] = None
        self.final_assessment_comment: Optional[str] = None
        self.final_assessment_detail: Optional[str] = None
        self.assessments: Optional[List[Assessment]] = None
        self.probation_extend_expected_end_date: Optional[str] = None
        self.extended_probation_period_duration: Optional[int] = None
        self.extended_probation_period_unit: Optional[Enum] = None
        self.probation_outcome: Optional[Enum] = None
        init(self, d, self._types)

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


class ProbationInfoBuilder(object):
    def __init__(self) -> None:
        self._probation_info = ProbationInfo()

    def employment_id(self, employment_id: str) -> "ProbationInfoBuilder":
        self._probation_info.employment_id = employment_id
        return self

    def probation_id(self, probation_id: str) -> "ProbationInfoBuilder":
        self._probation_info.probation_id = probation_id
        return self

    def probation_start_date(self, probation_start_date: str) -> "ProbationInfoBuilder":
        self._probation_info.probation_start_date = probation_start_date
        return self

    def probation_expected_end_date(self, probation_expected_end_date: str) -> "ProbationInfoBuilder":
        self._probation_info.probation_expected_end_date = probation_expected_end_date
        return self

    def actual_probation_end_date(self, actual_probation_end_date: str) -> "ProbationInfoBuilder":
        self._probation_info.actual_probation_end_date = actual_probation_end_date
        return self

    def initiating_time(self, initiating_time: str) -> "ProbationInfoBuilder":
        self._probation_info.initiating_time = initiating_time
        return self

    def submission_type(self, submission_type: Enum) -> "ProbationInfoBuilder":
        self._probation_info.submission_type = submission_type
        return self

    def initiator_id(self, initiator_id: str) -> "ProbationInfoBuilder":
        self._probation_info.initiator_id = initiator_id
        return self

    def probation_status(self, probation_status: Enum) -> "ProbationInfoBuilder":
        self._probation_info.probation_status = probation_status
        return self

    def self_review(self, self_review: str) -> "ProbationInfoBuilder":
        self._probation_info.self_review = self_review
        return self

    def notes(self, notes: str) -> "ProbationInfoBuilder":
        self._probation_info.notes = notes
        return self

    def process_id(self, process_id: str) -> "ProbationInfoBuilder":
        self._probation_info.process_id = process_id
        return self

    def converted_via_bpm(self, converted_via_bpm: bool) -> "ProbationInfoBuilder":
        self._probation_info.converted_via_bpm = converted_via_bpm
        return self

    def custom_fields(self, custom_fields: List[CustomFieldData]) -> "ProbationInfoBuilder":
        self._probation_info.custom_fields = custom_fields
        return self

    def final_assessment_status(self, final_assessment_status: Enum) -> "ProbationInfoBuilder":
        self._probation_info.final_assessment_status = final_assessment_status
        return self

    def final_assessment_result(self, final_assessment_result: Enum) -> "ProbationInfoBuilder":
        self._probation_info.final_assessment_result = final_assessment_result
        return self

    def final_assessment_score(self, final_assessment_score: float) -> "ProbationInfoBuilder":
        self._probation_info.final_assessment_score = final_assessment_score
        return self

    def final_assessment_grade(self, final_assessment_grade: Enum) -> "ProbationInfoBuilder":
        self._probation_info.final_assessment_grade = final_assessment_grade
        return self

    def final_assessment_comment(self, final_assessment_comment: str) -> "ProbationInfoBuilder":
        self._probation_info.final_assessment_comment = final_assessment_comment
        return self

    def final_assessment_detail(self, final_assessment_detail: str) -> "ProbationInfoBuilder":
        self._probation_info.final_assessment_detail = final_assessment_detail
        return self

    def assessments(self, assessments: List[Assessment]) -> "ProbationInfoBuilder":
        self._probation_info.assessments = assessments
        return self

    def probation_extend_expected_end_date(self, probation_extend_expected_end_date: str) -> "ProbationInfoBuilder":
        self._probation_info.probation_extend_expected_end_date = probation_extend_expected_end_date
        return self

    def extended_probation_period_duration(self, extended_probation_period_duration: int) -> "ProbationInfoBuilder":
        self._probation_info.extended_probation_period_duration = extended_probation_period_duration
        return self

    def extended_probation_period_unit(self, extended_probation_period_unit: Enum) -> "ProbationInfoBuilder":
        self._probation_info.extended_probation_period_unit = extended_probation_period_unit
        return self

    def probation_outcome(self, probation_outcome: Enum) -> "ProbationInfoBuilder":
        self._probation_info.probation_outcome = probation_outcome
        return self

    def build(self) -> "ProbationInfo":
        return self._probation_info
