Changeset View
Changeset View
Standalone View
Standalone View
swh/lister/cli.py
Show First 20 Lines • Show All 72 Lines • ▼ Show 20 Lines | def cli(db_url, lister, create_tables, drop_tables, with_data): | ||||
elif lister == 'pypi': | elif lister == 'pypi': | ||||
from .pypi.models import ModelBase | from .pypi.models import ModelBase | ||||
from .pypi.lister import PyPILister | from .pypi.lister import PyPILister | ||||
_lister = PyPILister(override_config=override_conf) | _lister = PyPILister(override_config=override_conf) | ||||
elif lister == 'npm': | elif lister == 'npm': | ||||
from .npm.models import IndexingModelBase as ModelBase | from .npm.models import IndexingModelBase as ModelBase | ||||
from .npm.models import NpmVisitModel | |||||
from .npm.lister import NpmLister | from .npm.lister import NpmLister | ||||
_lister = NpmLister(override_config=override_conf) | _lister = NpmLister(override_config=override_conf) | ||||
if drop_tables: | |||||
NpmVisitModel.metadata.drop_all(_lister.db_engine) | |||||
if create_tables: | |||||
NpmVisitModel.metadata.create_all(_lister.db_engine) | |||||
else: | else: | ||||
raise ValueError('Only supported listers are %s' % SUPPORTED_LISTERS) | raise ValueError('Only supported listers are %s' % SUPPORTED_LISTERS) | ||||
if drop_tables: | if drop_tables: | ||||
ModelBase.metadata.drop_all(_lister.db_engine) | ModelBase.metadata.drop_all(_lister.db_engine) | ||||
if create_tables: | if create_tables: | ||||
ModelBase.metadata.create_all(_lister.db_engine) | ModelBase.metadata.create_all(_lister.db_engine) | ||||
if with_data and insert_minimum_data: | if with_data and insert_minimum_data: | ||||
insert_minimum_data(_lister) | insert_minimum_data(_lister) | ||||
if __name__ == '__main__': | if __name__ == '__main__': | ||||
cli() | cli() |