X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=osm_nbi%2Fengine.py;h=5bdbb7e021a23b0be5bd7b106bfde7af84de3afe;hb=3d6881c23aea64d391e1e648c615f9dbd89e00ad;hp=0cd0666ab2b6efe0df91d4652710161e709b4c1d;hpb=0952a48159c11b6d31fff6617f04f06351ed79f3;p=osm%2FNBI.git diff --git a/osm_nbi/engine.py b/osm_nbi/engine.py index 0cd0666..5bdbb7e 100644 --- a/osm_nbi/engine.py +++ b/osm_nbi/engine.py @@ -15,7 +15,7 @@ import logging import yaml -from osm_common import dbmongo, dbmemory, fslocal, msglocal, msgkafka, version as common_version +from osm_common import dbmongo, dbmemory, fslocal, fsmongo, msglocal, msgkafka, version as common_version from osm_common.dbbase import DbException from osm_common.fsbase import FsException from osm_common.msgbase import MsgException @@ -68,7 +68,7 @@ class Engine(object): # Add new versions here } - def __init__(self): + def __init__(self, token_cache): self.db = None self.fs = None self.msg = None @@ -78,6 +78,7 @@ class Engine(object): self.logger = logging.getLogger("nbi.engine") self.map_topic = {} self.write_lock = None + self.token_cache = token_cache def start(self, config): """ @@ -106,6 +107,9 @@ class Engine(object): if config["storage"]["driver"] == "local": self.fs = fslocal.FsLocal() self.fs.fs_connect(config["storage"]) + elif config["storage"]["driver"] == "mongo": + self.fs = fsmongo.FsMongo() + self.fs.fs_connect(config["storage"]) else: raise EngineException("Invalid configuration param '{}' at '[storage]':'driver'".format( config["storage"]["driver"])) @@ -123,7 +127,7 @@ class Engine(object): if config["authentication"]["backend"] == "keystone": self.auth = AuthconnKeystone(config["authentication"], self.db, None) else: - self.auth = AuthconnInternal(config["authentication"], self.db, dict()) + self.auth = AuthconnInternal(config["authentication"], self.db, self.token_cache) if not self.operations: if "resources_to_operations" in config["rbac"]: resources_to_operations_file = config["rbac"]["resources_to_operations"]