Add preliminary file tree

Author: jyapayne

main.py

@@ -37,6 +37,7 @@
 from util_classes import ExistingProjectDialog
 from util_classes import BackgroundThread, Validator
 from util_classes import CompleterLineEdit, TagsCompleter
+from util_classes import TreeBrowser
 
 from PySide import QtGui, QtCore
 from PySide.QtGui import (QApplication, QHBoxLayout, QVBoxLayout)
@@ -989,6 +990,9 @@ def load_project(self, directory, readonly=False):
 
         self.set_window_icon()
         self.open_export_button.setEnabled(True)
+
+        self.tree_browser.init(directory, ['.*'])
+
         self.update_json = True
 
     def init_main_field(self, directory):
@@ -1159,23 +1163,48 @@ def create_export_settings(self):
 
         ex_setting_order = self.settings['order']['export_setting_order']
 
-        vlayout = self.create_layout(ex_setting_order, cols=4)
+        vlayout = self.create_layout(ex_setting_order, cols=1)
 
         output_name_layout = self.create_output_name_pattern_line()
 
         output_layout = self.create_output_directory_line()
 
         script_layout = self.create_script_layout()
 
+        hlayout = QtGui.QHBoxLayout()
+
+        platform_group = QtGui.QGroupBox('Platforms')
+        playout = QtGui.QVBoxLayout()
+        playout.addLayout(vlayout)
+        platform_group.setLayout(playout)
+
+        hlayout.addWidget(platform_group)
+
+        tree_layout = self.create_blacklist_layout()
+
+        tree_group = QtGui.QGroupBox('Files')
+        tree_group.setLayout(tree_layout)
+
+        hlayout.addWidget(tree_group)
+
         vbox = QtGui.QVBoxLayout()
-        vbox.addLayout(vlayout)
+        vbox.addLayout(hlayout)
         vbox.addLayout(output_name_layout)
         vbox.addLayout(output_layout)
         vbox.addLayout(script_layout)
 
         group_box.setLayout(vbox)
         return group_box
 
+    def create_blacklist_layout(self):
+        blacklist_layout = QtGui.QVBoxLayout()
+
+        self.tree_browser = TreeBrowser()
+
+        blacklist_layout.addWidget(self.tree_browser)
+
+        return blacklist_layout
+
     def create_output_name_pattern_line(self):
         output_name_layout = QtGui.QHBoxLayout()
 

util_classes.py

@@ -9,6 +9,129 @@
 import utils
 
 from PySide import QtGui, QtCore
+from PySide.QtCore import Qt
+
+class FileItem(QtGui.QTreeWidgetItem):
+    def __init__(self, parent=None, path=None):
+        super(FileItem, self).__init__(parent)
+        self.path = path
+
+class TreeBrowser(QtGui.QWidget):
+    def __init__(self, directory=None, checked_files=None,
+                 whitelist=None, blacklist=None, parent=None):
+        super(TreeBrowser, self).__init__(parent=parent)
+        self.root = QtGui.QTreeWidget()
+        self.root.setHeaderLabel('Included files')
+        self.root.itemChanged.connect(self.item_changed)
+        self.files = {}
+
+        self.paths = []
+
+        layout = QtGui.QVBoxLayout()
+        layout.addWidget(self.root)
+        self.setLayout(layout)
+
+        self.watcher = QtCore.QFileSystemWatcher()
+        self.watcher.directoryChanged.connect(self.directory_changed)
+        self.watcher.fileChanged.connect(self.file_changed)
+
+        self.directoryChanged = self.watcher.directoryChanged
+        self.fileChanged = self.watcher.fileChanged
+
+        self.init(directory, checked_files, whitelist, blacklist)
+
+    def init(self, directory=None, checked_files=None,
+             whitelist=None, blacklist=None):
+
+        if directory:
+            self.directory = directory + os.sep
+        else:
+            self.directory = directory
+
+        self.checked_files = checked_files or []
+
+        self.whitelist = whitelist or []
+        self.blacklist = blacklist or []
+
+        self.watcher.removePaths(self.watcher.files())
+
+        self.files = {}
+
+        self.root.clear()
+
+        self.generate_directory_widget()
+
+    def file_changed(self, path):
+        print(path)
+        pass
+
+    def directory_changed(self, path):
+        print(path)
+        pass
+
+    def get_abs_file_list(self):
+        return [os.path.join(self.directory, path) for path in self.files.keys()]
+
+    def get_checked_files(self):
+        pass
+
+    def item_changed(self, item, column):
+        self.files[item.path] = item.checkState(column)
+
+    def generate_directory_widget(self):
+        if self.directory is None:
+            return
+
+        parent_map = {'': self.root}
+
+        for root, dirs, files in os.walk(self.directory):
+            for directory in dirs:
+
+                proj_path = root.replace(self.directory, '')
+
+                parent = parent_map[proj_path]
+
+                path = os.path.join(proj_path, directory)
+
+                checked = Qt.Unchecked
+
+                for checked_file in self.checked_files:
+                    match = re.match('^'+checked_file, path)
+                    if match:
+                        checked = Qt.Checked
+
+                child = FileItem(parent, path)
+                child.setFlags(child.flags() | Qt.ItemIsTristate | Qt.ItemIsUserCheckable)
+                child.setText(0, directory)
+                child.setCheckState(0, checked)
+
+                self.files[path] = checked
+
+                parent_map[path] = child
+
+            for file in files:
+                proj_path = root.replace(self.directory, '')
+
+                parent = parent_map[proj_path]
+
+                path = os.path.join(proj_path, file)
+
+                checked = Qt.Unchecked
+
+                for checked_file in self.checked_files:
+                    match = re.match(checked_file, path)
+                    if match:
+                        checked = Qt.Checked
+
+                child = FileItem(parent, path)
+                child.setFlags(child.flags() | Qt.ItemIsUserCheckable)
+                child.setText(0, file)
+                child.setCheckState(0, checked)
+
+                self.files[path] = checked
+
+        self.watcher.addPaths(self.get_abs_file_list())
+
 
 class ExistingProjectDialog(QtGui.QDialog):
     def __init__(self, recent_projects, directory_callback, parent=None):
@@ -93,6 +216,7 @@ def project_clicked(self, _):
     def cancelled(self):
         self.close()
 
+
 class Validator(QtGui.QRegExpValidator):
     def __init__(self, regex, action, parent=None):
         self.exp = regex