reonokiy/swh-lister
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

swh.lister.cli: Add debian lister to the list of supported listers

Browse source
This commit is contained in:
Antoine R. Dumont (@ardumont) 2018-07-27 10:19:48 +02:00
parent 364786a2da
commit 6c54b64a8f
No known key found for this signature in database
GPG key ID: 52E2E9840D10C3B8
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
swh/lister/cli.py
View file

@ -12,7 +12,7 @@ import click
help='SQLAlchemy DB URL; see '
'<http://docs.sqlalchemy.org/en/latest/core/engines.html#database-urls>') # 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)