PYTHON

I started teaching myself Python a couple years ago on a friend's suggestion. I got a few pointers and lessons from some Python-veteran co-workers. It has been a little tough, but here a a few scripts I have come up with. They are intended to be used within Maya.

replace_paths_fileNode.py

I wrote this little snippet for a Texturing colleague. He asked for a way to replace all the urls from the file nodes for just a universal url. This way he could place all texture files into a single folder.

import maya.cmds as cmds class newPathFileNodeUI(): def __init__(self,name='newPathFileNode'): if cmds.window(name,query=True, exists=True): cmds.deleteUI(name) cmds.window(name, t=name,s=False) self.buidUI() cmds.showWindow() def buidUI(self): #Instructions column=cmds.columnLayout(w=400) cmds.text(al="left", l="\n Instructions: \n") cmds.text(al="left", l=" Type a new path to replace all current paths in all file nodes") cmds.separator(h=5) #TextField cmds.setParent(column) cmds.rowColumnLayout( numberOfColumns=4, columnWidth=[(1, 5), (2, 80), (3, 290), (4, 5)] ) cmds.separator(vis=False) cmds.text(al="right",l="new file path:") self.path_name = cmds.textField(w=265, h=20, ann="Name masking geo",tx="") cmds.separator(vis=False) #Buttons cmds.setParent(column) cmds.separator(h=20) cmds.rowColumnLayout( numberOfColumns=5, columnWidth=[(1, 40), (2, 140),(3,40), (4, 140), (5, 40)] ) cmds.separator(vis=False) cmds.button(l="REPLACE ALL", w=50, h=20, c=self.onReplaceClick) cmds.separator(vis=False) cmds.button(l="CANCEL", w=50, h=20, c=self.onCancelClick) cmds.separator(vis=False) # ----- MAIN FUNCTION ------ def onReplaceClick(self, *args): selection = cmds.ls(type='file') new_path = cmds.textField(self.path_name , query=True, tx=True) for x in selection: file_path = cmds.getAttr('{0}.fileTextureName'.format(x)) new = file_path.split('/') file_name = new[-1] cmds.setAttr('{0}.fileTextureName'.format(x),'{0}\{1}'.format(new_path,file_name),type='string') print(new_path+'\\' + file_name) cmds.deleteUI("newPathFileNode", window=True) # ---- EXIT ---- def onCancelClick(self, *args): cmds.deleteUI("newPathFileNode", window=True) newPathFileNodeUI()

paint_itBlack.py

This was partly a lesson/test from one of my Python veteran coworkers; he asked me to change the color input attribute on all shaders in the file into black. Tricky part is not all shaders shade the same input naming. It also had to respect other nodes like transparency.

import maya.cmds as cmds layer="_goBlack" # Lists to look up for materials and inputs shaderName = cmds.ls( materials=True ) global colorNode colorNode = ["color", "outColor"] global listFile listFile = [] listOutputs = [] layered_List = [] ramp_List = [] ### Find the input > output connection of each shader node def input_2_color_List( shader='shader', inputType='color'): source = cmds.listConnections( '{0}.{1}'.format(shader,inputType), d=False, s=True, p=True ) listOutputs.append(source) ### This section makes a list for the color node in the shader ## def shaderExlusion(cx="inputs",inList="layeredList"): for x in range(len(shaderName)): if cmds.objExists('{0}.{1}'.format(shaderName[x],cx)): inList.append(shaderName[x]) else: pass # Function appends listed named format to the colorNode list def color_listing(name="color[{0}].color_Color", top=10): counting = [1*x for x in range(top)] for x in range(len(counting)): x = name.format(counting[x]) yield x colorNode.append(x) #Hides masterLayer and creates new layer with variable naming def newRenderLayer(): cmds.setAttr ("defaultRenderLayer.renderable", 0) allObj = list(cmds.ls(g=True)) grab2Layer= cmds.select(allObj) if cmds.objExists(layer) is True: None else: cmds.createRenderLayer(n='{0}'.format(layer)) cmds.editRenderLayerGlobals(currentRenderLayer=layer) cmds.select(cl=True) # MAIN Function. It checks for the connection of each file to shader, disconnects and changes color & spec to black def blacked(fileTex = "file1", shader = "shader" , specular = "specularColor"): global colorNode if shader in ramp_List: limit = cmds.getAttr('{0}.color'.format(shader), s=True) color_Color = list(color_listing("color[{0}].color_Color",limit)) else: pass if shader in layered_List: limit = cmds.getAttr('{0}.inputs'.format(shader), s=True) inputsColor = list(color_listing("inputs[{0}].color",limit)) else: pass for x in range(len(colorNode)): try: connected = cmds.isConnected('{0}'.format(fileTex), '{0}.{1}'.format(shader,colorNode[x])) if connected == True: dx = cmds.disconnectAttr('{0}'.format(fileTex), '{0}.{1}'.format(shader,colorNode[x])) if shader != "lambert1": goBlack = cmds.setAttr ("{0}.{1}".format(shader,colorNode[x]), 0, 0, 0) noSpec = cmds.setAttr ("{0}.{1}".format(shader,specular), 0, 0, 0) if shader in ramp_List: del color_Color [:] colorNode = ["color", "outColor"] else: pass if shader in layered_List: del inputsColor[:] colorNode = ["color", "outColor"] else: pass except: pass #Function check for every file node in listFile def deFiler(mode="color"): for x in range(len(listFile)): blacked(listFile[x],mode) #Applies color node to de-File def rundown(): global listFile newRenderLayer() for x in range(len(shaderName)): try: input_2_color_List( shaderName[x], 'color') input_2_color_List( shaderName[x], 'inputs') input_2_color_List( shaderName[x], 'color_Color') except: None filter(None, listOutputs) listFile = [item for sublist in listOutputs for item in sublist] listFile = [item for sublist in listFile for item in sublist] listFile = [item for sublist in listFile for item in sublist] shaderExlusion("inputs", layered_List) shaderExlusion("color[1].color_Color",ramp_List) for x in range (len(shaderName)): deFiler(shaderName[x]) rundown()

prefix-2-renderLayer.py

Another lesson/test the same Python veteran buddy; he asked me to isolate an asset base on a prefix, move all the assets into a new layer, replace the shaders with a black hole matte while keeping the prefixed asset keep its shader.

import maya.cmds as cmds def prefix2RenderLayer(prefix = "ch_", layer="CHAR", shader="blackHole"): #define selection for prefix topnode = list(cmds.ls('{0}*'.format(prefix), tr=True, dag=True, ap=False, lf=False)) allObj = list(cmds.ls(tr=True, dag=True, ap=False, lf=False, cameras=False)) cameras = list(cmds.ls(cameras=True)) #function to select all objects BUT the prefix ones def blackHoleSelection(): grab2Layer= cmds.select(allObj) dropOrphans= cmds.select(topnode, d=True) #creates new layer with selection blackHoleSelection() if cmds.objExists(layer) is True: None else: cmds.createRenderLayer(n='{0}'.format(layer)) #creates black hole matte if cmds.objExists(shader) is True: None else: blackHoleMatte = cmds.shadingNode('surfaceShader', n='{0}'.format(shader), asShader=True) cmds.setAttr ("{0}.outMatteOpacity".format(shader), 0, 0, 0) #assigns black hole matte shader to the objects in new render layer cmds.editRenderLayerGlobals(currentRenderLayer=layer) blackHoleSelection() cmds.hyperShade(assign=blackHoleMatte) #this moves ch_ prefix objects to the assigned renderLayer for x in range(len(topnode)): dads = cmds.listRelatives(topnode[x],p=True) if dads == None: orphan = cmds.editRenderLayerMembers(layer,'{0}'.format(topnode[x])) else: None prefix2RenderLayer()