[enums] extract enums to separate file

nixview/data_models/tree_model.py

@@ -1,22 +1,16 @@
 import os
-import nixview.constants as cnst
 from PyQt5.QtCore import QAbstractItemModel, QModelIndex, Qt, QSize
 from PyQt5.QtGui import QIcon
 from PyQt5.QtWidgets import QTreeView, QTreeWidgetItem, QAbstractItemView, QHeaderView
-from enum import Enum
 
+import nixview.constants as cnst
 from nixview.util.file_handler import NodeType
 from nixview.util.descriptors import ItemDescriptor
+from nixview.util.enums import TreeType
 
 column_names = ['Name', 'Type', 'Value', 'Description', 'ID',]
 
 
-class TreeType(Enum):
-     Full = "full view"
-     Metadata = "metadata view"
-     Data = "data view"
-
-
 class NixTreeItem(QTreeWidgetItem):
     def __init__(self, node_descriptor, file_handler, parent=None):
         super().__init__(parent)

nixview/file_utils.py

@@ -1,11 +1,4 @@
 import nixio as nix
-from enum import Enum
-
-class PlotterTypes(Enum):
-     ImagePlotter = "image"
-     LinePlotter = "line"
-     CategoryPlotter = "category"
-     EventPlotter = "event"
      
 
 def guess_best_xdim(array):

nixview/util/enums.py

@@ -0,0 +1,30 @@
+from enum import Enum
+
+
+class NodeType(Enum):
+    Root = "root"
+    Section = "Section"
+    Block = "Block"
+    DataArray = "Data Array"
+    DataFrame = "Data Frame"
+    Property = "Property"
+    Dimension = "Dimension"
+    Source = "Source"
+    Tag = "Tag"
+    MultiTag = "Multi Tag"
+    Group = "Group"
+    Feature="Feature"
+
+
+class TreeType(Enum):
+    Full = "full view"
+    Metadata = "metadata view"
+    Data = "data view"
+
+
+class PlotterTypes(Enum):
+    ImagePlotter = "image"
+    LinePlotter = "line"
+    CategoryPlotter = "category"
+    EventPlotter = "event"
+

nixview/util/file_handler.py

@@ -1,23 +1,9 @@
 import os
 import nixio as nix
 import numpy as np
-from enum import Enum
 
 from nixview.util.descriptors import FileDescriptor, ItemDescriptor
-
-class NodeType(Enum):
-     Root = "root"
-     Section = "Section"
-     Block = "Block"
-     DataArray = "Data Array"
-     DataFrame = "Data Frame"
-     Property = "Property"
-     Dimension = "Dimension"
-     Source = "Source"
-     Tag = "Tag"
-     MultiTag = "Multi Tag"
-     Group = "Group"
-     Feature="Feature"
+from nixview.util.enums import NodeType