Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/storage/__init__.py
| Show First 20 Lines • Show All 42 Lines • ▼ Show 20 Lines | elif cls == "rabbitmq": | ||||
| from swh.provenance.storage.rabbitmq.client import ( | from swh.provenance.storage.rabbitmq.client import ( | ||||
| ProvenanceStorageRabbitMQClient, | ProvenanceStorageRabbitMQClient, | ||||
| ) | ) | ||||
| rmq_storage = ProvenanceStorageRabbitMQClient(**kwargs) | rmq_storage = ProvenanceStorageRabbitMQClient(**kwargs) | ||||
| if TYPE_CHECKING: | if TYPE_CHECKING: | ||||
| assert isinstance(rmq_storage, ProvenanceStorageInterface) | assert isinstance(rmq_storage, ProvenanceStorageInterface) | ||||
| return rmq_storage | return rmq_storage | ||||
| elif cls == "journal": | |||||
| from swh.journal.writer import get_journal_writer | |||||
| from swh.provenance.storage.journal import ProvenanceStorageJournal | |||||
| storage = get_provenance_storage(**kwargs["storage"]) | |||||
| journal = get_journal_writer(**kwargs["journal_writer"]) | |||||
| ret = ProvenanceStorageJournal(storage=storage, journal=journal) | |||||
| return ret | |||||
| raise ValueError | raise ValueError | ||||