Source code for nukescripts.searchreplace

# Copyright (c) 2009 The Foundry Visionmongers Ltd.  All Rights Reserved.

import nuke_internal as nuke
import re

def __NodeHasKnobWithName(node, name):
  try:
    node[name]
  except NameError:
    return False
  else:
    return True

def __NodeHasFileKnob(node):
  return __NodeHasKnobWithName(node, 'file')

def __NodeHasProxyKnob(node):
  return __NodeHasKnobWithName(node, 'proxy')

def __ReplaceKnobValue(searchstr, replacestr, knob):
  v = knob.value()
  if v:
    repl = re.sub(searchstr, replacestr, v)
    knob.setValue(repl)

def search_replace():
  """ Search/Replace in Reads and Writes. """
  fileKnobNodes = [i for i in nuke.selectedNodes(recursive=True) if __NodeHasFileKnob(i)]
  proxyKnobNodes = [i for i in nuke.selectedNodes(recursive=True) if __NodeHasProxyKnob(i)]
  if not fileKnobNodes and not proxyKnobNodes: raise ValueError("No nodes selected")

  p = nuke.Panel("Search/Replace in Reads and Writes")
  p.addSingleLineInput("Search for:", "rgbea")
  p.addSingleLineInput("Replace with:", "rgbea")
  success = p.show()
  if success == 1:
    searchstr = p.value("Search for:")
    replacestr = p.value("Replace with:")

    for i in fileKnobNodes: __ReplaceKnobValue(searchstr, replacestr, i['file'])
    for i in proxyKnobNodes: __ReplaceKnobValue(searchstr, replacestr, i['proxy'])