|
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