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()