Skip to content
Snippets Groups Projects
Commit 90b6b5b5 authored by Luke Campagnola's avatar Luke Campagnola
Browse files

python 3 fixes

parent 19be6959
No related branches found
No related tags found
No related merge requests found
from pyqtgraph.pgcollections import OrderedDict from pyqtgraph.pgcollections import OrderedDict
from Node import Node from .Node import Node
def isNodeClass(cls): def isNodeClass(cls):
try: try:
......
...@@ -19,11 +19,13 @@ getNodeTree = LIBRARY.getNodeTree ...@@ -19,11 +19,13 @@ getNodeTree = LIBRARY.getNodeTree
getNodeType = LIBRARY.getNodeType getNodeType = LIBRARY.getNodeType
# Add all nodes to the default library # Add all nodes to the default library
for modName in ['Data', 'Display', 'Filters', 'Operators']: from . import Data, Display, Filters, Operators
mod = __import__(modName, globals(), locals(), [], -1) for mod in [Data, Display, Filters, Operators]:
#mod = getattr(__import__('', fromlist=[modName], level=1), modName)
#mod = __import__(modName, level=1)
nodes = [getattr(mod, name) for name in dir(mod) if isNodeClass(getattr(mod, name))] nodes = [getattr(mod, name) for name in dir(mod) if isNodeClass(getattr(mod, name))]
for node in nodes: for node in nodes:
LIBRARY.addNodeType(node, [(modName,)]) LIBRARY.addNodeType(node, [(mod.__name__.split('.')[-1],)])
#NODE_LIST = OrderedDict() ## maps name:class for all registered Node subclasses #NODE_LIST = OrderedDict() ## maps name:class for all registered Node subclasses
#NODE_TREE = OrderedDict() ## categorized tree of Node subclasses #NODE_TREE = OrderedDict() ## categorized tree of Node subclasses
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment