浏览代码

visualizer tool enabled and disabled by env var

Joseph Brandenburg 6 月之前
父节点
当前提交
1cecf1eead
共有 2 个文件被更改,包括 5 次插入6 次删除
  1. 5 5
      __init__.py
  2. 0 1
      ops_nodegroup.py

+ 5 - 5
__init__.py

@@ -39,11 +39,11 @@ while (classLists):
 
 interface_classes = []
 
-# try:
-#     from .visualize import MantisVisualizeNode, MantisVisualizeOutput, MantisVisualizeTree
-#     classes.extend([MantisVisualizeTree, MantisVisualizeNode, MantisVisualizeOutput, ])
-# except ImportError:
-#     pass # this feature is optional
+
+from os import environ
+if environ.get("ENABLEVIS"):
+    from .visualize import MantisVisualizeNode, MantisVisualizeOutput, MantisVisualizeTree
+    classes.extend([MantisVisualizeTree, MantisVisualizeNode, MantisVisualizeOutput, ])
 
 import nodeitems_utils
 from nodeitems_utils import NodeCategory, NodeItem

+ 0 - 1
ops_nodegroup.py

@@ -252,7 +252,6 @@ class ExecuteNodeTree(Operator):
         from os import environ
         start_time = time()
         do_profile=False
-        print (environ.get("DOPROFILE"))
         if environ.get("DOPROFILE"):
             do_profile=True
         pass_error = True