diff --git a/setup.py b/setup.py --- a/setup.py +++ b/setup.py @@ -50,6 +50,10 @@ extras_require={'testing': parse_requirements('test')}, vcversioner={}, include_package_data=True, + entrypoints=''' + [swh.workers] + loader.package=swh.loader.package:register + ''', classifiers=[ "Programming Language :: Python :: 3", "Intended Audience :: Developers", diff --git a/swh/loader/package/__init__.py b/swh/loader/package/__init__.py --- a/swh/loader/package/__init__.py +++ b/swh/loader/package/__init__.py @@ -4,6 +4,8 @@ # See top-level LICENSE file for more information +from typing import Any, Mapping + try: from swh.loader.core._version import __version__ # type: ignore except ImportError: @@ -17,3 +19,9 @@ ) } } + + +def register() -> Mapping[str, Any]: + return { + 'task_modules': ['%s.tasks' % __name__], + } diff --git a/swh/loader/package/tasks.py b/swh/loader/package/tasks.py --- a/swh/loader/package/tasks.py +++ b/swh/loader/package/tasks.py @@ -19,7 +19,7 @@ @shared_task(name=__name__ + '.LoadDebian') -def load_debian(*, url, date, packages): +def load_deb(*, url, date, packages): return DebianLoader(url, date, packages).load()