Added some error checks for sanity

[biaweb_qt.git] / templates_dialog.py

diff --git a/templates_dialog.py b/templates_dialog.py
index 64cc70c..20f2e18 100644 (file)
--- a/templates_dialog.py
+++ b/templates_dialog.py
@@ -16,12 +16,20 @@ class TemplatesDialog (PyQt4.QtGui.QDialog, ui_templates_dialog.Ui_TemplatesDial
 
                # populate the templates
                tpls = biaweb_db.get_templates (self.current_db)
-               for tpl_name, tpl_content in tpls:
-                       self.templates.addItem (str (tpl_name))
+               # if templates populated
+               if tpls:
+                       for tpl_name, tpl_content in tpls:
+                               self.templates.addItem (str (tpl_name))
+               else:
+                       PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in retrieving templates")
 
        def onEdit (self):
                # get the currently selected list item
-               tpl_name = str (self.templates.currentItem ().data (0).toString ())
+               sel_item = self.templates.currentItem ()
+               # if no item is selected return
+               if sel_item is None:
+                       return
+               tpl_name = str (sel_item.data (0).toString ())
 
                # get the template content from the database
                tpl_str = biaweb_db.get_template_text (self.current_db, tpl_name)