Differential D5947 Diff 21385 swh/provenance/__init__.py

Changeset View

Standalone View

swh/provenance/init.py

	from typing import TYPE_CHECKING			from typing import TYPE_CHECKING
	import warnings

	from .postgresql.db_utils import connect			from .postgresql.db_utils import connect

	if TYPE_CHECKING:			if TYPE_CHECKING:
	from swh.provenance.archive import ArchiveInterface			from .archive import ArchiveInterface
	from swh.provenance.provenance import ProvenanceInterface			from .provenance import ProvenanceInterface, ProvenanceStorageInterface


	def get_archive(cls: str, **kwargs) -> "ArchiveInterface":			def get_archive(cls: str, **kwargs) -> "ArchiveInterface":
	if cls == "api":			if cls == "api":
	from swh.provenance.storage.archive import ArchiveStorage
	from swh.storage import get_storage			from swh.storage import get_storage

				from .storage.archive import ArchiveStorage

	return ArchiveStorage(get_storage(**kwargs["storage"]))			return ArchiveStorage(get_storage(**kwargs["storage"]))
	elif cls == "direct":			elif cls == "direct":
	from swh.provenance.postgresql.archive import ArchivePostgreSQL			from .postgresql.archive import ArchivePostgreSQL

	return ArchivePostgreSQL(connect(kwargs["db"]))			return ArchivePostgreSQL(connect(kwargs["db"]))
	else:			else:
	raise NotImplementedError			raise NotImplementedError


	def get_provenance(cls: str, **kwargs) -> "ProvenanceInterface":			def get_provenance(**kwargs) -> "ProvenanceInterface":
				from .backend import ProvenanceBackend

				return ProvenanceBackend(get_provenance_storage(**kwargs))


				def get_provenance_storage(cls: str, **kwargs) -> "ProvenanceStorageInterface":
	if cls == "local":			if cls == "local":
				from .postgresql.provenancedb_base import ProvenanceDBBase

	conn = connect(kwargs["db"])			conn = connect(kwargs["db"])
	if "with_path" in kwargs:			flavor = ProvenanceDBBase(conn).flavor
	warnings.warn(			if flavor == "with-path":
	douarddaUnsubmitted Done Inline Actions it would be best to have the removal of this bwcompat support (and deprecation warning) in a dedicated revision, but meh. douardda: it would be best to have the removal of this bwcompat support (and deprecation warning) in a…
	"Usage of the 'with-path' config option is deprecated. "			from .postgresql.provenancedb_with_path import ProvenanceWithPathDB
	"The db flavor is now used instead.",
	DeprecationWarning,			return ProvenanceWithPathDB(conn)
	)			else:
				from .postgresql.provenancedb_without_path import ProvenanceWithoutPathDB
	with_path = kwargs.get("with_path")
	from swh.provenance.backend import ProvenanceBackend			return ProvenanceWithoutPathDB(conn)

	prov = ProvenanceBackend(conn)
	if with_path is not None:
	flavor = "with-path" if with_path else "without-path"
	if prov.storage.flavor != flavor:
	raise ValueError(
	"The given flavor does not match the flavor stored in the backend."
	)
	return prov
	else:			else:
	raise NotImplementedError			raise NotImplementedError