Changeset View

Standalone View

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":		elif cls == "rabbitmq":
from .api.client import RemoteProvenanceStorage		from .api.client import ProvenanceStorageRabbitMQClient

storage = RemoteProvenanceStorage(**kwargs)		rmq_storage = ProvenanceStorageRabbitMQClient(**kwargs)
assert isinstance(storage, ProvenanceStorageInterface)		if TYPE_CHECKING:
return storage		assert isinstance(rmq_storage, ProvenanceStorageInterface)
		return rmq_storage

		elif cls in ["remote", "restapi"]:
		from .api.client import ProvenanceStorageRPCClient

		if cls == "remote":
		warnings.warn(
		'"remote" class is deprecated for provenance storage, please '
		'use "restapi" class instead.',
		DeprecationWarning,
		)

		rpc_storage = ProvenanceStorageRPCClient(**kwargs)
		if TYPE_CHECKING:
		assert isinstance(rpc_storage, ProvenanceStorageInterface)
		return rpc_storage

else:
raise ValueError		raise ValueError