1 #######################################################################################
2 # Copyright ETSI Contributors and Others.
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
16 #######################################################################################
17 from osm_common
import dbmemory
, dbmongo
18 from osm_mon
.core
.config
import Config
22 def __init__(self
, config
: Config
):
23 if config
.get("database", "driver") == "mongo":
24 self
.common_db
= dbmongo
.DbMongo()
25 elif config
.get("database", "driver") == "memory":
26 self
.common_db
= dbmemory
.DbMemory()
29 "Unknown database driver {}".format(config
.get("section", "driver"))
31 self
.common_db
.db_connect(config
.get("database"))
33 def get_vnfr(self
, nsr_id
: str, member_index
: str):
34 vnfr
= self
.common_db
.get_one(
35 "vnfrs", {"nsr-id-ref": nsr_id
, "member-vnf-index-ref": member_index
}
39 def get_vnfrs(self
, nsr_id
: str = None, vim_account_id
: str = None):
40 if nsr_id
and vim_account_id
:
41 raise NotImplementedError("Only one filter is currently supported")
44 self
.get_vnfr(nsr_id
, member
["member-vnf-index"])
45 for member
in self
.get_nsr(nsr_id
)["nsd"]["constituent-vnfd"]
48 vnfrs
= self
.common_db
.get_list("vnfrs", {"vim-account-id": vim_account_id
})
50 vnfrs
= self
.common_db
.get_list("vnfrs")
53 def get_nsr(self
, nsr_id
: str):
54 nsr
= self
.common_db
.get_one("nsrs", {"id": nsr_id
})
58 return self
.common_db
.get_list("vnfds")
60 def get_monitoring_vnfds(self
):
61 return self
.common_db
.get_list(
62 "vnfds", {"vdu.monitoring-parameter": {"$exists": "true"}}
65 def decrypt_vim_password(self
, vim_password
: str, schema_version
: str, vim_id
: str):
66 return self
.common_db
.decrypt(vim_password
, schema_version
, vim_id
)
68 def get_vim_accounts(self
):
69 return self
.common_db
.get_list("vim_accounts")
71 def get_vim_account(self
, vim_account_id
: str) -> dict:
72 vim_account
= self
.common_db
.get_one("vim_accounts", {"_id": vim_account_id
})
73 vim_account
["vim_password"] = self
.decrypt_vim_password(
74 vim_account
["vim_password"], vim_account
["schema_version"], vim_account_id
76 vim_config_encrypted_dict
= {
77 "1.1": ("admin_password", "nsx_password", "vcenter_password"),
85 vim_config_encrypted
= vim_config_encrypted_dict
["default"]
86 if vim_account
["schema_version"] in vim_config_encrypted_dict
.keys():
87 vim_config_encrypted
= vim_config_encrypted_dict
[
88 vim_account
["schema_version"]
90 if "config" in vim_account
:
91 for key
in vim_account
["config"]:
92 if key
in vim_config_encrypted
:
93 vim_account
["config"][key
] = self
.decrypt_vim_password(
94 vim_account
["config"][key
],
95 vim_account
["schema_version"],
103 vnf_member_index
: str,
108 q_filter
= {"action_type": action_type
}
110 q_filter
["tags.ns_id"] = nsr_id
112 q_filter
["tags.vnf_member_index"] = vnf_member_index
114 q_filter
["tags.vdu_id"] = vdu_id
116 q_filter
["tags.vdu_name"] = vdu_name
117 alert
= self
.common_db
.get_one(
118 table
="alerts", q_filter
=q_filter
, fail_on_empty
=False
122 def update_alert_status(self
, uuid
: str, alarm_status
: str):
123 modified_count
= self
.common_db
.set_one(
124 "alerts", {"uuid": uuid
}, {"alarm_status": alarm_status
}
126 return modified_count
128 def create_nslcmop(self
, nslcmop
: dict):
129 self
.common_db
.create("nslcmops", nslcmop
)
131 def get_nslcmop(self
, nsr_id
: str, operation_type
: str, since
: str):
134 q_filter
["nsInstanceId"] = nsr_id
136 q_filter
["lcmOperationType"] = operation_type
138 q_filter
["startTime"] = {"$gt": since
}
139 ops
= self
.common_db
.get_list(table
="nslcmops", q_filter
=q_filter
)