Maya Python3 Support: modernize "safe" options

Author: andrewkaufman

python/IECoreMaya/BoxParameterUI.py

@@ -36,6 +36,7 @@
 
 import IECore
 import IECoreMaya
+from six.moves import range
 
 class BoxParameterUI( IECoreMaya.ParameterUI ) :
 

python/IECoreMaya/ClassVectorParameterUI.py

@@ -38,6 +38,7 @@
 
 import IECore
 import IECoreMaya
+from six.moves import range
 
 ## A ParameterUI for ClassVectorParameters. Supports the following Parameter userData entries :
 #

python/IECoreMaya/DAGPathParameter.py

@@ -164,7 +164,7 @@ def getDAGPathValue( self ) :
 			return dp
 		except:
 			if self.mustExist :
-				raise Exception, "Node '%s' does not exist!" % dagNodePath
+				raise Exception("Node '%s' does not exist!" % dagNodePath)
 			return None
 
 IECore.registerRunTimeTyped( DAGPathParameter )

python/IECoreMaya/DAGPathVectorParameter.py

@@ -36,7 +36,7 @@
 from maya.OpenMaya import *
 import re
 
-from DAGPathParameter import DAGPathParameter
+from .DAGPathParameter import DAGPathParameter
 
 """
 Parameter class for specifying a list of Maya DAG paths.
@@ -172,7 +172,7 @@ def getDAGPathVectorValue( self ) :
 				result.append( dp )
 			except:
 				if self.mustExist :
-					raise Exception, "Node '%s' does not exist!" % dagNodePath
+					raise Exception("Node '%s' does not exist!" % dagNodePath)
 
 		return result
 

python/IECoreMaya/FileBrowser.py

@@ -296,8 +296,8 @@ def __getItemsForPath( self, path, *args ) :
 
 		try:
 			fullDirContents = os.listdir( self.__path )
-		except Exception, e :
-			print e
+		except Exception as e :
+			print(e)
 			maya.cmds.evalDeferred( 'import maya.cmds; maya.cmds.confirmDialog( b="OK", title="Error retrieving file list...", message="%s" )' % e )
 			return
 

python/IECoreMaya/FnDagNode.py

@@ -38,15 +38,16 @@
 import maya.cmds
 
 import IECore
-import StringUtil
+from . import StringUtil
+import six
 
 ## This class extends Maya's MFnDagNode to add assorted helper functions.
 class FnDagNode( maya.OpenMaya.MFnDagNode ) :
 
 	## \param obj - MObject, This can also be a string or an MObjectHandle.
 	def __init__( self, obj ) :
 
-		if isinstance( obj, str ) or isinstance( obj, unicode ) :
+		if isinstance( obj, str ) or isinstance( obj, six.text_type ) :
 
 			obj = StringUtil.dependencyNodeFromString( obj )
 

python/IECoreMaya/FnOpHolder.py

@@ -32,7 +32,7 @@
 #
 ##########################################################################
 
-from FnParameterisedHolder import FnParameterisedHolder
+from .FnParameterisedHolder import FnParameterisedHolder
 
 import maya.cmds
 

python/IECoreMaya/FnParameterisedHolder.py

@@ -41,8 +41,10 @@
 
 import IECore
 
-import _IECoreMaya
-import StringUtil
+from . import _IECoreMaya
+from . import StringUtil
+import six
+from six.moves import zip
 
 ## A function set for operating on the various IECoreMaya::ParameterisedHolder
 # types. This allows setting and getting of plug and parameter values, and
@@ -53,7 +55,7 @@ class FnParameterisedHolder( maya.OpenMaya.MFnDependencyNode ) :
 	# either be an MObject or a node name in string or unicode form.
 	def __init__( self, object ) :
 
-		if isinstance( object, str ) or isinstance( object, unicode ) :
+		if isinstance( object, str ) or isinstance( object, six.text_type ) :
 			object = StringUtil.dependencyNodeFromString( object )
 
 		maya.OpenMaya.MFnDependencyNode.__init__( self, object )
@@ -173,7 +175,7 @@ def parameterPlugPath( self, parameter ) :
 	# of the maya plug or its OpenMaya.MPlug instance.
 	def plugParameter( self, plug ) :
 
-		if isinstance( plug, str ) or isinstance( plug, unicode ) :
+		if isinstance( plug, str ) or isinstance( plug, six.text_type ) :
 			plug = StringUtil.plugFromString( plug )
 
 		return _IECoreMaya._parameterisedHolderPlugParameter( self, plug )
@@ -298,7 +300,7 @@ def _classParameterStates( self, parameter=None, parentParameterPath="", result=
 
 			classInfo = parameter.getClasses( True )
 			if classInfo :
-				classInfo = zip( *classInfo )
+				classInfo = list(zip( *classInfo ))
 			else :
 				classInfo = [ [], [], [], [] ]
 

python/IECoreMaya/FnSceneShape.py

@@ -42,7 +42,9 @@
 import IECore
 import IECoreScene
 import IECoreMaya
-import StringUtil
+from . import StringUtil
+import six
+from six.moves import range
 
 
 ## A function set for operating on the IECoreMaya::SceneShape type.
@@ -73,7 +75,7 @@ class FnSceneShape( maya.OpenMaya.MFnDagNode ) :
 	# either be an MObject or a node name in string or unicode form.
 	# Note: Most of the member functions assume that this function set is initialized with the full dag path.
 	def __init__( self, object ) :
-		if isinstance( object, basestring ) :
+		if isinstance( object, six.string_types ) :
 			object = StringUtil.dagPathFromString( object )
 
 		maya.OpenMaya.MFnDagNode.__init__( self, object )
@@ -163,7 +165,7 @@ def selectedComponentNames( self ) :
 	## Selects the components specified by the passed names.
 	def selectComponentNames( self, componentNames ) :
 		if not isinstance( componentNames, set ) :
-			if isinstance( componentNames, basestring ):
+			if isinstance( componentNames, six.string_types ):
 				componentNames = set( (componentNames, ) )
 			else:
 				componentNames = set( componentNames )
@@ -284,7 +286,7 @@ def __createChild( self, childName, sceneFile, sceneRoot, drawGeo = False, drawC
 		# Set visible if I have any of the draw flags in my hierarchy, otherwise set hidden
 		if drawTagsFilter:
 			childTags = fnChild.sceneInterface().readTags( IECoreScene.SceneInterface.EveryTag )
-			commonTags = filter( lambda x: str(x) in childTags, drawTagsFilter.split() )
+			commonTags = [x for x in drawTagsFilter.split() if str(x) in childTags]
 			if not commonTags:
 				dgMod.newPlugValueBool( fnChildTransform.findPlug( "visibility" ), False )
 			else:

python/IECoreMaya/FnTransientParameterisedHolderNode.py

@@ -37,8 +37,8 @@
 import maya.OpenMaya
 import maya.cmds
 import IECoreMaya
-import _IECoreMaya
-import StringUtil
+from . import _IECoreMaya
+from . import StringUtil
 
 class FnTransientParameterisedHolderNode( IECoreMaya.FnParameterisedHolder ) :