X-Git-Url: https://harishankar.org/repos/?p=biaweb_qt.git;a=blobdiff_plain;f=main_window.py;h=799fdf8b02d10a6b317fc82f03e97131fc390b68;hp=3e70b10f4c95f8e276972100ed9d649a797ff43f;hb=cd074c8da38f6cc0ad808783ca7e631967dfdd47;hpb=5f4cc1e3f6bf2ffbf624dfc19d24c1b4eb2e15db diff --git a/main_window.py b/main_window.py index 3e70b10..799fdf8 100644 --- a/main_window.py +++ b/main_window.py @@ -2,12 +2,15 @@ # Main Window class import PyQt4 +import sys + import ui_main_window 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 -import sys class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow): def __init__ (self): @@ -18,26 +21,35 @@ 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 + return False self.categories.clear () for (id, name, desc, stub) in recs: qrow = PyQt4.QtGui.QTreeWidgetItem ([str(id), str(name)]) self.categories.addTopLevelItem (qrow) + # return true when successful + return True + # 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 + return False self.articles.clear () for item in recs: qrow = PyQt4.QtGui.QTreeWidgetItem ([str(item[0]), str(item[1])]) self.articles.addTopLevelItem (qrow) + # return true when successful + return True # when the view is refreshed def onViewRefresh (self): @@ -67,14 +79,14 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow): keywords = str (artdlg.keywords.text ()).strip () summary = str (artdlg.summary.toPlainText ()).strip () content = str (artdlg.content.toPlainText ()).strip () - catid, ok = artdlg.category.itemData (artdlg.category.currentIndex ()).toInt () + cid, ok = artdlg.category.itemData (artdlg.category.currentIndex ()).toInt () # if catid is not an integer then abort if not ok: return rating = artdlg.rating.value () stub = str (artdlg.stub.text ()).strip () - ret = biaweb_db.create_article (self.current_db, title, summary, keywords, content, catid, + ret = biaweb_db.create_article (self.current_db, title, summary, keywords, content, cid, stub, rating) if ret: PyQt4.QtGui.QMessageBox.information (self, "Success", "Article successfully created") @@ -85,7 +97,54 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow): # when edit article is triggered def onArticleEdit (self): - pass + if self.current_db is None: + PyQt4.QtGui.QMessageBox.critical (self, "Error", + "Cannot edit article. You need to create or open a website first") + else: + # get the selected article + artid = self.get_selected_item_id (self.articles) + catid = self.get_selected_item_id (self.categories) + + # no article is selected + if artid is None: + PyQt4.QtGui.QMessageBox.critical (self, "Error", "No article selected") + return + article = biaweb_db.get_article (self.current_db, artid) + # if article cannot be read + if not article: + PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in reading article") + return + artdlg = artd.ArticleDialog (self) + + artdlg.article_title.setText (article[1]) + artdlg.summary.setPlainText (article[2]) + artdlg.keywords.setText (article[3]) + artdlg.content.setPlainText (article[4]) + # populate category combo box and set active category to the article's category + cats = biaweb_db.get_categories (self.current_db) + artdlg.populate_categories (cats, article[7]) + artdlg.stub.setText (article[8]) + artdlg.rating.setValue (article[9]) + + # ok is pressed + if artdlg.exec_ () == PyQt4.QtGui.QDialog.Accepted: + title = str (artdlg.article_title.text ()).strip () + summary = str (artdlg.summary.toPlainText ()).strip () + keywords = str (artdlg.keywords.text ()).strip () + content = str (artdlg.content.toPlainText ()).strip () + cid, ok = artdlg.category.itemData (artdlg.category.currentIndex()).toInt () + # if cat id is not an integer + if not ok: + return + rating = artdlg.rating.value () + stub = str (artdlg.stub.text ()).strip () + ret = biaweb_db.update_article (self.current_db, artid, title, + summary, keywords, content, cid, stub, rating) + if ret: + PyQt4.QtGui.QMessageBox.information (self, "Success", "Article successfully updated") + self.repopulate_articles (catid) + else: + PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in updating article") # when delete article is triggered def onArticleDelete (self): @@ -100,11 +159,11 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow): PyQt4.QtGui.QMessageBox.critical (self, "Error", "No article selected") return # get confirmation on delete - conf = PyQt4.QtGui.QMessageBox.question (self, "Confirm", + flag = PyQt4.QtGui.QMessageBox.question (self, "Confirm", "Are you sure you wish to delete the selected article?", PyQt4.QtGui.QMessageBox.Yes, PyQt4.QtGui.QMessageBox.No) # confirmed - if conf == PyQt4.QtGui.QMessageBox.Yes: + if flag == PyQt4.QtGui.QMessageBox.Yes: ret = biaweb_db.delete_article (self.current_db, artid) if not ret: PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in deleting article") @@ -125,7 +184,7 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow): catid = self.get_selected_item_id (self.categories) self.repopulate_articles (catid) - # when configuration menu is activated + # when configuration menu is triggered def onConfiguration (self): if self.current_db is None: PyQt4.QtGui.QMessageBox.critical (self, "Error", @@ -163,6 +222,15 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow): PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in updating configuration") + # when templates menu is triggered + def onTemplates (self): + if self.current_db is None: + PyQt4.QtGui.QMessageBox.critical (self, "Error", + "Cannot edit templates. You need to create or open a website first") + else: + 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 def get_selected_item_id (self, twidget): selitems = twidget.selectedItems () @@ -190,7 +258,7 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow): cat = biaweb_db.get_category (self.current_db, catid) # if the category cannot be retrieved from database if not cat: - PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error reading category") + PyQt4.QtGui.QMessageBox.critical (self, "Error", "SQLite 3 error in reading category") return # set the data in the dialog @@ -267,9 +335,21 @@ class MainWindow (PyQt4.QtGui.QMainWindow, ui_main_window.Ui_MainWindow): filename = PyQt4.QtGui.QFileDialog.getOpenFileName (self, "Open Site Database") if filename: self.current_db = str (filename) - self.setWindowTitle ("BiaWeb - " + self.current_db) - self.repopulate_categories () - self.repopulate_articles () + + # added to check whether categories are loaded successfully or not + # if not, then reset the current_db to None + loaded_cats = self.repopulate_categories () + loaded_arts = self.repopulate_articles () + # if failed in loading either categories or articles (note: checking against + # False and not None) + if loaded_cats == False or loaded_arts == False: + self.current_db = None + PyQt4.QtGui.QMessageBox.critical (self, "Error", + "SQLite 3 error in loading site database. This appears to be an invalid BiaWeb database") + else: + # set the window title to the database + self.setWindowTitle ("BiaWeb - " + self.current_db) + # file new menu is clicked def onFileNew (self): @@ -300,6 +380,23 @@ 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_ () + + # about menu is triggered + def onAbout (self): + PyQt4.QtGui.QMessageBox.about (self, "BiaWeb Qt", + "A static website/weblog content management system

\ +Copyright © 2010 Harishankar
\ +Licensed under GNU/GPL v3") + # file quit is clicked def onFileQuit (self): sys.exit (0)