Changeset 437:750a9bfbc6dd for src/mlx/gui/delaycodes.py

Timestamp:

02/23/13 08:46:11 (11 years ago)

Author:

István Váradi <ivaradi@…>

Branch:

default

Phase:

public

Message:

The main delay code handling logic is present (re #154)

File:

• src/mlx/gui/delaycodes.py (modified) (8 diffs)

src/mlx/gui/delaycodes.py

@@ -185 +185 @@
 #------------------------------------------------------------------------------
 
+class CheckButton(gtk.CheckButton):
+    """A check button that contains a reference to a row in the delay code
+    data table."""
+    def __init__(self, delayCodeRow):
+        """Construct the check button."""
+        super(CheckButton, self).__init__()
+        self.delayCodeRow = delayCodeRow
+
+#------------------------------------------------------------------------------
+
 CAPTION = 1
 
 DELAYCODE = 2
 
-_data1 = ( ["Num", "Code", "Title", "Description"],
+_data1 = ( lambda row: row[0].strip(),
+           ["Num", "Code", "Title", "Description"],
            [ (CAPTION, "Others"),
              (DELAYCODE, ("      6", "OA  ", "NO GATES/STAND AVAILABLE",
@@ -203 +214 @@
                           "Error with passenger or baggage details")) ] )
 
-_data2 = ( ["MA", "IATA", "Description"],
+_data2 = ( lambda row: row[0].strip(),
+           ["MA", "IATA", "Description"],
            [ (CAPTION, "Passenger and baggage"),
-             (DELAYCODE, ("    012", "01",
+             (DELAYCODE, ("    012", "01   ",
                           "Late shipping of parts and/or materials")),
              (DELAYCODE, ("    111", "11",
@@ -223 +235 @@
         super(DelayCodeTable, self).__init__()
 
+        self._delayCodeData = None
+
         self._treeView = None
 
-        self._listStore = gtk.ListStore(str, str)
-        self._treeView = gtk.TreeView(self._listStore)
+        self._treeView = gtk.TreeView(gtk.ListStore(str, str))
         self._treeView.set_rules_hint(True)
 
@@ -232 +245 @@
 
         self._alignments = []
+        self._checkButtons = []
 
         self._eventBox = gtk.EventBox()
@@ -244 +258 @@
 
         self._previousWidth = 0
+
+    @property
+    def delayCodes(self):
+        """Get a list of the delay codes checked by the user."""
+        codes = []
+
+        if self._delayCodeData is not None:
+            codeExtractor = self._delayCodeData[0]
+            for checkButton in self._checkButtons:
+                if checkButton.get_active():
+                    codes.append(codeExtractor(checkButton.delayCodeRow))
+
+        return codes
 
     def allocate_column_sizes(self, allocation):
@@ -263 +290 @@
             data = _data2
 
+        self._delayCodeData = data
+
         columns = self._treeView.get_columns()
         for column in columns:
             self._treeView.remove_column(column)
 
-        (headers, rows) = data
+        (_extractor, headers, rows) = data
         numColumns = len(headers) + 1
         numRows = len(rows)
@@ -287 +316 @@
 
         self._alignments = []
+        self._checkButtons = []
 
         firstDelayCodeRow = True
@@ -301 +331 @@
                 self._table.set_row_spacing(i, 8)
             elif type==DELAYCODE:
+                checkButton = CheckButton(elements)
+                self._checkButtons.append(checkButton)
                 alignment = Alignment(xalign = 0.5, yalign = 0.5, xscale = 1.0)
-                alignment.add(gtk.CheckButton())
+                alignment.add(checkButton)
                 self._table.attach(alignment, 0, 1, i, i+1)
                 if firstDelayCodeRow: