serverlessworkflow · tsurdilo · Feb 9, 2022 · Feb 7, 2022
diff --git a/serverlessworkflow/sdk/action.py b/serverlessworkflow/sdk/action.py
@@ -1,28 +1,37 @@
+from typing import Union
+
+from serverlessworkflow.sdk.action_data_filter import ActionDataFilter
+from serverlessworkflow.sdk.event_ref import EventRef
+from serverlessworkflow.sdk.function_ref import FunctionRef
+from serverlessworkflow.sdk.sleep import Sleep
+from serverlessworkflow.sdk.sub_flow_ref import SubFlowRef
+
+
 class Action:
-    id = None
-    name = None
-    functionRef = None
-    eventRef = None
-    subFlowRef = None
-    sleep = None
-    retryRef = None
-    nonRetryableErrors = None
-    retryableErrors = None
-    actionDataFilter = None
-    condition = None
+    id: str = None
+    name: str = None
+    functionRef: Union[str, FunctionRef] = None
+    eventRef: EventRef = None
+    subFlowRef: Union[str, SubFlowRef] = None
+    sleep: Sleep = None
+    retryRef: str = None
+    nonRetryableErrors: [str] = None
+    retryableErrors: [str] = None
+    actionDataFilter: ActionDataFilter = None
+    condition: str = None
 
     def __init__(self,
-                 id=None,
-                 name=None,
-                 functionRef=None,
-                 eventRef=None,
-                 subFlowRef=None,
-                 sleep=None,
-                 retryRef=None,
-                 nonRetryableErrors=None,
-                 retryableErrors=None,
-                 actionDataFilter=None,
-                 condition=None,
+                 id: str = None,
+                 name: str = None,
+                 functionRef: Union[str, FunctionRef] = None,
+                 eventRef: EventRef = None,
+                 subFlowRef: Union[str, SubFlowRef] = None,
+                 sleep: Sleep = None,
+                 retryRef: str = None,
+                 nonRetryableErrors: [str] = None,
+                 retryableErrors: [str] = None,
+                 actionDataFilter: ActionDataFilter = None,
+                 condition: str = None,
                  **kwargs):
 
         # duplicated

diff --git a/serverlessworkflow/sdk/action_data_filter.py b/serverlessworkflow/sdk/action_data_filter.py
@@ -0,0 +1,34 @@
+class ActionDataFilter:
+    fromStateData: str = None
+    useResults: bool = None
+    results: str = None
+    toStateData: str = None
+
+    def __init__(self,
+                 fromStateData: str = None,
+                 useResults: bool = None,
+                 results: str = None,
+                 toStateData: str = None,
+                 **kwargs):
+
+        # duplicated
+        for local in list(locals()):
+            if local in ["self", "kwargs"]:
+                continue
+            value = locals().get(local)
+            if not value:
+                continue
+            if value == "true":
+                value = True
+            # duplicated
+
+            self.__setattr__(local.replace("_", ""), value)
+
+        # duplicated
+        for k in kwargs.keys():
+            value = kwargs[k]
+            if value == "true":
+                value = True
+
+            self.__setattr__(k.replace("_", ""), value)
+            # duplicated
diff --git a/serverlessworkflow/sdk/eventdatafilter.py → serverlessworkflow/sdk/auth_def.py b/serverlessworkflow/sdk/eventdatafilter.py → serverlessworkflow/sdk/auth_def.py
@@ -1,12 +1,26 @@
-class Eventdatafilter:
-    useData = None
-    data = None
-    toStateData = None
+from enum import Enum
+from typing import Union
+
+from serverlessworkflow.sdk.basic_props_def import BasicPropsDef
+from serverlessworkflow.sdk.bearer_props_def import BearerPropsDef
+from serverlessworkflow.sdk.oauth2props_def import Oauth2PropsDef
+
+
+class Scheme(Enum):
+    BASIC = "basic"
+    BEARER = "bearer"
+    OAUTH2 = "oauth2"
+
+
+class AuthDef:
+    name: str = None
+    scheme: Scheme = None
+    properties: Union[str, Union[BasicPropsDef, BearerPropsDef, Oauth2PropsDef]] = None
 
     def __init__(self,
-                 useData=None,
-                 data=None,
-                 toStateData=None,
+                 name: str = None,
+                 scheme: Scheme = None,
+                 properties: Union[str, Union[BasicPropsDef, BearerPropsDef, Oauth2PropsDef]] = None,
                  **kwargs):
 
         # duplicated

diff --git a/serverlessworkflow/sdk/basic_props_def.py b/serverlessworkflow/sdk/basic_props_def.py
@@ -0,0 +1,35 @@
+from serverlessworkflow.sdk.metadata import Metadata
+
+
+class BasicPropsDef:
+    username: str = None
+    password: str = None
+    metadata: Metadata = None
+
+    def __init__(self,
+                 username: str = None,
+                 password: str = None,
+                 metadata: Metadata = None,
+                 **kwargs):
+
+        # duplicated
+        for local in list(locals()):
+            if local in ["self", "kwargs"]:
+                continue
+            value = locals().get(local)
+            if not value:
+                continue
+            if value == "true":
+                value = True
+            # duplicated
+
+            self.__setattr__(local.replace("_", ""), value)
+
+        # duplicated
+        for k in kwargs.keys():
+            value = kwargs[k]
+            if value == "true":
+                value = True
+
+            self.__setattr__(k.replace("_", ""), value)
+            # duplicated
diff --git a/serverlessworkflow/sdk/basicpropsdef.py → serverlessworkflow/sdk/bearer_props_def.py b/serverlessworkflow/sdk/basicpropsdef.py → serverlessworkflow/sdk/bearer_props_def.py
@@ -1,12 +1,13 @@
-class Basicpropsdef:
-    username = None
-    password = None
-    metadata = None
+from serverlessworkflow.sdk.metadata import Metadata
+
+
+class BearerPropsDef:
+    token: str = None
+    metadata: Metadata = None
 
     def __init__(self,
-                 username=None,
-                 password=None,
-                 metadata=None,
+                 token: str = None,
+                 metadata: Metadata = None,
                  **kwargs):
 
         # duplicated

diff --git a/serverlessworkflow/sdk/branch .py → serverlessworkflow/sdk/branch.py b/serverlessworkflow/sdk/branch .py → serverlessworkflow/sdk/branch.py
@@ -1,16 +1,16 @@
+from serverlessworkflow.sdk.action import Action
+from serverlessworkflow.sdk.branch_timeout import BranchTimeOut
+
+
 class Branch:
-    name = None
-    timeouts = None
-    actionExecTimeout = None
-    branchExecTimeout = None
-    actions = None
+    name: str = None
+    timeouts: BranchTimeOut = None
+    actions: [Action] = None
 
     def __init__(self,
-                 name=None,
-                 timeouts=None,
-                 actionExecTimeout=None,
-                 branchExecTimeout=None,
-                 actions=None,
+                 name: str = None,
+                 timeouts: BranchTimeOut = None,
+                 actions: [Action] = None,
                  **kwargs):
 
         # duplicated

diff --git a/serverlessworkflow/sdk/branch_timeout.py b/serverlessworkflow/sdk/branch_timeout.py
@@ -0,0 +1,3 @@
+class BranchTimeOut:
+    actionExecTimeOut: str = None  # ActionExecTimeOut
+    branchExecTimeOut: str = None  # BranchExecTimeOut
diff --git a/serverlessworkflow/sdk/callback_state.py b/serverlessworkflow/sdk/callback_state.py
@@ -0,0 +1,66 @@
+from typing import Union
+
+from serverlessworkflow.sdk.action import Action
+from serverlessworkflow.sdk.callback_state_timeout import CallbackStateTimeOut
+from serverlessworkflow.sdk.end import End
+from serverlessworkflow.sdk.error import Error
+from serverlessworkflow.sdk.event_data_filter import EventDataFilter
+from serverlessworkflow.sdk.metadata import Metadata
+from serverlessworkflow.sdk.state_data_filter import StateDataFilter
+from serverlessworkflow.sdk.transition import Transition
+
+
+class CallbackState:
+    id: str = None
+    name: str = None
+    type: 'callback' = None
+    action: Action = None
+    eventRef: str = None
+    timeouts: CallbackStateTimeOut = None
+    eventDataFilter: EventDataFilter = None
+    stateDataFilter: StateDataFilter = None
+    onErrors: [Error] = None
+    transition: Union[str, Transition] = None
+    end: Union[bool, End] = None
+    compensatedBy: str = None
+    usedForCompensation: bool = None
+    metadata: Metadata = None
+
+    def __init__(self,
+                 id: str = None,
+                 name: str = None,
+                 type: 'callback' = None,
+                 action: Action = None,
+                 eventRef: str = None,
+                 timeouts: CallbackStateTimeOut = None,
+                 eventDataFilter: EventDataFilter = None,
+                 stateDataFilter: StateDataFilter = None,
+                 onErrors: [Error] = None,
+                 transition: Union[str, Transition] = None,
+                 end: Union[bool, End] = None,
+                 compensatedBy: str = None,
+                 usedForCompensation: bool = None,
+                 metadata: Metadata = None,
+                 **kwargs):
+        # duplicated
+        for local in list(locals()):
+            if local in ["self", "kwargs"]:
+                continue
+            value = locals().get(local)
+            if not value:
+                continue
+            if value == "true":
+                value = True
+            # duplicated
+
+            self.__setattr__(local.replace("_", ""), value)
+
+            # duplicated
+
+        for k in kwargs.keys():
+            value = kwargs[k]
+            if value == "true":
+                value = True
+
+            self.__setattr__(k.replace("_", ""), value)
+            # duplicated
diff --git a/serverlessworkflow/sdk/callback_state_timeout.py b/serverlessworkflow/sdk/callback_state_timeout.py
@@ -0,0 +1,7 @@
+from serverlessworkflow.sdk.state_exec_timeout import StateExecTimeOut
+
+
+class CallbackStateTimeOut:
+    stateExecTimeOut: StateExecTimeOut = None
+    actionExecTimeOut: str = None  # ActionExecTimeOut
+    eventTimeOut: str = None  # EventTimeOut
diff --git a/serverlessworkflow/sdk/callbackstate.py b/serverlessworkflow/sdk/callbackstate.py
diff --git a/serverlessworkflow/sdk/continue_as_def.py b/serverlessworkflow/sdk/continue_as_def.py
@@ -0,0 +1,39 @@
+from typing import Union, Dict
+
+from serverlessworkflow.sdk.workflow_exec_timeout import WorkflowExecTimeOut
+
+
+class ContinueAsDef:
+    workflowId: str = None
+    version: str = None
+    data: Union[str, Dict[str, Dict]] = None
+    workflowExecTimeOut: WorkflowExecTimeOut = None
+
+    def __init__(self,
+                 workflowId: str = None,
+                 version: str = None,
+                 data: Union[str, Dict[str, Dict]] = None,
+                 workflowExecTimeOut: WorkflowExecTimeOut = None,
+                 **kwargs):
+
+        # duplicated
+        for local in list(locals()):
+            if local in ["self", "kwargs"]:
+                continue
+            value = locals().get(local)
+            if not value:
+                continue
+            if value == "true":
+                value = True
+            # duplicated
+
+            self.__setattr__(local.replace("_", ""), value)
+
+        # duplicated
+        for k in kwargs.keys():
+            value = kwargs[k]
+            if value == "true":
+                value = True
+
+            self.__setattr__(k.replace("_", ""), value)
+            # duplicated