tasks.py
No OneTemporary
Actions

Size

1 KB

Subscribers

None

tasks.py
View Options

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

	import dateutil.parser

	from swh.scheduler.task import Task

	from .loader import GitLoader
	from .updater import BulkUpdater
	from .reader import GitSha1RemoteReaderAndSendToQueue


	# TODO: rename to LoadRemoteGitRepository
	class UpdateGitRepository(Task):
	"""Import a git repository from a remote location"""
	task_queue = 'swh_loader_git'

	def run(self, repo_url, base_url=None):
	"""Import a git repository"""
	loader = BulkUpdater()
	loader.log = self.log

	return loader.load(repo_url, base_url)


	class LoadDiskGitRepository(Task):
	"""Import a git repository from disk"""
	task_queue = 'swh_loader_git'

	def run(self, origin_url, directory, date):
	"""Import a git repository, cloned in `directory` from `origin_url` at
	`date`."""

	loader = GitLoader()
	loader.log = self.log

	return loader.load(origin_url, directory, dateutil.parser.parse(date))


	class ReaderGitRepository(Task):
	task_queue = 'swh_reader_git'

	def run(self, repo_url, base_url=None):
	"""Read a git repository from a remote location and send sha1 to
	archival.

	"""
	loader = GitSha1RemoteReaderAndSendToQueue()
	loader.log = self.log

	return loader.load(repo_url)