Added about box and application icon

[biaweb_qt.git] / generate_dialog.py

diff --git a/generate_dialog.py b/generate_dialog.py
index eb49933..d1882d2 100644 (file)
--- a/generate_dialog.py
+++ b/generate_dialog.py
@@ -14,6 +14,31 @@ class GenerateDialog (PyQt4.QtGui.QDialog, ui_generate_dialog.Ui_SiteGenerateDia
                # set the database
                self.current_db = currentdb
 
+       # to return the list of items from a tree widget
+       def get_list_from_tree (self, treewidget):
+               lstfiles = []
+               iter = PyQt4.QtGui.QTreeWidgetItemIterator (treewidget)
+
+               # iterate through the list of files
+               item = iter.value ()
+               # while there is still a valid item
+               while item:
+                       # get the text in columns 0 and 1
+                       src = str (item.text (0))
+                       dst = str (item.text (1))
+                       # add it to the list
+                       lstfiles.append ( (src, dst) )
+                       # increase iterator by 1
+                       iter += 1
+                       item = iter.value ()
+
+               return lstfiles
+
+       # when site generate button is clicked
+       def onSiteGenerate (self):
+               files_list = self.get_list_from_tree (self.additional_files)
+               folder_list = self.get_list_from_tree (self.additional_folders)
+
        # when folder add is clicked
        def onFolderAdd (self):
                folder_to_add = PyQt4.QtGui.QFileDialog.getExistingDirectory (self,