|
7 | 7 | from serverlessworkflow.sdk.end import End
|
8 | 8 | from serverlessworkflow.sdk.error import Error
|
9 | 9 | from serverlessworkflow.sdk.event_data_filter import EventDataFilter
|
10 |
| -from serverlessworkflow.sdk.hydration import ComplexTypeOf, ArrayTypeOf, HydratableParameter, SimpleTypeOf, \ |
11 |
| - UnionTypeOf, Fields |
12 | 10 | from serverlessworkflow.sdk.metadata import Metadata
|
13 | 11 | from serverlessworkflow.sdk.state import State
|
14 | 12 | from serverlessworkflow.sdk.state_data_filter import StateDataFilter
|
| 13 | +from serverlessworkflow.sdk.swf_base import ComplexTypeOf, ArrayTypeOf, HydratableParameter, SimpleTypeOf, \ |
| 14 | + UnionTypeOf, SwfBase |
15 | 15 | from serverlessworkflow.sdk.transition import Transition
|
16 | 16 |
|
17 | 17 |
|
18 |
| -class CallbackState(State): |
| 18 | +class CallbackState(State, SwfBase): |
19 | 19 | id: str = None
|
20 | 20 | name: str = None
|
21 | 21 | type: str = None
|
@@ -48,11 +48,11 @@ def __init__(self,
|
48 | 48 | metadata: Metadata = None,
|
49 | 49 | **kwargs):
|
50 | 50 |
|
51 |
| - Fields(locals(), kwargs, Fields.f_hydration).set_to_object(self) |
| 51 | + _default_values = {'type': 'callback', 'usedForCompensation': False, } |
| 52 | + SwfBase.__init__(self, locals(), kwargs, SwfBase.default_hydration, _default_values) |
52 | 53 |
|
53 | 54 | @staticmethod
|
54 | 55 | def f_hydration(p_key, p_value):
|
55 |
| - |
56 | 56 | if p_key == 'action':
|
57 | 57 | return HydratableParameter(value=p_value).hydrateAs(ComplexTypeOf(Action))
|
58 | 58 |
|
|
0 commit comments