Exemplo n.º 1
0
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.'
Exemplo n.º 2
0
    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()