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

lister.cli: Fix broken imports

Browse source
This commit is contained in:
Antoine R. Dumont (@ardumont) 2018-07-17 14:04:36 +02:00
parent d88f1b60c9
commit 63cff5b337
No known key found for this signature in database
GPG key ID: 52E2E9840D10C3B8
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
swh/lister/cli.py
View file

@ -26,19 +26,19 @@ def cli(db_url, lister, create_tables, drop_tables):
override_conf = {'lister_db_url': db_url}
if lister == 'github':
from .github import models
from .github.models import IndexingModelBase as ModelBase
from .github.lister import GitHubLister
_lister = GitHubLister(api_baseurl='https://api.github.com',
override_config=override_conf)
elif lister == 'bitbucket':
from .bitbucket import models
from .bitbucket.models import IndexingModelBase as ModelBase
from .bitbucket.lister import BitBucketLister
_lister = BitBucketLister(api_baseurl='https://api.bitbucket.org/2.0',
override_config=override_conf)
elif lister == 'gitlab':
from .gitlab import models
from .gitlab.models import ModelBase
from .gitlab.lister import GitLabLister
_lister = GitLabLister(api_baseurl='https://gitlab.com/api/v4/',
override_config=override_conf)
@ -46,10 +46,10 @@ def cli(db_url, lister, create_tables, drop_tables):
raise ValueError('Only supported listers are %s' % supported_listers)
if drop_tables:
models.ModelBase.metadata.drop_all(_lister.db_engine)
ModelBase.metadata.drop_all(_lister.db_engine)
if create_tables:
models.ModelBase.metadata.create_all(_lister.db_engine)
ModelBase.metadata.create_all(_lister.db_engine)
if __name__ == '__main__':