Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/__init__.py
Show First 20 Lines • Show All 90 Lines • ▼ Show 20 Lines | elif cls == "mongodb": | ||||
from pymongo import MongoClient | from pymongo import MongoClient | ||||
from .mongo.backend import ProvenanceStorageMongoDb | from .mongo.backend import ProvenanceStorageMongoDb | ||||
dbname = kwargs["db"].pop("dbname") | dbname = kwargs["db"].pop("dbname") | ||||
db = MongoClient(**kwargs["db"]).get_database(dbname) | db = MongoClient(**kwargs["db"]).get_database(dbname) | ||||
return ProvenanceStorageMongoDb(db) | return ProvenanceStorageMongoDb(db) | ||||
elif cls == "remote": | |||||
from .api.client import RemoteProvenanceStorage | |||||
storage = RemoteProvenanceStorage(**kwargs) | |||||
assert isinstance(storage, ProvenanceStorageInterface) | |||||
return storage | |||||
else: | |||||
raise ValueError | raise ValueError |