From b388c7c68c4c32bc3c2a2d3e0ecdcf8847531595 Mon Sep 17 00:00:00 2001
From: Jan Grewe <jan.grewe@g-node.org>
Date: Sun, 17 Jan 2021 23:42:49 +0100
Subject: [PATCH] [enums] extract enums to separate file

---
 nixview/data_models/tree_model.py | 10 ++--------
 nixview/file_utils.py             |  7 -------
 nixview/util/enums.py             | 30 ++++++++++++++++++++++++++++++
 nixview/util/file_handler.py      | 16 +---------------
 4 files changed, 33 insertions(+), 30 deletions(-)
 create mode 100644 nixview/util/enums.py

diff --git a/nixview/data_models/tree_model.py b/nixview/data_models/tree_model.py
index 25abca0..e1f191c 100644
--- a/nixview/data_models/tree_model.py
+++ b/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)
diff --git a/nixview/file_utils.py b/nixview/file_utils.py
index 44af2a5..e73e5bf 100644
--- a/nixview/file_utils.py
+++ b/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):
diff --git a/nixview/util/enums.py b/nixview/util/enums.py
new file mode 100644
index 0000000..0937c61
--- /dev/null
+++ b/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"
+
diff --git a/nixview/util/file_handler.py b/nixview/util/file_handler.py
index e719f21..a39a3ea 100644
--- a/nixview/util/file_handler.py
+++ b/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