# 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 .agency_protection import AgencyProtection
from .agency_protection import AgencyProtection


class ProtectSearchAgencyResponseBody(object):
    _types = {
        "is_onboarded": bool,
        "onboarded_in_protection": bool,
        "onboarded_protection": AgencyProtection,
        "protection_list": List[AgencyProtection],
    }

    def __init__(self, d=None):
        self.is_onboarded: Optional[bool] = None
        self.onboarded_in_protection: Optional[bool] = None
        self.onboarded_protection: Optional[AgencyProtection] = None
        self.protection_list: Optional[List[AgencyProtection]] = None
        init(self, d, self._types)

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


class ProtectSearchAgencyResponseBodyBuilder(object):
    def __init__(self) -> None:
        self._protect_search_agency_response_body = ProtectSearchAgencyResponseBody()

    def is_onboarded(self, is_onboarded: bool) -> "ProtectSearchAgencyResponseBodyBuilder":
        self._protect_search_agency_response_body.is_onboarded = is_onboarded
        return self

    def onboarded_in_protection(self, onboarded_in_protection: bool) -> "ProtectSearchAgencyResponseBodyBuilder":
        self._protect_search_agency_response_body.onboarded_in_protection = onboarded_in_protection
        return self

    def onboarded_protection(self, onboarded_protection: AgencyProtection) -> "ProtectSearchAgencyResponseBodyBuilder":
        self._protect_search_agency_response_body.onboarded_protection = onboarded_protection
        return self

    def protection_list(self, protection_list: List[AgencyProtection]) -> "ProtectSearchAgencyResponseBodyBuilder":
        self._protect_search_agency_response_body.protection_list = protection_list
        return self

    def build(self) -> "ProtectSearchAgencyResponseBody":
        return self._protect_search_agency_response_body
