From db53ae02f41f5de19471ca315861a8fc4942d969 Mon Sep 17 00:00:00 2001 From: "Michael D. M. Dryden" <mdryden@chem.utoronto.ca> Date: Tue, 26 Aug 2014 14:46:59 -0400 Subject: [PATCH] Documentation and code cleanup of /interface/chronoamp.py --- .../dstat-interface/interface/chronoamp.py | 31 +++++++++---------- 1 file changed, 15 insertions(+), 16 deletions(-) diff --git a/dstat-interface/dstat-interface/interface/chronoamp.py b/dstat-interface/dstat-interface/interface/chronoamp.py index 06b28b4..f1eb596 100644 --- a/dstat-interface/dstat-interface/interface/chronoamp.py +++ b/dstat-interface/dstat-interface/interface/chronoamp.py @@ -2,18 +2,15 @@ import gtk -class chronoamp: +class Chronoamp: def __init__(self): self.builder = gtk.Builder() self.builder.add_from_file('interface/chronoamp.glade') self.builder.connect_signals(self) self.statusbar = self.builder.get_object('statusbar') - self.potential = self.builder.get_object('potential_entry') - self.time = self.builder.get_object('time_entry') self.model = self.builder.get_object('ca_list') self.treeview = self.builder.get_object('treeview') - self.cell_renderer = gtk.CellRendererText() self.treeview.insert_column_with_attributes(-1, "Time", @@ -21,15 +18,18 @@ class chronoamp: self.treeview.insert_column_with_attributes(-1, "Potential", self.cell_renderer, text=0).set_expand(True) - self.treeviewselection = self.treeview.get_selection() - self.treeviewselection.set_mode(gtk.SELECTION_MULTIPLE) + self.selection = self.treeview.get_selection() + self.selection.set_mode(gtk.SELECTION_MULTIPLE) def on_add_button_clicked(self, widget): + """Add current values in potential_entry and time_entry to model.""" + self.statusbar.remove_all(0) try: - potential = int(self.potential.get_text()) - time = int(self.time.get_text()) + potential = int( + self.builder.get_object('potential_entry').get_text()) + time = int(self.builder.get_object('time_entry').get_text()) if (potential > 1499 or potential < -1500): raise ValueError("Potential out of range") @@ -38,20 +38,19 @@ class chronoamp: self.model.append([potential, time]) - except ValueError as e: - self.statusbar.push(0, str(e)) - except TypeError as e: - self.statusbar.push(0, str(e)) + except ValueError as err: + self.statusbar.push(0, str(err)) + except TypeError as err: + self.statusbar.push(0, str(err)) def on_remove_button_clicked(self, widget): + """Remove currently selected items from model.""" # returns 2-tuple: treemodel, list of paths of selected rows - self.selected_rows = list(self.treeviewselection.get_selected_rows()[1]) - + self.selected_rows = list(self.selection.get_selected_rows()[1]) self.referencelist = [] for i in self.selected_rows: - x=gtk.TreeRowReference(self.model, i) - self.referencelist.append(x) + self.referencelist.append(gtk.TreeRowReference(self.model, i)) for i in self.referencelist: self.model.remove(self.model.get_iter(i.get_path())) -- GitLab