From 63cff5b3373edc1b9e03a475988305427008ffbd Mon Sep 17 00:00:00 2001 From: "Antoine R. Dumont (@ardumont)" Date: Tue, 17 Jul 2018 14:04:36 +0200 Subject: [PATCH] lister.cli: Fix broken imports --- swh/lister/cli.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/swh/lister/cli.py b/swh/lister/cli.py index 1f0d6f1..17dbbe6 100644 --- a/swh/lister/cli.py +++ b/swh/lister/cli.py @@ -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__':