1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import warnings
from mistral_lib import serialization
class ActionContext(serialization.MistralSerializable):
def __init__(self, security_ctx=None, execution_ctx=None):
self.security = security_ctx
self.execution = execution_ctx
def _deprecation_warning(self, name):
warnings.warn(
"context.{0} is deprecated from the context passed to actions. "
"Please use context.security.{0}. It will be removed in a future "
"release.", DeprecationWarning
)
def __getattribute__(self, name):
deprecated = [
"auth_cacert", "auth_token", "auth_uri", "expires_at", "insecure",
"is_target", "is_trust_scoped", "project_id", "project_name",
"redelivered", "region_name", "service_catalog", "trust_id",
"user_name"
]
if name in deprecated:
self._deprecation_warning(name)
return getattr(self.security, name)
return super(ActionContext, self).__getattribute__(name)
class SecurityContext(object):
def __init__(self, auth_uri=None, auth_cacert=None, insecure=None,
service_catalog=None, region_name=None, is_trust_scoped=None,
redelivered=None, expires_at=None, trust_id=None,
is_target=None, project_id=None, project_name=None,
user_name=None, auth_token=None):
self.auth_uri = auth_uri
self.auth_cacert = auth_cacert
self.insecure = insecure
self.service_catalog = service_catalog
self.region_name = region_name
self.is_trust_scoped = is_trust_scoped
self.redelivered = redelivered
self.expires_at = expires_at
self.trust_id = trust_id
self.is_target = is_target
self.project_id = project_id
self.project_name = project_name
self.user_name = user_name
self.auth_token = auth_token
class ExecutionContext(object):
def __init__(self, workflow_execution_id=None, task_execution_id=None,
action_execution_id=None, workflow_name=None,
callback_url=None, task_id=None, with_items_index=0,
task_rerun_no=0, task_rerun_id=None,
workflow_propagated_headers=None):
self.workflow_execution_id = workflow_execution_id
self.task_execution_id = task_execution_id
self.action_execution_id = action_execution_id
self.workflow_name = workflow_name
self.callback_url = callback_url
self.with_items_index = with_items_index
self.task_rerun_no = task_rerun_no
self.task_rerun_id = task_rerun_id
self.workflow_propagated_headers = workflow_propagated_headers
if task_id is not None:
self.task_execution_id = task_id
self._deprecate_task_id_warning()
def _deprecate_task_id_warning(self):
warnings.warn(
"context.execution.task_id was deprecated in the Queens cycle. "
"Please use context.execution.task_execution_id. It will be "
"removed in a future release.", DeprecationWarning
)
@property
def task_id(self):
self._deprecate_task_id_warning()
return self.task_execution_id
class ActionContextSerializer(serialization.DictBasedSerializer):
def serialize_to_dict(self, entity):
return {
'security': vars(entity.security),
'execution': vars(entity.execution),
}
def deserialize_from_dict(self, entity_dict):
return ActionContext(
security_ctx=SecurityContext(**entity_dict['security']),
execution_ctx=ExecutionContext(**entity_dict['execution'])
)
serialization.register_serializer(ActionContext, ActionContextSerializer())
|
