lsp

python-client/windmill-api/windmill_api/models/flow_module.py

@@ -0,0 +1,215 @@
+from typing import Any, Dict, List, Type, TypeVar, Union
+
+import attr
+
+from ..models.flow_module_input_transforms import FlowModuleInputTransforms
+from ..models.flow_module_retry import FlowModuleRetry
+from ..models.flow_module_sleep_type_0 import FlowModuleSleepType0
+from ..models.flow_module_sleep_type_1 import FlowModuleSleepType1
+from ..models.flow_module_stop_after_if import FlowModuleStopAfterIf
+from ..models.flow_module_value_type_0 import FlowModuleValueType0
+from ..models.flow_module_value_type_1 import FlowModuleValueType1
+from ..models.flow_module_value_type_2 import FlowModuleValueType2
+from ..models.flow_module_value_type_3 import FlowModuleValueType3
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="FlowModule")
+
+
+@attr.s(auto_attribs=True)
+class FlowModule:
+    """
+    Attributes:
+        input_transforms (FlowModuleInputTransforms):
+        value (Union[FlowModuleValueType0, FlowModuleValueType1, FlowModuleValueType2, FlowModuleValueType3]):
+        stop_after_if (Union[Unset, FlowModuleStopAfterIf]):
+        sleep (Union[FlowModuleSleepType0, FlowModuleSleepType1, Unset]):
+        summary (Union[Unset, str]):
+        suspend (Union[Unset, int]):
+        retry (Union[Unset, FlowModuleRetry]):
+    """
+
+    input_transforms: FlowModuleInputTransforms
+    value: Union[FlowModuleValueType0, FlowModuleValueType1, FlowModuleValueType2, FlowModuleValueType3]
+    stop_after_if: Union[Unset, FlowModuleStopAfterIf] = UNSET
+    sleep: Union[FlowModuleSleepType0, FlowModuleSleepType1, Unset] = UNSET
+    summary: Union[Unset, str] = UNSET
+    suspend: Union[Unset, int] = UNSET
+    retry: Union[Unset, FlowModuleRetry] = UNSET
+    additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        input_transforms = self.input_transforms.to_dict()
+
+        if isinstance(self.value, FlowModuleValueType0):
+            value = self.value.to_dict()
+
+        elif isinstance(self.value, FlowModuleValueType1):
+            value = self.value.to_dict()
+
+        elif isinstance(self.value, FlowModuleValueType2):
+            value = self.value.to_dict()
+
+        else:
+            value = self.value.to_dict()
+
+        stop_after_if: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.stop_after_if, Unset):
+            stop_after_if = self.stop_after_if.to_dict()
+
+        sleep: Union[Dict[str, Any], Unset]
+        if isinstance(self.sleep, Unset):
+            sleep = UNSET
+
+        elif isinstance(self.sleep, FlowModuleSleepType0):
+            sleep = UNSET
+            if not isinstance(self.sleep, Unset):
+                sleep = self.sleep.to_dict()
+
+        else:
+            sleep = UNSET
+            if not isinstance(self.sleep, Unset):
+                sleep = self.sleep.to_dict()
+
+        summary = self.summary
+        suspend = self.suspend
+        retry: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.retry, Unset):
+            retry = self.retry.to_dict()
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "input_transforms": input_transforms,
+                "value": value,
+            }
+        )
+        if stop_after_if is not UNSET:
+            field_dict["stop_after_if"] = stop_after_if
+        if sleep is not UNSET:
+            field_dict["sleep"] = sleep
+        if summary is not UNSET:
+            field_dict["summary"] = summary
+        if suspend is not UNSET:
+            field_dict["suspend"] = suspend
+        if retry is not UNSET:
+            field_dict["retry"] = retry
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        input_transforms = FlowModuleInputTransforms.from_dict(d.pop("input_transforms"))
+
+        def _parse_value(
+            data: object,
+        ) -> Union[FlowModuleValueType0, FlowModuleValueType1, FlowModuleValueType2, FlowModuleValueType3]:
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                value_type_0 = FlowModuleValueType0.from_dict(data)
+
+                return value_type_0
+            except:  # noqa: E722
+                pass
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                value_type_1 = FlowModuleValueType1.from_dict(data)
+
+                return value_type_1
+            except:  # noqa: E722
+                pass
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                value_type_2 = FlowModuleValueType2.from_dict(data)
+
+                return value_type_2
+            except:  # noqa: E722
+                pass
+            if not isinstance(data, dict):
+                raise TypeError()
+            value_type_3 = FlowModuleValueType3.from_dict(data)
+
+            return value_type_3
+
+        value = _parse_value(d.pop("value"))
+
+        _stop_after_if = d.pop("stop_after_if", UNSET)
+        stop_after_if: Union[Unset, FlowModuleStopAfterIf]
+        if isinstance(_stop_after_if, Unset):
+            stop_after_if = UNSET
+        else:
+            stop_after_if = FlowModuleStopAfterIf.from_dict(_stop_after_if)
+
+        def _parse_sleep(data: object) -> Union[FlowModuleSleepType0, FlowModuleSleepType1, Unset]:
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                _sleep_type_0 = data
+                sleep_type_0: Union[Unset, FlowModuleSleepType0]
+                if isinstance(_sleep_type_0, Unset):
+                    sleep_type_0 = UNSET
+                else:
+                    sleep_type_0 = FlowModuleSleepType0.from_dict(_sleep_type_0)
+
+                return sleep_type_0
+            except:  # noqa: E722
+                pass
+            if not isinstance(data, dict):
+                raise TypeError()
+            _sleep_type_1 = data
+            sleep_type_1: Union[Unset, FlowModuleSleepType1]
+            if isinstance(_sleep_type_1, Unset):
+                sleep_type_1 = UNSET
+            else:
+                sleep_type_1 = FlowModuleSleepType1.from_dict(_sleep_type_1)
+
+            return sleep_type_1
+
+        sleep = _parse_sleep(d.pop("sleep", UNSET))
+
+        summary = d.pop("summary", UNSET)
+
+        suspend = d.pop("suspend", UNSET)
+
+        _retry = d.pop("retry", UNSET)
+        retry: Union[Unset, FlowModuleRetry]
+        if isinstance(_retry, Unset):
+            retry = UNSET
+        else:
+            retry = FlowModuleRetry.from_dict(_retry)
+
+        flow_module = cls(
+            input_transforms=input_transforms,
+            value=value,
+            stop_after_if=stop_after_if,
+            sleep=sleep,
+            summary=summary,
+            suspend=suspend,
+            retry=retry,
+        )
+
+        flow_module.additional_properties = d
+        return flow_module
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties