def save_script(self): filePath = QtGui.QFileDialog.getSaveFileName(self, u"Save Script", "/", u"CDAT Scripts (*.py)") rows, columns = self.ss.getRowsAndColumns() cells = [] plotters = [] for r in range(rows): for c in range(columns): cell = self.ss.getCell(r, c) cells.append(cell) plotters.append(cell.getPlotters()[:-1]) var_manager = manager() all_files = var_manager.files.values() all_variables = {} for f in all_files: for var in f.vars: all_variables[var.data_key()] = var used_variables = [] # Now that we have all variables sorted out, let's grab relevant ones for pgroup in plotters: for plotter in pgroup: if plotter.can_plot() is False: continue for v in plotter.variables: if v is None: continue v_key = v.id if v_key in all_variables: used_variables.append(all_variables[v_key]) del all_variables[v_key] export_script(filePath[0], used_variables, plotters, rows=rows, columns=columns)
def files(self): if self._files is None: f = ensure_iterable(self.module.get_files()) m = manager() self._files = [m.add_file(file) for file in f] return self._files