X-Git-Url: https://harishankar.org/repos/?p=biaweb_qt.git;a=blobdiff_plain;f=templates_dialog.py;h=db61404f0ef15560705d79afa918ab651eaa6958;hp=efd5bd72e0688b1b989542f8e287b1b0063ee590;hb=1e98086e7040991b7aa6904fbac30f256a84f299;hpb=b517d55320f54925d862aca9dedb363b6d3b156d

diff --git a/templates_dialog.py b/templates_dialog.py
index efd5bd7..db61404 100644
--- a/templates_dialog.py
+++ b/templates_dialog.py
@@ -16,20 +16,37 @@ 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 = 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, str (tpl_name))
+		tpl_str = biaweb_db.get_template_text (self.current_db, tpl_name)
 
 		# if template string cannot be obtained for some reason
-		if tpl_str == False:
+		if tpl_str is False:
 			PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in retrieving template")
 		else:
 			# open edit dialog with title "template name" and text as the template string
 			edlg = editor_dialog.EditorDialog (self, tpl_name, tpl_str)
-			edlg.exec_ ()
+			# if editing is confirmed
+			if edlg.exec_ () == PyQt4.QtGui.QDialog.Accepted:
+				upd_tpl_str = str (edlg.text.toPlainText ()).strip ()
+
+				# update the template with the new template string
+				ret = biaweb_db.update_template (self.current_db, tpl_name, upd_tpl_str)
+				# if update failed
+				if ret is False:
+					PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in updating template")
+