A small helper function for finding the node type from tne nuke

[Python]
import nuke

#~ node type finding small function
def find_node_type(nodeList):

“””
Small helper function for find the node type from nuke.

:param nodeList: This should be a list

:type nodeList: list

:return: nodes and node type

:rtype: dict

Example

>>> import find_node_type as fns
>>> node_to_find = [“roto_a”,”write_b”,”shuffle_mode”]
>>> node_found = fns.find_node_type(node_to_find)
>>> print node_found
>>> {‘Roto’: [‘roto_a’], ‘Write’: [‘write_b’], ‘Shuffle’:[‘shuffle_mode’]}

“””

#~ return var and this is dict type
found_nodes = {}

#~ getting all available nodes from the nuke scene file
get_nuke_node_list = nuke.root().nodes()

#~ looping throuh the list provided by user
for each_node in nodeList:

#~ loop throuh the nodes we found in the nuke scene file
for each_nd in get_nuke_node_list:

#~ getting node class type from node
node_class = each_nd.Class()
if each_nd.name().find(each_node) != -1:

#~ if the node is there in the dict then add to the list
#~ or create the key and add value
if found_nodes.has_key(node_class):
found_nodes[node_class].append(each_nd.name())
else:
found_nodes[node_class]=[]
found_nodes[node_class].append(each_nd.name())

#~ return what we found from the scene
return(found_nodes)

[/Python]