# Code generated by Lark OpenAPI.

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


class SecurityLevel(object):
    _types = {
        "is_risk": bool,
        "risk_banner_level": str,
        "risk_banner_reason": str,
        "is_header_from_external": bool,
        "via_domain": str,
        "spam_banner_type": str,
        "spam_user_rule_id": str,
        "spam_banner_info": str,
    }

    def __init__(self, d=None):
        self.is_risk: Optional[bool] = None
        self.risk_banner_level: Optional[str] = None
        self.risk_banner_reason: Optional[str] = None
        self.is_header_from_external: Optional[bool] = None
        self.via_domain: Optional[str] = None
        self.spam_banner_type: Optional[str] = None
        self.spam_user_rule_id: Optional[str] = None
        self.spam_banner_info: Optional[str] = None
        init(self, d, self._types)

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


class SecurityLevelBuilder(object):
    def __init__(self) -> None:
        self._security_level = SecurityLevel()

    def is_risk(self, is_risk: bool) -> "SecurityLevelBuilder":
        self._security_level.is_risk = is_risk
        return self

    def risk_banner_level(self, risk_banner_level: str) -> "SecurityLevelBuilder":
        self._security_level.risk_banner_level = risk_banner_level
        return self

    def risk_banner_reason(self, risk_banner_reason: str) -> "SecurityLevelBuilder":
        self._security_level.risk_banner_reason = risk_banner_reason
        return self

    def is_header_from_external(self, is_header_from_external: bool) -> "SecurityLevelBuilder":
        self._security_level.is_header_from_external = is_header_from_external
        return self

    def via_domain(self, via_domain: str) -> "SecurityLevelBuilder":
        self._security_level.via_domain = via_domain
        return self

    def spam_banner_type(self, spam_banner_type: str) -> "SecurityLevelBuilder":
        self._security_level.spam_banner_type = spam_banner_type
        return self

    def spam_user_rule_id(self, spam_user_rule_id: str) -> "SecurityLevelBuilder":
        self._security_level.spam_user_rule_id = spam_user_rule_id
        return self

    def spam_banner_info(self, spam_banner_info: str) -> "SecurityLevelBuilder":
        self._security_level.spam_banner_info = spam_banner_info
        return self

    def build(self) -> "SecurityLevel":
        return self._security_level