# 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 .field_permission_access import FieldPermissionAccess


class DataModelPermissionAccess(object):
    _types = {
        "api_id": str,
        "api_name": str,
        "allowed_access": List[int],
        "fields": List[FieldPermissionAccess],
        "data_model_type": str,
        "display_name": Dict[str, str],
    }

    def __init__(self, d=None):
        self.api_id: Optional[str] = None
        self.api_name: Optional[str] = None
        self.allowed_access: Optional[List[int]] = None
        self.fields: Optional[List[FieldPermissionAccess]] = None
        self.data_model_type: Optional[str] = None
        self.display_name: Optional[Dict[str, str]] = None
        init(self, d, self._types)

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


class DataModelPermissionAccessBuilder(object):
    def __init__(self) -> None:
        self._data_model_permission_access = DataModelPermissionAccess()

    def api_id(self, api_id: str) -> "DataModelPermissionAccessBuilder":
        self._data_model_permission_access.api_id = api_id
        return self

    def api_name(self, api_name: str) -> "DataModelPermissionAccessBuilder":
        self._data_model_permission_access.api_name = api_name
        return self

    def allowed_access(self, allowed_access: List[int]) -> "DataModelPermissionAccessBuilder":
        self._data_model_permission_access.allowed_access = allowed_access
        return self

    def fields(self, fields: List[FieldPermissionAccess]) -> "DataModelPermissionAccessBuilder":
        self._data_model_permission_access.fields = fields
        return self

    def data_model_type(self, data_model_type: str) -> "DataModelPermissionAccessBuilder":
        self._data_model_permission_access.data_model_type = data_model_type
        return self

    def display_name(self, display_name: Dict[str, str]) -> "DataModelPermissionAccessBuilder":
        self._data_model_permission_access.display_name = display_name
        return self

    def build(self) -> "DataModelPermissionAccess":
        return self._data_model_permission_access