From 6c54b64a8ff52f894c6cbc08308001217a42a5be Mon Sep 17 00:00:00 2001 From: "Antoine R. Dumont (@ardumont)" Date: Fri, 27 Jul 2018 10:19:48 +0200 Subject: [PATCH] swh.lister.cli: Add debian lister to the list of supported listers --- swh/lister/cli.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/swh/lister/cli.py b/swh/lister/cli.py index 17dbbe6..0eacaea 100644 --- a/swh/lister/cli.py +++ b/swh/lister/cli.py @@ -12,7 +12,7 @@ import click help='SQLAlchemy DB URL; see ' '') # noqa @click.option('--lister', required=1, - type=click.Choice(['github', 'gitlab', 'bitbucket']), + type=click.Choice(['github', 'gitlab', 'bitbucket', 'debian']), help='Lister to act upon') @click.option('--create-tables', is_flag=True, default=False, help='create tables') @@ -22,7 +22,7 @@ def cli(db_url, lister, create_tables, drop_tables): """Initialize db model according to lister. """ - supported_listers = ['github', 'gitlab', 'bitbucket'] + supported_listers = ['github', 'gitlab', 'bitbucket', 'debian'] override_conf = {'lister_db_url': db_url} if lister == 'github': @@ -42,6 +42,11 @@ def cli(db_url, lister, create_tables, drop_tables): from .gitlab.lister import GitLabLister _lister = GitLabLister(api_baseurl='https://gitlab.com/api/v4/', override_config=override_conf) + elif lister == 'debian': + from .debian.lister import DebianLister + ModelBase = DebianLister.MODEL + _lister = DebianLister() + else: raise ValueError('Only supported listers are %s' % supported_listers)