X-Git-Url: https://harishankar.org/repos/?p=biaweb_qt.git;a=blobdiff_plain;f=generate_dialog.py;fp=generate_dialog.py;h=d1882d2d355b7b71bfe4c9948e1e3f4ef6da3bb9;hp=eb49933ea6cac02c9f79b9f52540f0d36d39f508;hb=bfa2d7e7b048794a95590ee79ba41aa861835cfa;hpb=61047f72eb7d61e1c0efd7f22729e16fbfb97505

diff --git a/generate_dialog.py b/generate_dialog.py
index eb49933..d1882d2 100644
--- 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,