# Code generated by Lark OpenAPI.

from typing import Optional

from lark_oapi.core.construct import init
from .shift import Shift


class CreateShiftResponseBody(object):
    _types = {
        "shift": Shift,
    }

    def __init__(self, d=None):
        self.shift: Optional[Shift] = None
        init(self, d, self._types)

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


class CreateShiftResponseBodyBuilder(object):
    def __init__(self) -> None:
        self._create_shift_response_body = CreateShiftResponseBody()

    def shift(self, shift: Shift) -> "CreateShiftResponseBodyBuilder":
        self._create_shift_response_body.shift = shift
        return self

    def build(self) -> "CreateShiftResponseBody":
        return self._create_shift_response_body
