Redshift rewire

import hou

def main():
    if not hou.selectedNodes():
        print ("No selected node")
        return
    curnode=hou.selectedNodes()[0]
    if  curnode.type().category().name()!='Vop':
        print("No Shop selected!")
        return
    result = None
    for node in curnode.parent().children():
        if node.type().name() == "redshift_material":
            result = node
    result.setInput(0,curnode,0)

main()