# 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 CompletionCondition(object):
    _types = {
        "condition_type": int,
        "duration": int,
        "playback_duration": int,
    }

    def __init__(self, d=None):
        self.condition_type: Optional[int] = None
        self.duration: Optional[int] = None
        self.playback_duration: Optional[int] = None
        init(self, d, self._types)

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


class CompletionConditionBuilder(object):
    def __init__(self) -> None:
        self._completion_condition = CompletionCondition()

    def condition_type(self, condition_type: int) -> "CompletionConditionBuilder":
        self._completion_condition.condition_type = condition_type
        return self

    def duration(self, duration: int) -> "CompletionConditionBuilder":
        self._completion_condition.duration = duration
        return self

    def playback_duration(self, playback_duration: int) -> "CompletionConditionBuilder":
        self._completion_condition.playback_duration = playback_duration
        return self

    def build(self) -> "CompletionCondition":
        return self._completion_condition
