Changeset View

Standalone View

	# Copyright (C) 2016-2017 The Software Heritage developers			# Copyright (C) 2016-2017 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

	from swh.scheduler.task import Task			from swh.scheduler.task import Task
	from swh.vault.cookers import get_cooker			from swh.vault.cookers import get_cooker


	class SWHCookingTask(Task):			class SWHCookingTask(Task):
	"""Main task to cook a bundle."""			"""Main task to cook a bundle."""

	task_queue = 'swh_vault_cooking'			task_queue = 'swh_vault_cooking'

	def run_task(self, config, obj_type, obj_id):			def run_task(self, config, obj_type, obj_id):
	cooker = get_cooker(obj_type)(config, obj_type, obj_id)			with get_cooker(obj_type)(config, obj_type, obj_id) as cooker:
	cooker.cook()			cooker.cook()