Differential D3124 Diff 11131 swh/deposit/loader/checker.py

Changeset View

Standalone View

swh/deposit/loader/checker.py

	# Copyright (C) 2017-2019 The Software Heritage developers			# Copyright (C) 2017-2019 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution			# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version			# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information			# See top-level LICENSE file for more information

	import logging			import logging

	from typing import Mapping			from typing import Mapping

				from swh.core.config import SWHConfig

	from swh.deposit.client import PrivateApiDepositClient			from swh.deposit.client import PrivateApiDepositClient


	logger = logging.getLogger(__name__)			logger = logging.getLogger(__name__)


	class DepositChecker:			class DepositChecker(SWHConfig):
	"""Deposit checker implementation.			"""Deposit checker implementation.

	Trigger deposit's checks through the private api.			Trigger deposit's checks through the private api.

	"""			"""

	def __init__(self, client=None):			CONFIG_BASE_FILENAME = "deposit/checker"

				DEFAULT_CONFIG = {
				"deposit": ("dict", {"url": "http://localhost:5006/1/private/", "auth": {},})
				}

				def __init__(self, config=None):
	super().__init__()			super().__init__()
	self.client = client if client else PrivateApiDepositClient()			if config is None:
				self.config = self.parse_config_file()
				else:
				self.config = config
				self.client = PrivateApiDepositClient(config=self.config["deposit"])

	def check(self, deposit_check_url: str) -> Mapping[str, str]:			def check(self, collection: str, deposit_id: str) -> Mapping[str, str]:
	status = None			status = None
				deposit_check_url = f"/{collection}/{deposit_id}/check/"
	try:			try:
	r = self.client.check(deposit_check_url)			r = self.client.check(deposit_check_url)
	status = "eventful" if r == "verified" else "failed"			status = "eventful" if r == "verified" else "failed"
	except Exception:			except Exception:
	logger.exception("Failure during check on '%s'" % (deposit_check_url,))			logger.exception("Failure during check on '%s'" % (deposit_check_url,))
	status = "failed"			status = "failed"
	return {"status": status}			return {"status": status}