def main(): input_foldername = UIUtils.open_folder(title='Select Source Folder') output_foldername = None if input_foldername: output_foldername = UIUtils.open_folder( title='Select Destination Folder') if not input_foldername or not output_foldername: exit() csv_filenames = filter(lambda name: name.lower().endswith('.csv'), os.listdir(input_foldername)) for filename in csv_filenames: process_file(input_foldername + '\\' + filename, output_foldername + '\\' + filename) print 'Filtering complete.'
def _run_config(self, treeview): model, it = treeview.get_selection().get_selected() if it: config = model.get(it, 4)[0] trs_folder = UIUtils.open_folder(title='Select TRS Folder...') if trs_folder: #will be None if user clicked 'cancel' or closed the dialog window export_folder = UIUtils.open_folder( title='Select Output Folder...') if export_folder: #trs_filenames = glob.glob(trs_folder + '\\*.trs') trs_filenames = self._get_trs_filenames(trs_folder) export_filenames = map( lambda name: '%s/%s-stats.csv' % (export_folder, os.path.basename(name)[:-4]), trs_filenames) phases = [ 'File %d of %d' % (i + 1, len(trs_filenames)) for i in range(len(trs_filenames)) ] dialog = ProgressDialog('Working...', phases) dialog.show() for j in range(len(trs_filenames)): exporter = StatsExporter( config, trs_filenames[j], export_filenames[j], summary_filename=export_folder + "/summary.csv") exporter.export() dialog.set_fraction(1.0) if j < len(trs_filenames) - 1: dialog.next_phase() dialog.ensure_finish() UIUtils.show_message_dialog('Export successful.') else: UIUtils.show_no_sel_dialog()