blob: 9f21cc66347361329510ce04e10cf6c00d268d3b (
plain)
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
|
# ============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=====================================================
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 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']
NetworkFunctionModel.query.filter(
NetworkFunctionModel.nf_name == nf_name). \
delete(synchronize_session='evaluate')
db.session.commit()
|
