tooledit: corrected data type for column 'orient' to int, add error message when loading invalid value

Author: hansu

lib/python/gladevcp/tooledit_gtk.glade

@@ -35,7 +35,7 @@
       <!-- column-name back -->
       <column type="gchararray"/>
       <!-- column-name orient -->
-      <column type="gchararray"/>
+      <column type="gint"/>
       <!-- column-name comments -->
       <column type="gchararray"/>
     </columns>

lib/python/gladevcp/tooledit_widget.py

@@ -226,7 +226,7 @@ def match_tool(model, path, iter, pathlist):
         except:
             print(_("tooledit_widget error: cannot select tool number"),toolnumber)
 
-    def add(self,widget,data=[1,0,0,'0','0','0','0','0','0','0','0','0','0','0','0','0',"comment"]):
+    def add(self,widget,data=[1,0,0,'0','0','0','0','0','0','0','0','0','0','0','0',0,"comment"]):
         self.model.append(data)
         self.num_of_col +=1
 
@@ -235,6 +235,17 @@ def set_filename(self,filename):
         self.toolfile = filename
         self.reload(None)
 
+    def warning_dialog(self, line_number):
+        message = _("Error in tool table line {0} in column ").format(line_number) + _("Orientation") + \
+            _(".\nValid range is 0 - 9.")
+        dialog = Gtk.MessageDialog(self.wTree.get_object("window1"),
+            Gtk.DialogFlags.DESTROY_WITH_PARENT,
+            Gtk.MessageType.ERROR, Gtk.ButtonsType.OK,
+            message)
+        dialog.show()
+        dialog.run()
+        dialog.destroy()
+
         # Reload the tool file into display
     def reload(self,widget):
         self.hash_code = self.md5sum(self.toolfile)
@@ -247,18 +258,23 @@ def reload(self,widget):
             return
         logfile = open(self.toolfile, "r").readlines()
         self.toolinfo = []
+        line_number = 0
         for rawline in logfile:
             # strip the comments from line and add directly to array
             # if index = -1 the delimiter ; is missing - clear comments
             index = rawline.find(";")
+            # skip lines beginning with a semicolon
+            if index == 0:
+                continue
             comment =''
             if not index == -1:
                 comment = (rawline[index+1:])
                 comment = comment.rstrip("\n")
                 line = rawline.rstrip(comment)
             else:
                 line = rawline
-            array = [0,0,0,'0','0','0','0','0','0','0','0','0','0','0','0','0',comment]
+            line_number += 1
+            array = [0,0,0,'0','0','0','0','0','0','0','0','0','0','0','0',0,comment]
             toolinfo_flag = False
             # search beginning of each word for keyword letters
             # offset 0 is the checkbutton so ignore it
@@ -278,11 +294,21 @@ def reload(self,widget):
                                 array[offset]= int(word.lstrip(i))
                             except:
                                 print(_("Tooledit widget int error"))
+                        elif offset == 15:
+                            try:
+                                # Accept also float for 'orientation' for backward compatibility
+                                value = int(float(word.lstrip(i)))
+                                array[offset] = value
+                                if value not in range(10):
+                                    self.warning_dialog(line_number)
+                                    break
+                            except:
+                                print(_("Tooledit widget float error"))
                         else:
                             try:
                                 array[offset]= locale.format("%10.4f", float(word.lstrip(i)))
                             except:
-                                print(_("Tooledit_widget float error"))
+                                print(_("Tooledit widget float error"))
                         break
             if toolinfo_flag:
                 self.toolinfo = array
@@ -292,16 +318,26 @@ def reload(self,widget):
         # Note we have to save the float info with a decimal even if the locale uses a comma
     def save(self,widget):
         if self.toolfile == None:return
+        liststore = self.model
+        # pre check before saving the file
+        # if not done before, the file will be saved only until the erroneous line and the rest will be lost
+        line_number = 0
+        for row in liststore:
+            values = [ value for value in row ]
+            line_number += 1
+            if values[15] > 9:
+                self.warning_dialog(line_number)
+                return
+
         file = open(self.toolfile, "w")
         #print self.toolfile
-        liststore = self.model
         for row in liststore:
             values = [ value for value in row ]
             #print values
             line = ""
             for num,i in enumerate(values):
                 if num == 0: continue
-                elif num in (1,2): # tool# pocket#
+                elif num in (1,2,15): # tool#, pocket#, orientation
                     line = line + "%s%d "%(KEYWORDS[num], i)
                 elif num == 16: # comments
                     test = i.strip()
@@ -421,17 +457,26 @@ def col_editted(self, widget, path, new_text, col, filter):
         elif filter == 'tool':
             (store_path,) = self.tool_filter.convert_path_to_child_path(path)
             path = store_path
-        
+
         if col in(1,2):
             try:
                 self.model[path][col] = int(new_text)
             except:
                 pass
-        elif col in range(3,16):
+        # validate input for float columns
+        elif col in range(3,15):
             try:
                 self.model[path][col] = locale.format("%10.4f",locale.atof(new_text))
             except:
                 pass
+        # validate input for orientation: check if int and valid range
+        elif col == 15:
+            try:
+                value = int(new_text)
+                if value in range(10):
+                    self.model[path][col] = value
+            except:
+                pass
         elif col == 16:
             try:
                 self.model[path][col] = (new_text)