add NODE_PROP_FILE file ext filter

Author: ArnoChenFx

NodeGraphQt/base/model.py

@@ -92,7 +92,7 @@ def __init__(self):
         }
 
     def add_property(self, name, value, items=None, range=None,
-                     widget_type=NODE_PROP, tab='Properties'):
+                     widget_type=NODE_PROP, tab='Properties', ext=None):
         """
         add custom property.
 
@@ -103,6 +103,7 @@ def add_property(self, name, value, items=None, range=None,
             range (tuple)): min, max values used by NODE_PROP_SLIDER.
             widget_type (int): widget type flag.
             tab (str): widget tab name.
+            ext (str) file ext for NODE_PROP_FILE
         """
         tab = tab or 'Properties'
 
@@ -122,6 +123,8 @@ def add_property(self, name, value, items=None, range=None,
                 self._TEMP_property_attrs[name]['items'] = items
             if range:
                 self._TEMP_property_attrs[name]['range'] = range
+            if ext:
+                self._TEMP_property_attrs[name]['ext'] = ext
         else:
             attrs = {self.type_: {name: {
                 'widget_type': widget_type,
@@ -131,6 +134,8 @@ def add_property(self, name, value, items=None, range=None,
                 attrs[self.type_][name]['items'] = items
             if range:
                 attrs[self.type_][name]['range'] = range
+            if ext:
+                attrs[self.type_][name]['ext'] = ext
             self._graph_model.set_node_common_properties(attrs)
 
     def set_property(self, name, value):

NodeGraphQt/base/node.py

@@ -234,7 +234,7 @@ def set_selected(self, selected=True):
         self.set_property('selected', selected)
 
     def create_property(self, name, value, items=None, range=None,
-                        widget_type=NODE_PROP, tab=None):
+                        widget_type=NODE_PROP, tab=None, ext=None):
         """
         Creates a custom property to the node.
 
@@ -247,7 +247,13 @@ def create_property(self, name, value, items=None, range=None,
             - :attr:`NodeGraphQt.constants.NODE_PROP_QCHECKBOX`
             - :attr:`NodeGraphQt.constants.NODE_PROP_QSPINBOX`
             - :attr:`NodeGraphQt.constants.NODE_PROP_COLORPICKER`
-            - :attr:`NodeGraphQt.constants.NODE_PROP_SLIDER`
+            - :attr:`NodeGraphQt.constants.NODE_PROP_FILE`
+            - :attr:`NodeGraphQt.constants.NODE_PROP_VECTOR2`
+            - :attr:`NodeGraphQt.constants.NODE_PROP_VECTOR3`
+            - :attr:`NodeGraphQt.constants.NODE_PROP_VECTOR4`
+            - :attr:`NodeGraphQt.constants.NODE_PROP_FLOAT`
+            - :attr:`NodeGraphQt.constants.NODE_PROP_INT`
+            - :attr:`NodeGraphQt.constants.NODE_PROP_BUTTON`
 
         See Also:
             :class:`NodeGraphQt.PropertiesBinWidget`
@@ -259,8 +265,9 @@ def create_property(self, name, value, items=None, range=None,
             range (tuple)): ``(min, max)`` values used by ``NODE_PROP_SLIDER``
             widget_type (int): widget flag to display in the ``PropertiesBinWidget``
             tab (str): name of the widget tab to display in the properties bin.
+            ext (str): file ext of ``NODE_PROP_FILE``
         """
-        self.model.add_property(name, value, items, range, widget_type, tab)
+        self.model.add_property(name, value, items, range, widget_type, tab, ext)
 
     def properties(self):
         """
@@ -557,9 +564,12 @@ def add_file_input(self, name, label='', text='', tab=None, ext="*"):
             tab (str): name of the widget tab to display in.
             ext (str): file ext
         """
-        self.create_property(
-            name, text, widget_type=NODE_PROP_FILE, tab=tab)
-        widget = NodeFilePath(self.view, name, label, text,ext)
+        self.model.add_property(name, text, None, None, NODE_PROP_FILE, tab, ext)
+        # use create_property will cause strange error
+        # self.create_property(
+        #     name, text, widget_type=NODE_PROP_FILE, tab=tab, ext=None)
+
+        widget = NodeFilePath(self.view, name, label, text, ext)
         widget.value_changed.connect(lambda k, v: self.set_property(k, v))
         self.view.add_widget(widget)
 

NodeGraphQt/widgets/properties.py

@@ -310,9 +310,13 @@ def __init__(self, parent=None):
 
         self._ledit.setStyleSheet("QLineEdit{border:1px solid}")
         _button.setStyleSheet("QPushButton{border:1px solid}")
+        self._ext = "*"
+
+    def set_ext(self, ext):
+        self._ext = ext
 
     def _on_select_file(self):
-        file_path = file_dialog.getOpenFileName(self)
+        file_path = file_dialog.getOpenFileName(self, ext_filter=self._ext)
         file = file_path[0] or None
         if file:
             self.set_value(file)
@@ -868,6 +872,8 @@ def _read_node(self, node):
                         prop_range = common_props[prop_name]['range']
                         widget.set_min(prop_range[0])
                         widget.set_max(prop_range[1])
+                    if 'ext' in common_props[prop_name].keys():
+                        widget.set_ext(common_props[prop_name]['ext'])
 
                 prop_window.add_widget(prop_name, widget, value,
                                        prop_name.replace('_', ' '))