2021-07-12

sceneControl: added 'removeKellerTab' function.

pythonpluginsUI/sceneControl/sceneControl.py

 WRITENODE_VERSION = 1.0
 USE_ENV_PROJECT_ROOT = 1
 WRITENODE_LABEL = 'exr compression: [value compression]'
+KELLERTABKNOBS = (
+    'extension', 'ext', 'ver', 'explore', 'divider3', 'playinrv', 'pushtorv', 'pushtorvappend', 'divider2', 'ignore',
+    'divider1', 'set',
+    'info', 'element', 'subtask', 'task', 'w_reload', 'useshotenv', 'keller')
+
 
 # globals
 sequenceGlobal=''
         self.menu = nuke.menu('Nuke')
         m = self.menu.addMenu('SceneControl')
         m.addCommand('SceneControl', 'import sceneControl; sceneControl.createSceneControl()',shortcut='F12')
-        m.addCommand('Increment Major', 'import sceneControl; sceneControl.increment_major_version()', tooltip="Increment Major version and save script.", shortcut='alt+shift+s')
-        m.addCommand('Increment Minor', 'import sceneControl; sceneControl.increment_minor_version()', tooltip="Increment Minor version and save script.", shortcut='alt+s')
+        m.addCommand('Increment Major', 'import sceneControl; sceneControl.increment_major_version()', tooltip='Increment Major version and save script.', shortcut='alt+shift+s')
+        m.addCommand('Increment Minor', 'import sceneControl; sceneControl.increment_minor_version()', tooltip='Increment Minor version and save script.', shortcut='alt+s')
         m.addCommand('Reload Config', 'import sceneControl; sceneControl.kenvDelete()')
 
-        m.addCommand('Helpers/Read/Set Missing Frames Mode', 'import sceneControl; sceneControl.setErrorMode()', tooltip="Sets the Error Mode on selected Read nodes.")
-        m.addCommand('Helpers/Read/Set colorspace on Nodes', 'import sceneControl; sceneControl.setOcioRole()', tooltip="Sets the OCIO role on selected Read / Write nodes.")
-        m.addCommand('Helpers/Read/Reload Reads', 'import sceneControl; sceneControl.refresh_read_nodes()', tooltip="Reload footage of selected Read / Write nodes.")
+        m.addCommand('Helpers/Read/Set Missing Frames Mode', 'import sceneControl; sceneControl.setErrorMode()', tooltip='Sets the Error Mode on selected Read nodes.')
+        m.addCommand('Helpers/Read/Set colorspace on Nodes', 'import sceneControl; sceneControl.setOcioRole()', tooltip='Sets the OCIO role on selected Read / Write nodes.')
+        m.addCommand('Helpers/Read/Reload Reads', 'import sceneControl; sceneControl.refresh_read_nodes()', tooltip='Reload footage of selected Read / Write nodes.')
         #
         m.addCommand('Helpers/Write/Ignore selected Nodes', 'import sceneControl; sceneControl.setIgnoreFlag(True)')
         m.addCommand('Helpers/Write/Manage selected Nodes','import sceneControl; sceneControl.setIgnoreFlag(False)')
-        m.addCommand('Helpers/Write/Set colorspace on Nodes', 'import sceneControl; sceneControl.setOcioRole()', tooltip="Sets the OCIO role on selected Read / Write nodes.")
-        m.addCommand('Helpers/Write/Update Write Nodes Tab', 'import sceneControl; sceneControl.updateWriteNodeConfig()', tooltip="Updates Keller Tab and functions of selected write nodes.")
+        m.addCommand('Helpers/Write/Set colorspace on Nodes', 'import sceneControl; sceneControl.setOcioRole()', tooltip='Sets the OCIO role on selected Read / Write nodes.')
+        m.addCommand('Helpers/Write/Update Write Nodes Tab', 'import sceneControl; sceneControl.updateWriteNodeConfig()', tooltip='Updates Keller Tab and functions of selected write nodes.')
+        m.addCommand('Helpers/Write/Remove Write Nodes Tab', 'import sceneControl; sceneControl.removeKellerTab()', tooltip='Removes Keller Tab on selected Write Nodes.')
 
         #internal
         m.addCommand('Debug/Enable Debug Mode', 'import sceneControl; sceneControl.setDebug(True)')
         self.menu.removeItem('Helpers/Write/Manage selected Nodes')
         self.menu.removeItem('Helpers/Write/Set colorspace on Nodes')
         self.menu.removeItem('Helpers/Write/Update Write Nodes Tab')
+        self.menu.removeItem('Helpers/Write/Remove Write Nodes Tab')
         #
         self.menu.removeItem('Debug/Enable Debug Mode')
         self.menu.removeItem('Debug/Disable Debug Mode')
             hexColour = int('%02x%02x%02x%02x' % (1 * 255, 0.666 * 255, 0.266 * 255, 1), 16)
             n['tile_color'].setValue(hexColour)
 
-
 def onCreateWrite():
     n = nuke.thisNode()
     createCustomWrite(n)
     return
 
-
 def onDestroy():
     print('SceneControl: Info. Nice meeting you. Removing Kenv now...')
     try:
 
 # SceneControl
 def getSceneControl():
-    c = nuke.exists("sceneCtrl")
+    c = nuke.exists('sceneCtrl')
     if c:
         sc = nuke.toNode('sceneCtrl')
     else:
 
 
 def createSceneControlUI():
-    n = nuke.Node("NoOp")
-    n.setName("sceneCtrl")
+    n = nuke.Node('NoOp')
+    n.setName('sceneCtrl')
     n['hide_input'].setValue(True)
 
     # % (r * 255, g * 255, b * 255, 1), 16): change r g and b to your liking
     hexColour = int('%02x%02x%02x%02x' % (0.2 * 255, 0.478 * 255, 0.647 * 255, 1), 16)
     n['tile_color'].setValue(hexColour)
-    n['label'].setValue("_\n__")
+    n['label'].setValue('_\n__')
 
     # ###########################
     # ---------------------------
     k = nuke.Tab_Knob('main', 'Main')
     n.addKnob(k)
 
-    # Scriptname
-    #k = nuke.String_Knob("scriptname", "Script")
-    #n.addKnob(k)
-    #n.knob("scriptname").setValue(scriptname)
-    #n.knob("scriptname").setEnabled(False)
-
     # Project
     k = nuke.String_Knob('project', 'Project')
     k.setTooltip('Project Shortname')
     n.knob('element').clearFlag(nuke.STARTLINE)
 
     # Update from KEnv
-    k = nuke.PyScript_Knob("reloadConfig", "Reload", "sceneControl.kenvDelete()")
-    k.setTooltip("Reload project config file.")
+    k = nuke.PyScript_Knob('reloadConfig', 'Reload', 'sceneControl.kenvDelete()')
+    k.setTooltip('Reload project config file.')
     n.addKnob(k)
 
     # Season
     n.knob('episode').setVisible(False)
 
     # _______________________________________________________________
-    k = nuke.Text_Knob("divider1", "")
+    k = nuke.Text_Knob('divider1', '')
     n.addKnob(k)
     # _______________________________________________________________
 
     # Sequence
     k = nuke.String_Knob('sequence', 'Seq')
     k.setTooltip(
-        "can be a sequence number or gen when doing generic tasks.\n\ncomp_001_010.comp_v01.ms.nk\ncomp_gen.footageIngest_v01.ms.nk")
+        'can be a sequence number or gen when doing generic tasks.\n\ncomp_001_010.comp_v01.ms.nk\ncomp_gen.footageIngest_v01.ms.nk')
     n.addKnob(k)
     n.knob('sequence').setValue(seqDef)
 
     # Shot
     k = nuke.String_Knob('shot', 'Shot')
     k.setTooltip(
-        "can be a shot number or gen in a multiShot context.\n\ncomp_001_010.comp_v01.ms.nk\ncomp_001_gen.comp_v01.ms.nk")
+        'can be a shot number or gen in a multiShot context.\n\ncomp_001_010.comp_v01.ms.nk\ncomp_001_gen.comp_v01.ms.nk')
     n.addKnob(k)
     n.knob('shot').setValue(shotDef)
     n.knob('shot').clearFlag(nuke.STARTLINE)
     n.knob('user').clearFlag(nuke.STARTLINE)
 
     # Divider _______________________________________________________________
-    k = nuke.Text_Knob("divider2", "")
+    k = nuke.Text_Knob('divider2', '')
     n.addKnob(k)
     # _______________________________________________________________
 
     k.setTooltip('Save file Preview.')
 
     #  Divider _______________________________________________________________
-    k = nuke.Text_Knob("divider3", "")
+    k = nuke.Text_Knob('divider3', '')
     n.addKnob(k)
     # _______________________________________________________________
 
     n.knob('fEnd').clearFlag(nuke.STARTLINE)
 
     # Button Get Framerange (ftrack)
-    k = nuke.PyScript_Knob("get_framerange", "Get Range", "sceneControl.get_framerange(nuke.thisNode())")
+    k = nuke.PyScript_Knob('get_framerange', 'Get Range', 'sceneControl.get_framerange(nuke.thisNode())')
     n.addKnob(k)
-    n.knob("get_framerange").setTooltip("Get Framerange from FTrack.\nPlease check you firewall rules and and allow ftrack ip 104.155.3.128 on port 443")
+    n.knob('get_framerange').setTooltip('Get Framerange from FTrack.\nPlease check you firewall rules and and allow ftrack ip 104.155.3.128 on port 443')
 
     # Button Set Framerange
-    k = nuke.PyScript_Knob("set_framerange", "Set Range", "sceneControl.set_framerange(nuke.thisNode())")
+    k = nuke.PyScript_Knob('set_framerange', 'Set Range', 'sceneControl.set_framerange(nuke.thisNode())')
     n.addKnob(k)
-    n.knob("set_framerange").setTooltip("Set Framerange from SceneControl Interface")
-    n.knob("set_framerange").clearFlag(nuke.STARTLINE)
+    n.knob('set_framerange').setTooltip('Set Framerange from SceneControl Interface')
+    n.knob('set_framerange').clearFlag(nuke.STARTLINE)
 
     # Comment
-    k = nuke.Multiline_Eval_String_Knob("comment", "Comment")
+    k = nuke.Multiline_Eval_String_Knob('comment', 'Comment')
     n.addKnob(k)
-    n.knob("comment").setValue(comment)
+    n.knob('comment').setValue(comment)
 
     # Update / publish
     if UPDATE_TOOLSET:
         # Button update Layer
-        k = nuke.PyScript_Knob("update_layer", "Update Layer", "sceneControl.update_layer()")
+        k = nuke.PyScript_Knob('update_layer', 'Update Layer', 'sceneControl.update_layer()')
         n.addKnob(k)
-        n.knob("update_layer").setFlag(nuke.STARTLINE)
+        n.knob('update_layer').setFlag(nuke.STARTLINE)
     
         # Button connect all Layers
-        k = nuke.PyScript_Knob("autoconnect_layers", "Connect Layers", "sceneControl.autoconnect_layers(nuke.thisNode())")
+        k = nuke.PyScript_Knob('autoconnect_layers', 'Connect Layers', 'sceneControl.autoconnect_layers(nuke.thisNode())')
         n.addKnob(k)
-        n.knob("autoconnect_layers").setFlag(nuke.STARTLINE)
+        n.knob('autoconnect_layers').setFlag(nuke.STARTLINE)
     
         # Button remove all layers
-        k = nuke.PyScript_Knob("remove_all_layers", "Remove All Layers", "sceneControl.remove_all_layers(nuke.thisNode())")
+        k = nuke.PyScript_Knob('remove_all_layers', 'Remove All Layers', 'sceneControl.remove_all_layers(nuke.thisNode())')
         n.addKnob(k)
-        n.knob("remove_all_layers").setTooltip("Removes all Layers in SceneControl.")
+        n.knob('remove_all_layers').setTooltip('Removes all Layers in SceneControl.')
     
         # Button delete all reads
-        k = nuke.PyScript_Knob("remove_scnctrl_reads", "Delete All Reads",
-                               "sceneControl.remove_scnctrl_reads(nuke.thisNode())")
+        k = nuke.PyScript_Knob('remove_scnctrl_reads', 'Delete All Reads',
+                               'sceneControl.remove_scnctrl_reads(nuke.thisNode())')
         n.addKnob(k)
-        n.knob("remove_scnctrl_reads").setTooltip("Deletes all reads created by SceneControl.")
-        n.knob("remove_scnctrl_reads").setFlag(nuke.STARTLINE)
-        n.knob("remove_scnctrl_reads").setVisible(False)
+        n.knob('remove_scnctrl_reads').setTooltip('Deletes all reads created by SceneControl.')
+        n.knob('remove_scnctrl_reads').setFlag(nuke.STARTLINE)
+        n.knob('remove_scnctrl_reads').setVisible(False)
     
         # Button publish Shot
-        k = nuke.PyScript_Knob("publish_shot", "Publish", "sceneControl.publish_shot()")
+        k = nuke.PyScript_Knob('publish_shot', 'Publish', 'sceneControl.publish_shot()')
         n.addKnob(k)
-        n.knob("publish_shot").setFlag(nuke.STARTLINE)
+        n.knob('publish_shot').setFlag(nuke.STARTLINE)
 
     # invisible helpers
     # helper
-    k = nuke.Int_Knob("layerNum", "Slots")
+    k = nuke.Int_Knob('layerNum', 'Slots')
     n.addKnob(k)
-    n.knob("layerNum").setValue(0)
-    n.knob("layerNum").setVisible(False)
+    n.knob('layerNum').setValue(0)
+    n.knob('layerNum').setVisible(False)
 
     # used for resolving expressions
-    k = nuke.EvalString_Knob("eval_helper", "eval_helper")
+    k = nuke.EvalString_Knob('eval_helper', 'eval_helper')
     n.addKnob(k)
-    n.knob("eval_helper").setValue("")
-    n.knob("eval_helper").setVisible(False)
+    n.knob('eval_helper').setValue('')
+    n.knob('eval_helper').setVisible(False)
     # for debug disable above
 
-    k = nuke.Boolean_Knob("stereoToggle", "Stereo Toggle")
+    k = nuke.Boolean_Knob('stereoToggle', 'Stereo Toggle')
     n.addKnob(k)
-    n.knob("stereoToggle").setValue(0)
-    n.knob("stereoToggle").setVisible(False)
+    n.knob('stereoToggle').setValue(0)
+    n.knob('stereoToggle').setVisible(False)
 
     # used for resolving expressions
-    k = nuke.EvalString_Knob("precomp_output", "Precomp Output")
+    k = nuke.EvalString_Knob('precomp_output', 'Precomp Output')
     n.addKnob(k)
-    n.knob("precomp_output").setValue("")
-    n.knob("precomp_output").setVisible(False)
+    n.knob('precomp_output').setValue('')
+    n.knob('precomp_output').setVisible(False)
     # for debug disable above
 
-    k = nuke.Boolean_Knob("debug", "Debug")
+    k = nuke.Boolean_Knob('debug', 'Debug')
     n.addKnob(k)
-    n.knob("debug").setValue(1)
-    n.knob("debug").setVisible(False)
+    n.knob('debug').setValue(1)
+    n.knob('debug').setVisible(False)
 
     # end helper
 
     k = sceneControl.knobs()
     kenvCheck()
     # debug
-    if k["debug"].getValue():
+    if k['debug'].getValue():
         print(('SceneControl: Debug. WriteTask: %s, WriteElement: %s, WriteInfo: %s') % (writeTask, writeElement, writeInfo))
 
     # useEnvProjectRoot
     if writeTask.lower() == 'scripttask':
         # get task from sceneControl and not from the write node
         # sceneControl task
-        task = k["task"].value()
+        task = k['task'].value()
         writeTask = task
         # KEnv query
         renderpath = kenvQuery('RENDER_2D_FILE', element=writeElement, task=writeTask, minor_version=None, info=writeInfo, ext=ext, USE_ENV_PROJECTROOT=useEnv)
     k = sceneControl.knobs()
     kenvCheck()
     #debug
-    if k["debug"].getValue():
+    if k['debug'].getValue():
         print(('SceneControl: Debug. Setting path on: %s') % n.name())
     subtask = n.knob('subtask').value()
     if subtask is None:
 def setIgnoreFlag(state):
     s = nuke.selectedNodes()
     if len(s) == 0:
-        nuke.message("Please select some Nodes first.")
+        nuke.message('Please select some Nodes first.')
         return
 
     for n in s:
 def setErrorMode():
     n = nuke.selectedNodes()
     if len(n) == 0:
-        nuke.message("Please select some Nodes first.")
+        nuke.message('Please select some Nodes first.')
         return
     w = nuke.nodes.Read()
     onerror = w['on_error'].values()
 
     sel = nuke.selectedNodes()
     for s in sel:
-        if s.Class() == "Read" or s.Class() == "DeepRead":
+        if s.Class() == 'Read' or s.Class() == 'DeepRead':
             s.knobs()['on_error'].setValue(ret)
 
 
 def setOcioRole():
     n = nuke.selectedNodes()
     if len(n) == 0:
-        nuke.message("No node(s) selected.")
+        nuke.message('No node(s) selected.')
 
     # create temporary write node
     w = nuke.nodes.Write(file_type='exr')
     ret = p.value('Colorspace')
     if ret is not None:
         for s in n:
-            if s.Class() == "Read" or s.Class() == "Write":
+            if s.Class() == 'Read' or s.Class() == 'Write':
                 s.knobs()['colorspace'].setValue(ret)
 
 
 
 def deselectAll():
     for each in nuke.allNodes():
-        each.knob("selected").setValue(False)
+        each.knob('selected').setValue(False)
     return
 
 
 def getFilename():
-    return nuke.Root()["name"].getValue()
+    return nuke.Root()['name'].getValue()
 
 
 
     nodes = nuke.allNodes()
     for n in nodes:
         if n.Class() == nodeclass:
-            if n.knob("label").getValue().lower() == nodename.lower():
+            if n.knob('label').getValue().lower() == nodename.lower():
                 return n
     return None
 
     # TODO: more automated update function.
     import sceneControl
 
-    # 'task' for legacy write nodes
-    customKnobs = (
-    'extension', 'ext', 'ver', 'explore', 'divider3', 'playinrv', 'pushtorv', 'pushtorvappend', 'divider2', 'ignore', 'divider1', 'set',
-    'info', 'element', 'subtask', 'task', 'w_reload', 'keller')
-
     # save these for reapplying later on
     saveKnobs = ('info', 'element', 'subtask', 'ignore')
 
     # get all reads
     nodes = [node for node in nuke.selectedNodes() if node.Class() == 'Write']
 
-    for n in nodes:
-        knobs = n.knobs()
-        info = ''
-        element = ''
-        subtask = ''
-        for c in customKnobs:
+    if len(nodes) >0:
+        for n in nodes:
+            knobs = n.knobs()
             # save config
             try:
                 info = knobs['info'].value()
                 ignore = knobs['ignore'].value()
             except:
                 pass
+            for c in KELLERTABKNOBS:
+                try:
+                    n.removeKnob(knobs[c])
+                except:
+                    pass
+
+            # create new Keller tab
+            sceneControl.createCustomWrite(n)
+
+            # refresh subtasks
+            sceneControl.kenvWriteReloadSubTasks(n)
+
+            # restore old values
             try:
-                n.removeKnob(knobs[c])
+                n.knobs()['info'].setValue(info)
             except:
-                pass
-
-        # create new Keller tab
-        sceneControl.createCustomWrite(n)
+                print('SceneControl: Error. Node %s. Cannot set value on knob: %s') % (n.name(), info)
+            try:
+                n.knobs()['element'].setValue(element)
+            except:
+                print('SceneControl: Error. Node %s. Cannot set value on knob: %s') % (n.name(), element)
+            try:
+                n.knobs()['subtask'].setValue(subtask)
+            except:
+                print('SceneControl: Error. Node %s. Cannot set value on knob: %s') % (n.name(), subtask)
+    else:
+        pass
+    return
 
-        # refresh subtasks
-        sceneControl.kenvWriteReloadSubTasks(n)
 
-        # restore old values
-        try:
-            n.knobs()['info'].setValue(info)
-        except:
-            print('SceneControl: Error. Node %s. Cannot set value on knob: %s') % (n.name(), info)
-        try:
-            n.knobs()['element'].setValue(element)
-        except:
-            print('SceneControl: Error. Node %s. Cannot set value on knob: %s') % (n.name(), element)
-        try:
-            n.knobs()['subtask'].setValue(subtask)
-        except:
-            print('SceneControl: Error. Node %s. Cannot set value on knob: %s') % (n.name(), subtask)
+def removeKellerTab():
+    # get all reads
+    nodes = [node for node in nuke.selectedNodes() if node.Class() == 'Write']
 
+    if len(nodes) >0:
+        for n in nodes:
+            knobs = n.knobs()
+            # save config
+            try:
+                info = knobs['info'].value()
+                element = knobs['element'].value()
+                subtask = knobs['subtask'].value()
+                ignore = knobs['ignore'].value()
+            except:
+                pass
+            for c in KELLERTABKNOBS:
+                try:
+                    n.removeKnob(knobs[c])
+                except:
+                    pass
+    else:
+        pass
     return
 
 
 
 def autoconnect_layers(node):
     n = nuke.thisNode()
-    nu = n["layerNum"].getValue()
+    nu = n['layerNum'].getValue()
     num = int(nu)
 
     for i in range(1, num + 1):
 def createCustomWrite(n):
     # Task
     '''
-    taskList = ("comp",)
+    taskList = ('comp',)
     try:
         taskList = sceneControl.kenvReloadTasks()
     except:
-        print("Our_Write: Error. Cannot call sceneControl.kenvReloadTasks()")
+        print('Our_Write: Error. Cannot call sceneControl.kenvReloadTasks()')
         pass
     '''
     try:
         # already existing
         return
     # create tab
-    print("Our_Write: Info. Creating KellerTab")
+    print('Our_Write: Info. Creating KellerTab')
     k = nuke.Tab_Knob('keller', 'Keller')
     n.addKnob(k)
 
 
 
     # _______________________________________________________________
-    k = nuke.Text_Knob("divider1", "")
+    k = nuke.Text_Knob('divider1', '')
     n.addKnob(k)
     # _______________________________________________________________
 
     # ignore
-    k = nuke.Boolean_Knob("ignore", "ignore in sceneControl")
+    k = nuke.Boolean_Knob('ignore', 'ignore in sceneControl')
     n.addKnob(k)
-    n.knob("ignore").setValue(0)
+    n.knob('ignore').setValue(0)
 
     # _______________________________________________________________
-    k = nuke.Text_Knob("divider2", "")
+    k = nuke.Text_Knob('divider2', '')
     n.addKnob(k)
     # _______________________________________________________________
 
     # play in new rv
-    k = nuke.PyScript_Knob("playinrv", "Play in RV", "playInRV.playInRV(nuke.thisNode(),0)")
+    k = nuke.PyScript_Knob('playinrv', 'Play in RV', 'playInRV.playInRV(nuke.thisNode(),0)')
     n.addKnob(k)
-    n.knob("playinrv").setTooltip("Opens new instance of RV")
-    n.knob("playinrv").setFlag(nuke.STARTLINE)
+    n.knob('playinrv').setTooltip('Opens new instance of RV')
+    n.knob('playinrv').setFlag(nuke.STARTLINE)
 
     # push to rv
-    k = nuke.PyScript_Knob("pushtorv", "Push to RV", "playInRV.playInRV(nuke.thisNode(),1)")
+    k = nuke.PyScript_Knob('pushtorv', 'Push to RV', 'playInRV.playInRV(nuke.thisNode(),1)')
     n.addKnob(k)
-    n.knob("pushtorv").setTooltip("Push to tagged RV. Will open RV and uses this instance to directly push sequences.")
+    n.knob('pushtorv').setTooltip('Push to tagged RV. Will open RV and uses this instance to directly push sequences.')
 
     # push to rv / append
-    k = nuke.PyScript_Knob("pushtorvappend", "Push to RV (append)", "playInRV.playInRV(nuke.thisNode(),2)")
+    k = nuke.PyScript_Knob('pushtorvappend', 'Push to RV (append)', 'playInRV.playInRV(nuke.thisNode(),2)')
     n.addKnob(k)
-    n.knob("pushtorvappend").setTooltip("Push to RV and append to existing sources")
+    n.knob('pushtorvappend').setTooltip('Push to RV and append to existing sources')
     n.knob("pushtorvappend").clearFlag(nuke.STARTLINE)
 
     k = nuke.Boolean_Knob("useshotenv", "use shot env")