from osm_common import dbmongo
from osm_policy_module.core.config import Config
+from osm_policy_module.core.exceptions import VdurNotFound
class CommonDbClient:
def __init__(self):
cfg = Config.instance()
self.common_db = dbmongo.DbMongo()
- self.common_db.db_connect({'host': cfg.OSMPOL_DATABASE_HOST,
- 'port': int(cfg.OSMPOL_DATABASE_PORT),
+ self.common_db.db_connect({'uri': cfg.OSMPOL_DATABASE_URI,
'name': 'osm'})
def get_vnfr(self, nsr_id: str, member_index: int):
nslcmop = self.common_db.get_one("nslcmops",
{"_id": nslcmop_id})
return nslcmop
+
+ def get_vdur(self, nsr_id, member_index, vdur_name):
+ vnfr = self.get_vnfr(nsr_id, member_index)
+ for vdur in vnfr['vdur']:
+ if vdur['name'] == vdur_name:
+ return vdur
+ raise VdurNotFound('vdur not found for nsr-id %s, member_index %s and vdur_name %s', nsr_id, member_index,
+ vdur_name)