# 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 .vault_task_file import VaultTaskFile
from .user import User


class VaultTask(object):
    _types = {
        "task_id": int,
        "name": str,
        "create_time": str,
        "update_time": str,
        "status": str,
        "fail_reason": str,
        "expire_time": str,
        "extract_key": str,
        "files": List[VaultTaskFile],
        "creator": User,
    }

    def __init__(self, d=None):
        self.task_id: Optional[int] = None
        self.name: Optional[str] = None
        self.create_time: Optional[str] = None
        self.update_time: Optional[str] = None
        self.status: Optional[str] = None
        self.fail_reason: Optional[str] = None
        self.expire_time: Optional[str] = None
        self.extract_key: Optional[str] = None
        self.files: Optional[List[VaultTaskFile]] = None
        self.creator: Optional[User] = None
        init(self, d, self._types)

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


class VaultTaskBuilder(object):
    def __init__(self) -> None:
        self._vault_task = VaultTask()

    def task_id(self, task_id: int) -> "VaultTaskBuilder":
        self._vault_task.task_id = task_id
        return self

    def name(self, name: str) -> "VaultTaskBuilder":
        self._vault_task.name = name
        return self

    def create_time(self, create_time: str) -> "VaultTaskBuilder":
        self._vault_task.create_time = create_time
        return self

    def update_time(self, update_time: str) -> "VaultTaskBuilder":
        self._vault_task.update_time = update_time
        return self

    def status(self, status: str) -> "VaultTaskBuilder":
        self._vault_task.status = status
        return self

    def fail_reason(self, fail_reason: str) -> "VaultTaskBuilder":
        self._vault_task.fail_reason = fail_reason
        return self

    def expire_time(self, expire_time: str) -> "VaultTaskBuilder":
        self._vault_task.expire_time = expire_time
        return self

    def extract_key(self, extract_key: str) -> "VaultTaskBuilder":
        self._vault_task.extract_key = extract_key
        return self

    def files(self, files: List[VaultTaskFile]) -> "VaultTaskBuilder":
        self._vault_task.files = files
        return self

    def creator(self, creator: User) -> "VaultTaskBuilder":
        self._vault_task.creator = creator
        return self

    def build(self) -> "VaultTask":
        return self._vault_task