Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/__init__.py
Show First 20 Lines • Show All 86 Lines • ▼ Show 20 Lines | if cls in ["local", "postgresql"]: | ||||
) | ) | ||||
elif cls == "mongodb": | elif cls == "mongodb": | ||||
from .mongo.backend import ProvenanceStorageMongoDb | from .mongo.backend import ProvenanceStorageMongoDb | ||||
engine = kwargs.get("engine", "pymongo") | engine = kwargs.get("engine", "pymongo") | ||||
return ProvenanceStorageMongoDb(engine=engine, **kwargs["db"]) | return ProvenanceStorageMongoDb(engine=engine, **kwargs["db"]) | ||||
elif cls == "rabbitmq": | |||||
from .api.client import ProvenanceStorageRabbitMQClient | |||||
rmq_storage = ProvenanceStorageRabbitMQClient(**kwargs) | |||||
if TYPE_CHECKING: | |||||
assert isinstance(rmq_storage, ProvenanceStorageInterface) | |||||
return rmq_storage | |||||
raise ValueError | raise ValueError |