import os import tkinter as tk import tkinter.ttk as ttk from tkinter import filedialog import pandas from . import validators from . import inout APP_STATE_1 = 'no valid files selected' APP_STATE_2 = 'no valid fields selected' APP_STATE_3 = 'ok, single files' APP_STATE_4 = 'ok, multiple files' class StatusPanel(tk.Frame): def __init__(self, parent): tk.Frame.__init__(self, parent.master) self._parent = parent self.label = tk.Label(self, bd=1, relief=tk.SUNKEN, anchor=tk.S) self.label.pack(fill=tk.X) self.pack(side=tk.BOTTOM, fill=tk.X) def set_text(self, text): self.label.config(text=text) self.label.update_idletasks() def clear_text(self): self.set_text('') class FilePanel(tk.Frame): def __init__(self, parent): tk.Frame.__init__(self, parent.master) self._parent = parent self.btn_files = tk.Button(self, text="Select Files", command=self._parent.select_files) self.btn_files.pack(pady=5, padx=5) self.label = tk.Label(self, anchor=tk.SW) self.label.pack(fill=tk.X, pady=5, padx=5) ttk.Separator(self, orient=tk.HORIZONTAL).pack(side=tk.BOTTOM, fill=tk.X) self.pack(side=tk.TOP, fill=tk.X) def set_text(self, text): self.label.config(text=text) self.label.update_idletasks() def clear_text(self): self.set_text('') class FieldPanel(tk.Frame): def __init__(self, parent): tk.Frame.__init__(self, parent.master) self._parent = parent self.is_single = tk.BooleanVar() self.is_single.set(True) self.rbtn_single = tk.Radiobutton( self, text='seperate excel files', variable=self.is_single, value=True, command=self.disable_listbox) self.rbtn_combine = tk.Radiobutton( self, text='one combined excel file', variable=self.is_single, value=False, command=self.enable_listbox) self.rbtn_single.pack(anchor=tk.W) self.rbtn_combine.pack(anchor=tk.W) self.listbox = tk.Listbox(self, selectmode=tk.MULTIPLE, activestyle='none', height=19) self.listbox.pack(fill=tk.X, ipady=5, ipadx=5, ) self.listbox.bind('<>', self._parent.toggle_run_button) self.pack(side=tk.TOP, fill=tk.X) def disable_radio_btn(self): self.rbtn_single.config(state="disabled") self.rbtn_combine.config(state="disabled") def disable_listbox(self): self.listbox.config(state="disabled") self._parent.toggle_run_button() def disable(self): self.disable_listbox() self.disable_radio_btn() def enable_radio_btn(self): self.rbtn_single.config(state="normal") self.rbtn_combine.config(state="normal") self.is_single.set(True) def enable_listbox(self): self.listbox.config(state="normal") self._parent.toggle_run_button() def enable(self): self.enable_listbox() self.enable_radio_btn() def set_listbox_content(self, choices, selection): self.enable_listbox() current = self.listbox.size() self.listbox.selection_clear(0, current) self.listbox.delete(0,current) for i, element in enumerate(choices): self.listbox.insert(tk.END, element) if element in selection: self.listbox.selection_set(i) self.disable_listbox() class ActionPanel(tk.Frame): def __init__(self, parent): tk.Frame.__init__(self, parent.master) self._parent = parent self.btn_go = tk.Button(self, text="GO!", command=self._parent.convert_files) self.btn_go.pack(side=tk.LEFT, pady=5, padx=5) self.btn_quit = tk.Button(self, text="Quit", command=self._parent.quit) self.btn_quit.pack(side=tk.RIGHT, pady=5, padx=5) self.pack(side=tk.BOTTOM, fill=tk.X) def disable(self): self.btn_go.config(state='disabled') def enable(self): self.btn_go.config(state='active') class Application(tk.Frame): def __init__(self, master): master.minsize(height=330, width=300) tk.Frame.__init__(self, master) self._master = master self.pack(fill=tk.BOTH) self.file_validator = None self.validated_files = None self.status_panel = StatusPanel(self) self.file_panel = FilePanel(self) self.field_panel = FieldPanel(self) self.action_panel = ActionPanel(self) self.reset() def reset(self): self.status_panel.clear_text() self.field_panel.disable() self.action_panel.disable() self.file_validator = None self.validated_files = None def select_files(self): self.reset() self.file_panel.clear_text() opts = { 'initialdir': '~/_signalyse test', 'filetypes': [('Signalyse Statistics', '.stx'), ('Signalyse Data', '.dat'), ('Sensovation Data', '.csv')], 'multiple': True} file_selection = tk.filedialog.askopenfilename(**opts) if not isinstance(file_selection, tuple) or len(file_selection) == 0: return self.file_validator = validators.guess_validator(file_selection) if self.file_validator is None: return self.validated_files = [f for f in validators.validate_files(file_selection, self.file_validator)] status = '%d of %d valid %s files found' % (len(self.validated_files), len(file_selection), self.file_validator.type) self.file_panel.set_text(status) if self._state == APP_STATE_1: return self.reset() self.field_panel.enable_radio_btn() self.field_panel.set_listbox_content(self.file_validator.data_fields, self.file_validator.defaults) self.toggle_run_button() def toggle_run_button(self, event=None): self.action_panel.disable() if self._state in (APP_STATE_3, APP_STATE_4): self.action_panel.enable() def quit(self): self._master.destroy() def convert_files(self): state = self._state if state not in (APP_STATE_3, APP_STATE_4): return self.reset() if state == APP_STATE_3: text = self._convert_to_separate_files() else: text = self._combine_data_files() self.status_panel.set_text(text + ' Done') @property def _state(self): c = self.field_panel.rbtn_single.config() if not self.validated_files or c['state'] == 'disabled': return APP_STATE_1 elif self.field_panel.is_single.get(): return APP_STATE_3 elif self.field_panel.listbox.curselection(): return APP_STATE_4 else: return APP_STATE_2 def _convert_to_separate_files(self): total = len(self.validated_files) text = 'Converting data file %d of %d to separate excel file ...' all_fields = self.file_validator.id_fields + self.file_validator.data_fields for i, data_file in enumerate(self.validated_files): self.status_panel.set_text(text % (i+1, total)) data_frame = inout.read_data_file(data_file, all_fields) data_frame = data_frame.set_index(data_file.validator.id_fields) name, ext = os.path.splitext(data_file.path) path = name + ext.replace('.', '_') + '.xlsx' inout.write_excel_file(path, data_frame) return text % (i+1, total) def _combine_data_files(self): total = len(self.validated_files) text = 'Combining data file %d of %d to single excel file ...' data_fields = [self.file_validator.data_fields[i] for i in self.field_panel.listbox.curselection()] selected_fields = self.file_validator.id_fields + data_fields data_frame_list = [] for i, data_file in enumerate(self.validated_files): self.status_panel.set_text(text % (i + 1, total)) data_frame = inout.read_data_file(data_file, selected_fields) data_frame['File Name'] = os.path.basename(data_file.path) data_frame_list.append(data_frame) master_frame = pandas.concat(data_frame_list, ignore_index=True) pivoted_df = master_frame.pivot(self.file_validator.id_fields[0],'File Name') col_grouper = dict() value_columns = [] for value_col, file_col in pivoted_df.columns: if value_col not in col_grouper: col_grouper[value_col] = [] value_columns.append(value_col) col_grouper[value_col].append(file_col) sorted_col_names = [] for value_col in value_columns: sorted_file_cols = sorted(col_grouper[value_col], key=lambda x: inout.natural_sort(x)) sorted_col_names.extend([(value_col, file_col) for file_col in sorted_file_cols]) sorted_df = pivoted_df.reindex_axis(sorted_col_names, axis=1) first_file = self.validated_files[0] directory = os.path.dirname(first_file.path) pseudo_extension = self.file_validator.extension.replace('.', '_') xls_path = os.path.join(directory, 'combined' + pseudo_extension + '.xlsx') inout.write_excel_file(xls_path, sorted_df) return text % (i + 1, total)