|
@@ -62,6 +62,8 @@ class Command(BaseCommand):
|
|
|
interpreter = options.get('interpreter') # Only when connectors are enabled. Later will deprecate `dialect`.
|
|
interpreter = options.get('interpreter') # Only when connectors are enabled. Later will deprecate `dialect`.
|
|
|
user = options['user']
|
|
user = options['user']
|
|
|
request = options['request']
|
|
request = options['request']
|
|
|
|
|
+ self.queries = options.get('queries')
|
|
|
|
|
+ self.tables = options.get('tables')
|
|
|
|
|
|
|
|
tables = 'tables_standard.json' if dialect not in ('hive', 'impala') else (
|
|
tables = 'tables_standard.json' if dialect not in ('hive', 'impala') else (
|
|
|
'tables_transactional.json' if has_concurrency_support() else 'tables.json'
|
|
'tables_transactional.json' if has_concurrency_support() else 'tables.json'
|
|
@@ -104,15 +106,16 @@ class Command(BaseCommand):
|
|
|
raise InstallException(_('No %s tables are available as samples') % dialect)
|
|
raise InstallException(_('No %s tables are available as samples') % dialect)
|
|
|
|
|
|
|
|
for table_dict in table_list:
|
|
for table_dict in table_list:
|
|
|
- full_name = '%s.%s' % (db_name, table_dict['table_name'])
|
|
|
|
|
- try:
|
|
|
|
|
- table = SampleTable(table_dict, dialect, db_name, interpreter=interpreter, request=request)
|
|
|
|
|
- if table.install(django_user):
|
|
|
|
|
- self.successes.append(_('Table %s installed.') % full_name)
|
|
|
|
|
- except Exception as ex:
|
|
|
|
|
- msg = str(ex)
|
|
|
|
|
- LOG.error(msg)
|
|
|
|
|
- self.errors.append(_('Could not install table %s: %s') % (full_name, msg))
|
|
|
|
|
|
|
+ if self.tables is None or table_dict['table_name'] in self.tables:
|
|
|
|
|
+ full_name = '%s.%s' % (db_name, table_dict['table_name'])
|
|
|
|
|
+ try:
|
|
|
|
|
+ table = SampleTable(table_dict, dialect, db_name, interpreter=interpreter, request=request)
|
|
|
|
|
+ if table.install(django_user):
|
|
|
|
|
+ self.successes.append(_('Table %s installed.') % full_name)
|
|
|
|
|
+ except Exception as ex:
|
|
|
|
|
+ msg = str(ex)
|
|
|
|
|
+ LOG.error(msg)
|
|
|
|
|
+ self.errors.append(_('Could not install table %s: %s') % (full_name, msg))
|
|
|
|
|
|
|
|
|
|
|
|
|
def install_queries(self, django_user, dialect, interpreter=None):
|
|
def install_queries(self, django_user, dialect, interpreter=None):
|
|
@@ -130,14 +133,15 @@ class Command(BaseCommand):
|
|
|
raise InstallException(_('No %s queries are available as samples') % dialect)
|
|
raise InstallException(_('No %s queries are available as samples') % dialect)
|
|
|
|
|
|
|
|
for design_dict in design_list:
|
|
for design_dict in design_list:
|
|
|
- design = SampleQuery(design_dict)
|
|
|
|
|
- try:
|
|
|
|
|
- design.install(django_user, interpreter=interpreter)
|
|
|
|
|
- self.successes.append(_('Query %s %s installed.') % (design_dict['name'], dialect))
|
|
|
|
|
- except Exception as ex:
|
|
|
|
|
- msg = str(ex)
|
|
|
|
|
- LOG.error(msg)
|
|
|
|
|
- self.errors.append(_('Could not install %s query: %s') % (dialect, msg))
|
|
|
|
|
|
|
+ if self.queries is None or design_dict['name'] in self.queries:
|
|
|
|
|
+ design = SampleQuery(design_dict)
|
|
|
|
|
+ try:
|
|
|
|
|
+ design.install(django_user, interpreter=interpreter)
|
|
|
|
|
+ self.successes.append(_('Query %s %s installed.') % (design_dict['name'], dialect))
|
|
|
|
|
+ except Exception as ex:
|
|
|
|
|
+ msg = str(ex)
|
|
|
|
|
+ LOG.error(msg)
|
|
|
|
|
+ self.errors.append(_('Could not install %s query: %s') % (dialect, msg))
|
|
|
|
|
|
|
|
|
|
|
|
|
class SampleTable(object):
|
|
class SampleTable(object):
|