X-Git-Url: https://harishankar.org/repos/?p=biaweb_qt.git;a=blobdiff_plain;f=main_window.py;h=bf223101d1426a150adb830ab9ab144e0806c517;hp=32e39f9910728c616659a388c362032407e1d18b;hb=9321b80905bca48d49a6b12a4d463c268736b166;hpb=c78ebaf4ae0e486c54907edf51159615f2b4f81e

diff --git a/main_window.py b/main_window.py
index 32e39f9..bf22310 100644
--- a/main_window.py
+++ b/main_window.py
@@ -9,6 +9,7 @@ import site_configuration_dialog as scd
 import category_dialog as catd
 import article_dialog as artd
 import templates_dialog as tpld
+import generate_dialog as gend
 import biaweb_db
 
 class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow):
@@ -20,7 +21,9 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow):
 	# refresh the category list
 	def repopulate_categories (self):
 		recs = biaweb_db.get_categories (self.current_db)
-		if not recs:
+
+		# check with False because None is returned when no records exist
+		if recs == False:
 			PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in getting the categories")
 			return
 
@@ -32,7 +35,9 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow):
 	# refresh the articles list
 	def repopulate_articles (self, catid=None):
 		recs = biaweb_db.get_articles (self.current_db, catid)
-		if not recs:
+
+		# check with False because None is returned when no records exist
+		if recs == False:
 			PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in getting the articles")
 			return
 
@@ -218,7 +223,7 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow):
 			PyQt4.QtGui.QMessageBox.critical (self, "Error",
 								"Cannot edit templates. You need to create or open a website first")
 		else:
-			tdlg = tpld.TemplatesDialog (self)
+			tdlg = tpld.TemplatesDialog (self, self.current_db)
 			tdlg.exec_ ()
 
 	# function to get the category or article ID from current selected item in a tree widget
@@ -358,6 +363,16 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow):
 				else:
 					PyQt4.QtGui.QMessageBox.critical (self, "Error", "System or SQLite 3 error in creating database")
 
+	# when generate site menu is triggered
+	def onGenerateSite (self):
+		if self.current_db is None:
+			PyQt4.QtGui.QMessageBox.critical (self, "Error",
+								"Cannot generate site. You need to create or open a website first")
+		else:
+			# run the generate site dialog
+			gendlg = gend.GenerateDialog (self, self.current_db)
+			gendlg.exec_ ()
+
 	# file quit is clicked
 	def onFileQuit (self):
 		sys.exit (0)