Changeset View
Changeset View
Standalone View
Standalone View
swh/loader/mercurial/tasks.py
- This file was moved from swh/loader/mercurial/tasks_from_disk.py.
# Copyright (C) 2020-2021 The Software Heritage developers | # Copyright (C) 2020-2021 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 typing import Optional | from typing import Optional | ||||
from celery import shared_task | from celery import shared_task | ||||
from swh.loader.mercurial.utils import parse_visit_date | from swh.loader.mercurial.utils import parse_visit_date | ||||
from .from_disk import HgArchiveLoaderFromDisk, HgLoaderFromDisk | from .loader import HgArchiveLoader, HgLoader | ||||
@shared_task(name=__name__ + ".LoadMercurialFromDisk") | @shared_task(name=__name__ + ".LoadMercurial") | ||||
def load_hg( | def load_hg( | ||||
*, url: str, directory: Optional[str] = None, visit_date: Optional[str] = None | *, url: str, directory: Optional[str] = None, visit_date: Optional[str] = None | ||||
): | ): | ||||
"""Mercurial repository loading | """Mercurial repository loading | ||||
Import a mercurial tarball into swh. | Import a mercurial tarball into swh. | ||||
Args: see :func:`HgLoaderFromDisk` constructor. | Args: see :func:`HgLoader` constructor. | ||||
""" | """ | ||||
loader = HgLoaderFromDisk.from_configfile( | loader = HgLoader.from_configfile( | ||||
url=url, directory=directory, visit_date=parse_visit_date(visit_date) | url=url, directory=directory, visit_date=parse_visit_date(visit_date) | ||||
) | ) | ||||
return loader.load() | return loader.load() | ||||
@shared_task(name=__name__ + ".LoadArchiveMercurialFromDisk") | @shared_task(name=__name__ + ".LoadArchiveMercurial") | ||||
def load_hg_from_archive( | def load_hg_from_archive( | ||||
*, url: str, archive_path: Optional[str] = None, visit_date: Optional[str] = None | *, url: str, archive_path: Optional[str] = None, visit_date: Optional[str] = None | ||||
): | ): | ||||
"""Import a mercurial tarball into swh. | """Import a mercurial tarball into swh. | ||||
Args: see :func:`HgArchiveLoaderFromDisk` constructor. | Args: see :func:`HgArchiveLoader` constructor. | ||||
""" | """ | ||||
loader = HgArchiveLoaderFromDisk.from_configfile( | loader = HgArchiveLoader.from_configfile( | ||||
url=url, archive_path=archive_path, visit_date=parse_visit_date(visit_date) | url=url, archive_path=archive_path, visit_date=parse_visit_date(visit_date) | ||||
) | ) | ||||
return loader.load() | return loader.load() |