def load_reports(self):
if self.reports and self.is_enabled():
importer = Importer()
for report_name, pkg_name in self.get_reports():
importer.import_file(report_name, pkg_name)
self._reports_loaded = True
def get_tables(self):
buf = StringIO()
i = Importer(buf=buf)
from steelscript.appfwk.apps.datasource.modules import analysis, html
for p in plugins.all():
for d in p.get_datasources():
i.import_file(*d)
return self.get_subclasses(DatasourceTable)
def handle(self, *args, **options):
self.stdout.write('Reloading report objects ... ')
management.call_command('clean_pyc', path=settings.PROJECT_ROOT)
self.importer = Importer(buf=self.stdout)
if options['report_id']:
# single report
report_id = options['report_id']
pk = int(report_id)
report = Report.objects.get(pk=pk)
management.call_command('clean',
applications=False,
report_id=report_id,
clear_cache=False,
clear_logs=False)
DeviceManager.clear()
self.import_module(report.sourcefile)
elif options['report_name']:
name = options['report_name']
try:
report = Report.objects.get(sourcefile__endswith=name)
sourcefile = report.sourcefile
management.call_command('clean',
applications=False,
report_id=report.id,
clear_cache=False,
clear_logs=False)
self.import_module(sourcefile)
except ObjectDoesNotExist:
self.import_module(name)
DeviceManager.clear()
elif options['namespace']:
reports = Report.objects.filter(namespace=options['namespace'])
self.capture_enabled(reports)
# clear all data from one namespace (module)
for report in reports:
management.call_command('clean',
applications=False,
report_id=report.id,
clear_cache=False,
clear_logs=False)
# ignore all dirs besides the one we cleared, then import
root_dir = settings.REPORTS_DIR
target_dir = options['namespace']
ignore_list = [
os.path.basename(os.path.normpath(x))
for x in os.listdir(root_dir) if x != target_dir
]
self.importer.import_directory(root_dir, ignore_list=ignore_list)
self.apply_enabled()
else:
self.capture_enabled()
# clear all data
management.call_command('clean',
applications=True,
report_id=None,
clear_cache=True,
clear_logs=False)
# start with fresh device instances
DeviceManager.clear()
report_dir = settings.REPORTS_DIR
self.importer.import_directory(report_dir, report_name=None)
self.apply_enabled()
