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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
|
"""AAI Element module."""
# Copyright 2022 Orange, Deutsche Telekom AG
#
# 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.
from dataclasses import dataclass, field
from typing import Dict, Iterator, List, Optional
from onapsdk.configuration import settings
from onapsdk.onap_service import OnapService
from onapsdk.utils.headers_creator import headers_aai_creator
from onapsdk.utils.jinja import jinja_env
from onapsdk.utils.gui import GuiItem, GuiList
from onapsdk.exceptions import RelationshipNotFound, ResourceNotFound
@dataclass
class Relationship:
"""Relationship class.
A&AI elements could have relationship with other A&AI elements.
Relationships are represented by this class objects.
"""
related_to: str
related_link: str
relationship_data: List[Dict[str, str]]
relationship_label: str = ""
related_to_property: List[Dict[str, str]] = field(default_factory=list)
def get_relationship_data(self, relationship_key: str) -> Optional[str]:
"""Get relationship data for given relationship key.
From list of relationship data get the value for
given key
Args:
relationship_key (str): Key to get relationship data value
Returns:
Optional[str]: Relationship value or None if relationship data
with provided ket doesn't exist
"""
for data in self.relationship_data:
if data["relationship-key"] == relationship_key:
return data["relationship-value"]
return None
class AaiElement(OnapService):
"""Mother Class of all A&AI elements."""
name: str = "AAI"
server: str = "AAI"
base_url = settings.AAI_URL
api_version = "/aai/" + settings.AAI_API_VERSION
headers = headers_aai_creator(OnapService.headers)
@classmethod
def get_guis(cls) -> GuiItem:
"""Retrieve the status of the AAI GUIs.
Only one GUI is referenced for AAI
the AAI sparky GUI
Return the list of GUIs
"""
gui_url = settings.AAI_GUI_SERVICE
aai_gui_response = cls.send_message(
"GET", "Get AAI GUI Status", gui_url)
guilist = GuiList([])
guilist.add(GuiItem(
gui_url,
aai_gui_response.status_code))
return guilist
class AaiResource(AaiElement):
"""A&AI resource class."""
@classmethod
def filter_none_key_values(cls, dict_to_filter: Dict[str, Optional[str]]) -> Dict[str, str]:
"""Filter out None key values from dictionary.
Iterate through given dictionary and filter None values.
Args:
dict_to_filter (Dict): Dictionary to filter out None
Returns:dataclasse init a field
Dict[str, str]: Filtered dictionary
"""
return dict(
filter(lambda key_value_tuple: key_value_tuple[1] is not None, dict_to_filter.items(),)
)
@property
def url(self) -> str:
"""Resource's url.
Returns:
str: Resource's url
"""
raise NotImplementedError
@property
def relationships(self) -> Iterator[Relationship]:
"""Resource relationships iterator.
Yields:
Relationship: resource relationship
Raises:
RelationshipNotFound: if request for relationships returned 404
"""
try:
generator = self.send_message_json("GET",
"Get object relationships",
f"{self.url}/relationship-list")\
.get("relationship", [])
for relationship in generator:
yield Relationship(
related_to=relationship.get("related-to"),
relationship_label=relationship.get("relationship-label"),
related_link=relationship.get("related-link"),
relationship_data=relationship.get("relationship-data"),
)
except ResourceNotFound as exc:
self._logger.error("Getting object relationships failed: %s", exc)
msg = (f'{self.name} relationships not found.'
f'Server: {self.server}. Url: {self.url}')
raise RelationshipNotFound(msg) from exc
@classmethod
def get_all_url(cls, *args, **kwargs) -> str:
"""Return an url for all objects of given class.
Returns:
str: URL to get all objects of given class
"""
raise NotImplementedError
@classmethod
def count(cls, *args, **kwargs) -> int:
"""Get the count number of all objects of given class.
Get the response, iterate through response (each class has different response)
-- the first key value is the count.
Returns:
int: Count of the objects
"""
return next(iter(cls.send_message_json(
"GET",
f"Get count of {cls.__name__} class instances",
f"{cls.get_all_url(*args, **kwargs)}?format=count"
)["results"][0].values()))
def add_relationship(self, relationship: Relationship) -> None:
"""Add relationship to aai resource.
Add relationship to resource using A&AI API
Args:
relationship (Relationship): Relationship to add
"""
self.send_message(
"PUT",
f"add relationship to {self.__class__.__name__}",
f"{self.url}/relationship-list/relationship",
data=jinja_env()
.get_template("aai_add_relationship.json.j2")
.render(relationship=relationship),
)
|
