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
|
# ============LICENSE_START===================================================
# Copyright (C) 2020 Nordix Foundation.
# ============================================================================
# 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.
#
# SPDX-License-Identifier: Apache-2.0
# ============LICENSE_END=====================================================
import re
from enum import Enum
from mod import pmsh_logging as logger, db
from mod.db_models import NetworkFunctionModel
class NetworkFunction:
def __init__(self, **kwargs):
""" Object representation of the NetworkFunction. """
self.nf_name = kwargs.get('nf_name')
self.orchestration_status = kwargs.get('orchestration_status')
@classmethod
def nf_def(cls):
return cls(nf_name=None, orchestration_status=None)
def __str__(self):
return f'nf-name: {self.nf_name}, orchestration-status: {self.orchestration_status}'
def __eq__(self, other):
return self.nf_name == other.nf_name and \
self.orchestration_status == other.orchestration_status
def __hash__(self):
return hash((self.nf_name, self.orchestration_status))
def create(self):
""" Creates a NetworkFunction database entry """
existing_nf = NetworkFunctionModel.query.filter(
NetworkFunctionModel.nf_name == self.nf_name).one_or_none()
if existing_nf is None:
new_nf = NetworkFunctionModel(nf_name=self.nf_name,
orchestration_status=self.orchestration_status)
db.session.add(new_nf)
db.session.commit()
return new_nf
else:
logger.debug(f'Network function {existing_nf} already exists,'
f' returning this network function..')
return existing_nf
@staticmethod
def get(nf_name):
""" Retrieves a network function
Args:
nf_name (str): The network function name
Returns:
NetworkFunctionModel object else None
"""
return NetworkFunctionModel.query.filter(
NetworkFunctionModel.nf_name == nf_name).one_or_none()
@staticmethod
def get_all():
""" Retrieves all network functions
Returns:
list: NetworkFunctionModel objects else empty
"""
return NetworkFunctionModel.query.all()
@staticmethod
def delete(**kwargs):
""" Deletes a network function from the database """
nf_name = kwargs['nf_name']
nf = NetworkFunctionModel.query.filter(
NetworkFunctionModel.nf_name == nf_name).one_or_none()
if nf:
db.session.delete(nf)
db.session.commit()
class NetworkFunctionFilter:
def __init__(self, **kwargs):
self.nf_sw_version = kwargs.get('swVersions')
self.nf_names = kwargs.get('nfNames')
self.regex_matcher = re.compile('|'.join(raw_regex for raw_regex in self.nf_names))
def is_nf_in_filter(self, nf_name, orchestration_status):
"""Match the nf name against regex values in Subscription.nfFilter.nfNames
Args:
nf_name: the AAI nf name.
orchestration_status: orchestration status of the nf
Returns:
bool: True if matched, else False.
"""
return self.regex_matcher.search(nf_name) and \
orchestration_status == OrchestrationStatus.ACTIVE.value
class OrchestrationStatus(Enum):
ACTIVE = 'Active'
INVENTORIED = 'Inventoried'
|