Introduction To Python PDF
Introduction To Python PDF
com
INTRODUCTION TO PYTHON
SCRIPTING
FOR MAYA ARTISTS
By Chad Vernon
Table of Contents
Introduction .................................................................................................................................................. 4
Additional Resources .................................................................................................................................... 4
Python in the Computer Graphics Industry .................................................................................................. 4
Some Programs that support Python: ...................................................................................................... 4
What is Python used for? .......................................................................................................................... 4
Introduction to Python ................................................................................................................................. 5
What is Python? ........................................................................................................................................ 5
The Python Interpreter ............................................................................................................................. 5
What is a Python Script? ........................................................................................................................... 5
The Interactive Prompt ............................................................................................................................. 5
Running Python Scripts From a Command Prompt .................................................................................. 7
Shebang Lines ....................................................................................................................................... 8
Running Scripts in Maya and within a Python Session ............................................................................. 9
Python Modules ...................................................................................................................................... 11
Data Types and Variables ............................................................................................................................ 11
Variables ................................................................................................................................................. 11
Numbers and Math ................................................................................................................................. 13
Strings ..................................................................................................................................................... 14
String Formatting ................................................................................................................................ 15
String Methods.................................................................................................................................... 17
Lists ......................................................................................................................................................... 18
Tuples ...................................................................................................................................................... 19
Dictionaries ............................................................................................................................................. 20
Booleans and Comparing Values ............................................................................................................ 20
www.chadvernon.com
Controlling the Flow of Execution............................................................................................................... 21
Conditionals ............................................................................................................................................ 21
Code Blocks ......................................................................................................................................... 22
While Loops............................................................................................................................................. 23
For Loops ................................................................................................................................................. 24
Functions ..................................................................................................................................................... 25
Function Arguments................................................................................................................................ 27
Scope ....................................................................................................................................................... 28
Lambda Expressions ................................................................................................................................ 28
Exceptions and Error Handling.................................................................................................................... 29
Modules ...................................................................................................................................................... 30
Module Packages .................................................................................................................................... 31
Built-In and Third Party Modules ............................................................................................................ 32
Files and Paths ............................................................................................................................................ 33
Classes and Object Oriented Programming ................................................................................................ 34
Quick Notes ............................................................................................................................................. 34
Classes ..................................................................................................................................................... 34
Inheritance and Polymorphism ............................................................................................................... 35
Documenting Your Code ............................................................................................................................. 37
Epytext ................................................................................................................................................ 39
reST ..................................................................................................................................................... 39
Google ................................................................................................................................................. 39
Numpydoc ........................................................................................................................................... 39
Coding Conventions .................................................................................................................................... 40
PEP8 Quick Notes .................................................................................................................................... 41
Writing Clean Code ..................................................................................................................................... 41
The DRY Principle .................................................................................................................................... 41
Use Clean Names .................................................................................................................................... 42
Naming Classes ................................................................................................................................... 43
Naming Methods ................................................................................................................................ 43
Avoid Abbreviations ............................................................................................................................ 44
Naming Booleans ................................................................................................................................ 44
Symmetrical Names ............................................................................................................................ 44
www.chadvernon.com
Working with Booleans ........................................................................................................................... 44
Use Ternaries .......................................................................................................................................... 45
Don't Use String as Types........................................................................................................................ 45
Don't Use Magic Numbers ...................................................................................................................... 46
Encapsulate Complex Conditionals ......................................................................................................... 46
Writing Clean Functions .......................................................................................................................... 47
Extracting a Method............................................................................................................................ 47
Return Early......................................................................................................................................... 47
Signs Your Function is Too Long.......................................................................................................... 48
Writing Clean Classes .............................................................................................................................. 48
High Cohesion ..................................................................................................................................... 48
Method Proximity ............................................................................................................................... 50
Comments ............................................................................................................................................... 51
Redundant Comments ........................................................................................................................ 51
Divider Comments .............................................................................................................................. 51
Zombie Comments .............................................................................................................................. 52
Clean Code Conclusion............................................................................................................................ 52
Python Outside of Maya Conclusion ........................................................................................................... 52
Python in Maya ........................................................................................................................................... 53
The Maya Python Command Documentation ........................................................................................ 54
Sample Scripts ............................................................................................................................................. 57
Calling MEL from Python............................................................................................................................. 61
Maya Python Standalone Applications ....................................................................................................... 61
Further Reading .......................................................................................................................................... 62
Conclusion ................................................................................................................................................... 63
www.chadvernon.com
Introduction
This workshop is geared towards students with little to no scripting/programming experience. By the
end of this workshop, you will have the knowledge to write and run Python scripts inside and outside of
Maya. You will not learn everything about Python from this workshop. This workshop includes all the
basic information you should know in order to be proficient in reading, writing, running, and modifying
Python scripts. The purpose of this workshop is not to make you expert Python scripters, but to give you
a solid foundation from which in further your Python studies.
Additional Resources
Maya
Modo
Nuke
Houdini
XSI
Massive
Blender
Photoshop
3ds max
Engineers can
www.chadvernon.com
Introduction to Python
What is Python?
Python is a general purpose scripting language used in many different industries. It is a relatively easy to
use and easy to learn language. Python is used in web services, hardware testing, game development,
animation production, interface development, database programming, and many other domains.
www.chadvernon.com
FIGURE 1 IDLE
www.chadvernon.com
www.chadvernon.com
To run this script from a command prompt or terminal window, you call the script with the python
command (which is on your system once you install Python).
D:\>C:\Python26\python myFirstPythonScript.py
You are running a python script.
8
You can add text together.
If you have the PATH environment variable (Google search path environment variable) set to include
Python (which should happen by default with the installer), you dont need to specify the path to the
Python executable.
D:\>python myFirstPythonScript.py
You are running a python script.
8
You can add text together.
You can also route the output of your script to a file to save it for later use:
D:\>python myFirstPythonScript.py > output.txt
Another way to run a Python script is to open it in IDLE and run it from within the editor as shown
below. This is a great way to quickly experiment with Python code as you learn and write new scripts
and applications.
Shebang Lines
On Linux and OSX, you can use something called a shebang line to allow you to run a Python script
without having to specify python. For example, you could just run:
$> myFirstPythonScript.py
To use a shebang line, just add the following to the first lines of your script:
www.chadvernon.com
#!/usr/bin/env/ python
www.chadvernon.com
Consult the Maya documentation for all the other variables you can set in the Maya.env file. If you have
multiple scripts with the same name in different directories, Python will use the first one it finds.
Notice when I import the module again, the script does not run:
10
www.chadvernon.com
Notice that when I reload the Python module, the result states it read the module from a .pyc file. A
.pyc file is a compiled Python file. When you import a Python module, Python compiles the code and
generates a .pyc file. You could distribute these .pyc files if you do not want people looking at your
code. Note however there are tools available on the internet that will easily decompile a pyc back into
source code. Import statements will work with .pyc files.
Python Modules
As you can see, Python modules are simply Python scripts that contain specific functionality. Since
Python is so widely used, you can find thousands of free Python modules on the internet that implement
various tasks such as networking, image manipulation, file handling, scientific computing, etc. To
interface with Maya, you import the maya.cmds module, which is the module that ships with Maya to
implement all the Maya commands.
Variables
The most common concept in almost all scripting and programming languages is the variable. A variable
is a storage container for some type of data:
11
www.chadvernon.com
FIGURE 9 VARIABLES
Variables allow us to store data in order to use it later. Variables, sometimes called identifiers, must
start with a non-numeric character or underscore (_) and may contain letters, numbers, and
underscores (_). Identifiers are case sensitive. It is always a good idea to name your variables with
descriptive names so your code is easy to read.
Legal variable names
joint_count
button1
teeth_geometry
_particle_effect
finger.nail
4vertexEdgeId
cluster-handle
Python is a dynamically typed language. This means that a variable can hold one type of value for a
while and then can hold another type later.
12
www.chadvernon.com
x = 5
x = 'Now x holds a string'
Not all languages allow you to do this. For example, in MEL, you have to declare a variable as an integer
and then that variable can only hold an integer. This is what is known as a statically typed language.
Floating-point
43
-9234
6
1.324
-23.5325
6.2
Python supports all the math operators that you would want to use on these numbers.
x
y
y
z
z
z
z
a
x
x
x
x
= 4 + 5
= x 8
= y * 2.2
= y / 1.2
= y // 3
= z ** 4
= -z
= 10 % 3
+= 2
-= 2
*= 2
/= 2
#
#
#
#
#
#
#
#
#
#
#
#
Addition
Subtraction
Multiplication
Division
Floor division
Power
Negation
Modulus (division remainder)
Addition and store the result back in x, same as x = x + 2
Subtraction and store the result back in x
Multiplication and store the result back in x
Division and store the result back in x
Notice in some of these statements, I use the same variable on both the right and left side of the
assignment operator (=). In most programming languages, the right side is evaluated first and then the
result is stored in the variable on the left. So in the statement y = y * 2.2, the expression
y * 2.2 is evaluated using the current value of y, in this case 1, and then the result (1 * 2.2) = 2.2 is
stored in the variable y.
Math operators have a precedence of operation. That is, some operators always execute before others
even when in the same expression. For example the following two lines give different results:
>>> print (5 + 3.2) * 2
16.4
>>> print 6 + 3.2 * 2
12.4
Multiplication and division always get evaluated before addition and subtraction. However, you can
control which expressions get evaluated first by using parentheses. Inner most parentheses always get
evaluated first.
13
www.chadvernon.com
3
3
3
3
6
2
*
*
*
*
(54
(54
(54
2
4
/ ((2 + 7) * 3)) 4
/ (9 * 3)) 4
/ 27) 4
4
When you mix integers and floating-point values, the result will always turn into a floating-point:
>>> 4 * 3.1
# Result: 12.4 #
Strings
Strings are text values. You can specify strings in single, double or triple quotes.
>>> 'This is a string'
>>> "This is also a string"
>>> """This string can span multiple lines"""
# Single quotes
# Double quotes
# Triple quotes
Single and double quoted strings are the same. The main thing to keep in mind when using strings are
escape sequences. Escape sequences are characters with special meaning. To specify an escape code,
you use the backslash character followed by a character code. For example, a tab is specified as \t, a
new line is specified as \n. You have to be mindful whenever escape sequences are involved because
they can lead to a lot of errors and frustration. For example, in Windows, paths are written using the
backslash: (e.x. C:\tools\new). If you save this path into a string, you get unexpected results:
>>> print "C:\tools\new"
C:
ools
Ew
Python reads the backslashes as an escape sequence. It thinks the \t is a tab and the \n is a new line.
To get the expected results, you either use escaped backslashes or a raw string:
>>> print "C:\\tools\\new"
C:\tools\new
>>> print r"C:\tools\new"
C:\tools\new
The easiest thing to do in this case is to always use forward slashes for paths because Python will find
the right file whether on Windows or not.
You can concatenate strings together with the + operator.
14
www.chadvernon.com
>>> x = 'I am '
>>> y = '25 years old.'
>>> print x + y
I am 25 years old.
Python will throw an error saying you cannot concatenate a string and an integer. You can fix this in a
couple different ways. One way is to convert the integer to a string.
>>> x = 'I am '
>>> y = 25
>>> print 'I am ' + str(y)
I am 25
String Formatting
String formatting allows you to code multiple string substitutions in a compact way. You use string
formatting with the format function available on all strings.
>>> x = 'I am'
>>> y = '32 years old.'
>>> print '{0} {1}'.format(x, y)
I am 32 years old.
>>> y = 32
>>> print 'I am {0} years old.'.format(y)
I am 32 years old.
The format function allows you do automatically convert multiple variables into a string. The numbers
within the curly braces represent the index of argument within the format function.
15
www.chadvernon.com
>>> '{0}, {1}, {2}'.format('a', 'b', 'c')
'a, b, c'
>>> '{}, {}, {}'.format('a', 'b', 'c') # Python 2.7+ only
'a, b, c'
>>> '{2}, {1}, {0}'.format('a', 'b', 'c')
'c, b, a'
>>> '{2}, {1}, {0}'.format(*'abc')
# unpacking argument sequence
'c, b, a'
>>> '{0}{1}{0}'.format('abra', 'cad')
# arguments' indices can be repeated
'abracadabra'
You can also use the format function to control decimal precision, string alignment, as well as adding
zero-padding to strings. To read more about string formatting, visit
https://docs.python.org/2/library/string.html#format-string-syntax
You can also format strings with the % operator.
>>> x = 'I am '
>>> y = '25 years old.'
>>> print %s%s % (x, y)
I am 25 years old.
>>> x = 'I am '
>>> y = 25
>>> print '%s%d % (x, y)
I am 25
To use the % operator, you provide a format string on the left of the % operator and the values you
want to print on the right of the % operator. Different types of values have different format codes.
Common codes include:
%s
%d, %i %f
-
string
integer
floating-point
In the second example above, '%s%d' % (x, y) contains two format codes, a string followed by an
integer. On the right side of the % operator, we need to specify a value for each of the format codes in
the order they appear in the format string. String formatting not only lets us code in a more compact
way, it also lets us format values for output:
>>> print
005
>>> print
+06
>>> print
-04
>>> print
4.42424
'%03d' % 5
'%+03d' % 6
'%+03d' % -4
'%.5f' % 4.4242353534
16
www.chadvernon.com
Like the format method, the % operator allows us to specify decimal precision, how many spaces a
number should be printed with, whether to include the sign, etc. This is especially useful when printing
out large tables of data. To read more about the % operator in string formatting, visit:
https://docs.python.org/2/library/stdtypes.html#string-formatting
I tend to prefer using the format method over the % operator because format is the more modern
method.
String Methods
Methods are chunks of code that perform some type of operation. We will learn more about methods,
also known as functions, in more detail later on, but now is a good time to introduce you to the syntax
of calling a method. We call a method using the dot operator (.):
object.method()
An object is an instance of a particular type. For example, we could have a string object.
>>> x = 'This is a string.'
>>> y = 'Another string.'
# x is a string object
# y is another string object
When we read the code, some_object.some_method(), we say we are calling the method
named some_method from the object called some_object. Most objects have many callable
methods. The format method described in the previous section is a string method. Below are some of
the methods found in string objects:
>>> x = 'This is my example string.'
>>> print x.lower()
this is my example string.
>>> print x.upper()
THIS IS MY EXAMPLE STRING.
>>> print x.replace('my', 'your')
This is your example string.
>>> print x.title()
This Is My Example String.
>>> print x.endswith('ng.')
True
>>> print x.endswith('exam')
False
Some of the methods have arguments in the parentheses. Many methods allow you to pass in values to
perform operations based on the arguments. For example x.replace('my', 'your') will return a
copy of string x with all of the 'my' instances replaced with 'your'. To view a list of the available
string methods, visit: https://docs.python.org/2/library/stdtypes.html#string-methods
There are many methods available for many different types of objects and there is no need to memorize
them. You will begin to memorize them after using them a lot. You can find help documentation for all
objects with the help command.
help(str)
17
www.chadvernon.com
The help command will print out the documentation associated with an object, class, or function.
Lists
Lists are sequences or arrays of data. Lists allow us to use organized groups of data in our scripts.
>>> list_of_lights = ['keyLight',
>>> print list_of_lights[0]
#
keyLight
>>> print list_of_lights[2]
#
backLight
>>> print list_of_lights[3]
#
fillLight
>>> print list_of_lights[-1]
#
fillLight
>>> print list_of_lights[0:2]
#
['keyLight', 'rimLight']
>>> print len(list_of_lights)
#
4
We can access elements in a list with a numeric index. The indices start at index 0 and increment with
each value in the list. We can also access subsets of lists with slicing
>>> print list_of_lights[0:2], list_of_lights[:3], list_of_lights[:-1]
['keyLight', 'rimLight'] ['keyLight', 'rimLight', 'backLight'] ['keyLight',
'rimLight', 'backLight']
>>> list_of_lights[0:2] = ['newLight', 'greenLight']
>>> print list_of_light
['newLight', 'greenLight', 'backLight', 'fillLight']
When an index is negative, it counts from the end of the list back. Lists can also contain mixed types of
data including other lists.
>>> my_list = [3, 'food', [another_list, 4 * 2, 'dog'], 80.3]
>>> print my_list[2][1]
8
When you try to access an element that does not exist, Python will throw an error.
>>> print my_list[34]
# Error: list index out of range
# Traceback (most recent call last):
#
File "<maya console>", line 1, in <module>
# IndexError: list index out of range #
18
www.chadvernon.com
>>> list_of_lights = ['keyLight', 'rimLight', 'backLight', 'fillLight']
>>> list_of_lights.sort()
>>> print listOfLights
['backLight', 'fillLight', 'keyLight', 'rimLight']
>>> list_of_lights.reverse()
>>> print listOfLights
['rimLight', 'keyLight', 'fillLight', 'backLight']
>>> list_of_lights.append('newLight')
>>> print list_of_lights
['rimLight', 'keyLight', 'fillLight', 'backLight', 'newLight']
The concept of indexing is not unique to lists. We can actually access strings the same way.
>>> x = 'Eat your vegetables'
>>> print x[0:6]
Eat yo
>>> print x[:-9]
Eat your v
>>> print x[5:]
our vegetables
>>> print x.find('veg')
9
>>> print x.split('e')
['Eat your v', 'g', 'tabl', 's']
You can think of strings as lists of characters. The main difference is strings cannot be edited in place
with indices. For example, the following is illegal.
>>> x[4] = 't'
# Error: 'str' object does not support item assignment
# Traceback (most recent call last):
#
File "<maya console>", line 1, in <module>
# TypeError: 'str' object does not support item assignment #
Strings are what are known as immutable objects. Once they are created, they cannot be changed. Lists
on the other hand are mutable objects, meaning they can change internally.
Tuples
Tuples are the same as lists except they are immutable. They cannot be changed once created.
>>> my_tuple = (5, 4.2, 'cat')
>>> my_tuple[1] = 3
# Error: 'tuple' object does not support item assignment
# Traceback (most recent call last):
#
File "<maya console>", line 1, in <module>
# TypeError: 'tuple' object does not support item assignment #
What is the purpose of tuples? There are aspects of Python that use or return tuples. Id say 99.9% of
the time, youll be using lists. Just be aware that you cannot change a tuple when one eventually pops
up.
19
www.chadvernon.com
Dictionaries
Dictionaries are like lists except their elements do not have to be accessed with numeric indices.
Dictionaries are a type of look-up table or hash map. They are useful in storing unordered types of data.
>>> characters = {'male': ['Derrick', 'Chad', 'Ryan'], 'female': ['Olivia', 'Sarah',
'Zoe']}
>>> print characters['male']
['Derrick', 'Chad', 'Ryan']
>>> print characters['female']
['Olivia', 'Sarah', 'Zoe']
>>> # Same functionality, different syntax:
>>> characters = dict(male=['Derrick', 'Chad', 'Ryan'], female=['Olivia', 'Sarah',
'Zoe'])
>>> print characters['male']
['Derrick', 'Chad', 'Ryan']
>>> print characters['female']
['Olivia', 'Sarah', 'Zoe']
# Some methods of dictionaries
>>> print characters.keys()
['male', 'female']
>>> print characters.values()
[['Derrick', 'Chad', 'Ryan'], ['Olivia', 'Sarah', 'Zoe']]
>>> print 'male' in characters.keys()
True
>>> print 'clown' in characters.keys()
False
# Elements can be added on the fly
>>> characters['alien'] = ['ET', 'Alf']
>>> characters[6] = 4.5
20
www.chadvernon.com
>>> x == y # tests if x is equal to y
>>> x != y # tests if x does not equal y
>>> x > y
# tests if x is greater than y
>>> x < y
# tests if x is less than y
>>> x >= y # tests if x is greater than or equal to y
>>> x <= y # tests if x is less than or equal to y
>>> x = 4.3
>>> y = 2
>>> 5 == 5, 5 == 8, x == y, x == 4.3
(True, False, False, True)
>>> x <= y, x > y, x == y + 2.3, x >= 93
(False, True, True, False)
>>> a = 'alpha'
>>> b = 'beta'
>>> a > b, a == 'alpha', a < b
(False, True, True)
''
[]
{}
0
None
#
#
#
#
#
Empty strings
Empty lists
Empty dictionaries
0 valued numbers
An empty value
'text'
[2, 3]
{3: 4.5}
2.3
#
#
#
#
Conditionals
To execute code only if a condition is True or False, you use the if, elif, and else statements.
The if statement runs a block of code (the indented portion) if the corresponding condition is True.
21
www.chadvernon.com
>>> x = 5
>>> if x == 5:
>>>
x += 3
>>> print x
8
An if statement can have a corresponding else statement which runs if the condition in the if statement
is False.
>>>
>>>
>>>
>>>
>>>
>>>
10
x = 5
if x < 5:
x += 3
else:
x *= 2
print x
# Optional else
x = 5
if x > 5 and not x == 2:
x += 2
elif x == 5:
# Optional elif
x += 4
elif x == 9:
# Optional elif
x -= 3
else:
# Optional else
x *= 2
print x
In a sequence of
If the condition is True,
the code in its corresponding code block is executed. If the condition is False, the code block is skipped
and execution continues to the next else or elif (else if) statement if one exists. else statements
must always be preceded by an if or elif statement but can be left out if not needed. elif
statements must always be preceded by an if statement but can left out if not needed.
if/elif/else statements, the first if statement is evaluated as True or False.
Code Blocks
Code blocks are chunks of code associated with another statement of code. Take the following code for
example.
22
www.chadvernon.com
>>> x = 'big boy'
>>> y = 7
>>> if x == big boy and y < 8:
>>>
print Big Boy!
>>>
y += 3
The last two lines are in the code block associated with the if statement. When the condition of the if
statement is evaluated as True, execution enters the indented portion of the script. Code blocks in
Python are specified by the use of whitespace and indentation. You can use any number of spaces or
even tabs, you just have to be consistent throughout your whole script. However, even though you can
use any amount of whitespace, the Python standard is 4 spaces. Code blocks can be nested in other
code blocks, you just need to make sure your indentation is correct.
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
16
>>>
9
x = 5
y = 9
if x == 5 or y > 3:
x += 3
if x == 8:
x *= 2
elif y == 7:
y -= 3
if x == 16 or y < 21:
x -= 10
else:
y *= 3
else:
x += 2
print x
print y
if/elif/else statements that are chained together need to be on the same indentation level.
Read
though the previous example and work out the flow of execution in your head or on paper.
While Loops
While loops allow you to run code while a condition is True.
>>> x = 5
>>> while x > 1:
>>>
x -= 1
>>>
print x
4
3
2
1
In the above example, the condition is tested as True, so execution enters the code block of the while
loop. When execution reaches the print statement, the value of x has been decremented and execution
returns to the while statement where the condition is tested again. This loop continues until the
23
www.chadvernon.com
condition is False. You have to take care that the condition eventually evaluates to False or else you will
get an infinite loop.
>>> x = 5
>>> while x > 1:
>>>
x += 1
In the above example, x will continue to increment and the condition will always be True. This is called
an infinite loop. If you create one of these, youll have to shut down your program, Ctrl-Alt-Delete, or
force quit out of Maya.
Sometimes you will want to exit out of a loop early or skip certain iterations in a loop. These can be
done with the break and continue commands.
>>> x = 0
>>> while x < 10:
>>>
x += 1
>>>
if x % 2:
>>>
continue
>>>
if x == 8:
>>>
break
>>>
print x
>>> else:
>>>
x = 2
2
4
6
>>> # This code causes an infinite loop, try to find out why.
>>> x = 0
>>> while x < 10:
>>>
if x % 2:
>>>
continue
>>>
if x == 8:
>>>
break
>>>
print x
>>>
x += 1
For Loops
For loops iterate over sequence objects such as lists, tuples, and strings. Sequence objects are data
types comprised of multiple elements. The elements are usually accessed by square brackets (e.g.
my_list[3]) as youve seen previously. However, it is often useful to be able to iterate through all of
the elements in a sequence.
24
www.chadvernon.com
>>> someItems = ['truck', 'car', 'semi']
>>> for x in someItems:
>>>
print x
truck
car
semi
You can iterate through two lists of the same length with the zip function.
>>>
>>>
>>>
>>>
>>>
The
The
The
Like the while loop, for loops support the continue, break, and else statements.
>>> for i in range(0, 10, 2):
# range(start, stop, step)
>>>
if i % 2 == 1:
>>>
continue
>>>
if i > 7:
>>>
break
>>>
print i,
>>> else:
# Optional else
>>>
print 'Exited loop without break'
0 2 4 6
Functions
Previously, weve seen functions and methods built in to Python (such as range and zip) and built in to
different data types (string, list, and dictionary methods). Functions allow us to create reusable chunks
of code that we can call throughout our scripts. Functions are written in the form
25
www.chadvernon.com
def function_name(optional, list, of, arguments):
# statements
>>> def my_print_function():
>>>
print 'woohoo!'
>>>
>>> my_print_function()
woohoo!
>>> for x in range(3):
>>>
my_print_function()
woohoo!
woohoo!
woohoo!
Functions also accept arguments that get passed into your function.
>>> def print_my_own_range(start, stop, step):
>>>
x = start
>>>
while x < stop:
>>>
print x
>>>
x += step
>>>
>>> print_my_own_range(0, 5, 2)
0
2
4
26
www.chadvernon.com
Function Arguments
Function parameters (arguments) can be passed to functions a few different ways. The first is positional
where the arguments are matched in order left to right:
>>> def func(x, y, z):
>>>
pass
>>>
>>> func(1, 2, 3)
# Uses x = 1, y = 2, z = 3
You can also specify the names of arguments you are passing if you only want to pass certain
arguments. These are called keyword arguments. This is the method used in the Maya commands.
>>> def func(x=1, y=3, z=10):
>>>
pass
>>>
>>> func(y=5)
# Uses x = 1, y = 5, z = 10
Often you will see code where the arguments are unknown and both *arg and **kwargs will be used:
27
www.chadvernon.com
>>> def func(*args, **kwargs):
>>>
print args
>>>
print kwargs
>>>
>>> func('a', 'b', joints=1, x=2, y=3, z=4)
('a', 'b')
{'y': 3, 'joints': 1, 'z': 4, 'x': 2}
Scope
Scope is the place where variables and functions are valid. Depending on what scope you create a
variable, it may or may not be valid in other areas of your code.
>>> def my_function():
>>>
x = 1
# x is in the local scope of my_function
The enclosing module (the .py file you create the variable in) is a global scope.
Global scope spans a single file only.
Each call to a function is a new local scope.
Assigned names are local, unless declared global.
Examples:
>>>
>>>
>>>
>>>
>>>
>>>
x = 10
def func():
x = 20
>>>
>>>
>>>
>>>
>>>
>>>
>>>
x = 10
def func():
global x
x = 20
func()
print x
func()
print x
>>> x = 10
>>> def func():
>>>
print x
>>> func()
Lambda Expressions
Lambda expressions are basically a way of writing short functions. Normal functions are usually of the
form:
28
www.chadvernon.com
def name(arg1, arg2):
statements...
This is useful because we can embed functions straight into the code that uses it. For example, say we
had the following code:
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
def increment(x):
return x + 1
def decrement(x):
return x 1
def crazy(x):
return x / 2.2 ** 3.0
D = {'f1': increment, 'f2': decrement, 'f3': crazy}
D['f1'](2)
A drawback of this is that the function definitions are declared elsewhere in the file. Lambda
expressions allow us to achieve the same effect as follows:
>>> D = {
'f1': (lambda x: x + 1),
'f2': (lambda x: x - 1),
'f3': (lambda x: x / 2.2 ** 3.0)
}
>>> D['f1'](2)
Note that lambda bodies are a single expression, not a block of statements. It is similar to what you put
in a def return statement. When you are just starting out using Python in Maya, you probably wont be
using many lambda expressions, but just be aware that they exist.
29
www.chadvernon.com
>>>
>>>
>>>
>>>
>>>
>>>
x = [1,2,3]
try:
x[10]
except IndexError:
print "What are you trying to pull?"
print "Continuing program..."
Another variation includes both the else and finally statements, both of which are optional.
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
x = [1,2,3]
try:
x[10]
except IndexError:
print 'What are you trying to pull?'
else:
# Will only run if no exception was raised
print 'No exception was encountered'
finally:
print 'This will run whether there is an exception or not'
print "Continuing program..."
You can raise your own exception if you want to prevent the script from running:
>>> if func() == 'bad return value':
>>>
raise RuntimeError('Something bad happened')
There are many different types of built-in exceptions that you can raise. You can even create your own
types of exceptions. To read more about the different types of built-in exceptions visit the
documentation here: https://docs.python.org/2/library/exceptions.html
Exceptions that I regularly use include:
RuntimeError
ValueError
TypeError
NotImplementedError
Modules
We learned in the beginning that Python modules are simply .py files full of Python code. These
modules can be full of functions, variables, classes (more on classes later), and other statements. We
also learned that to load a Python module from within Maya or another interactive prompt, we need to
import the module. When we import a module, we gain access to all the functionality of that module.
30
www.chadvernon.com
# mymathmodule.py
def add(x, y):
return x + y
def subtract(x, y):
return x y
Or
import mymathmodule as mm
mm.add(1, 2)
Or
from mymathmodule import add
add(1, 2)
Or
from mymathmodule import *
add(1, 2)
subtract(1, 2)
Remember, we can only import a module once per Python session. If we were to update the code in
mymathmodule.py, we wouldnt have access to the updates until we reload the module.
reload(mymathmodule)
We can also import modules into other modules. If we have one module full of some really useful
functions, we can import that module into other scripts we write (which are also modules) in order to
gain access to those functions in our current module.
Module Packages
Packages allow us to organize our Python modules into organized directory structures. Instead of
placing all of our modules into one flat directory, we can group our modules in subdirectories based on
functionality.
31
www.chadvernon.com
Any code you put in the __init__.py file of a package gets executed with the package. However, it is
usually good practice not to execute any complex code when the user imports your module or package.
32
www.chadvernon.com
import
import
import
import
import
import
import
import
import
import
import
sys
os
shutil
struct
json
xmllib
math
random
re
argparse
logging
#
#
#
#
#
#
#
#
#
#
#
System commands
Generic module to the operating system
Contains file copying functions
Deals with binary data
Tools to deal with JSON data
XML parser module
Contains many useful math operations
Random number module
Regular expressions module
Command-line option parser
Customizable logging functionality
There are also hundreds of third-party modules available online. For example, the PIL module contains
many functions that deal with image manipulation. To find out what functions are in a module, run the
help command or view online documentation.
#
#
#
#
#
#
#
#
#
Heres a more practical example of opening a maya ascii file and renaming myBadSphere to
myGoodSphere.
>>>
>>>
>>>
>>>
>>>
>>>
>>>
>>>
Python also makes it easy to find files and traverse directory hierarchies.
33
www.chadvernon.com
# Get all the Maya files in a directory
import os
maya_files = []
for content in os.listdir('C:/somedirectory'):
if content.endswith('.ma') or content.endswith('.mb'):
maya_files.append(content)
# Traverse a directory and all subdirectories for fbx files
fbx_files = []
for root, dirs, files in os.walk('/my/tools/directory'):
print 'Currently searching {0}'.format(root)
for file_name in files:
if file_name.endswith('.fbx'):
fbx_files.append(os.path.join(root, file_name))
#
#
#
#
returns
returns
returns
returns
Happy
True
"haphazard"
4
You are free to use Python without using any of its object oriented functionality by just sticking with
functions and groups of statements as we have been doing throughout these notes. However, if you
would like to create larger scale applications and systems, I recommend learning more about object
oriented programming. The Maya API and PyMEL, the popular Maya commands replacement module,
are built upon the notions of object oriented programming so if you want to use API functionality or
PyMEL in your scripts, you should understand the principles of OOP.
Note that while I give a brief introduction to object-oriented programming in this paper, a few pages
cannot do the topic any justice. I recommend reading more about object-oriented programming in the
endless resources found online and in books.
Classes
Classes are the basic building blocks of object oriented programming. With classes, we can create
independent instances of a common object. It is kind of like duplicating a cube a few times. They are all
cubes, but they have their own independent attributes.
34
www.chadvernon.com
class Shape(object):
def __init__(self, name):
self.name = name
def print_me(self):
print 'I am a shape named {0}.'.format(self.name)
>>> shape1 = shape(name='myFirstShape')
>>> shape2 = shape(name='mySecondShape')
>>> shape1.print_me()
I am a shape named myFirstshape.
>>>shape2.print_me()
I am a shape named mySecondShape.
The above example shows a simple class that contains one data member (name), and two
functions. Functions that begin with a double underscore usually have a special meaning in Python. The
__init__ function of a class is a special function called a constructor. It allows us to construct a new
instance of an object. In the example, we create two independent instances of a shape object: shape1
and shape2. Each of these instances contains its own copy of the name attribute defined in the class
definition. In the shape1 instance, the value of name is myFirstShape. In the shape2 instance, the
value of name is mySecondShape. Notice we dont pass in any value for the self argument. We dont
pass in any value for the self argument because the self argument refers to the particular instance of a
class.
The first argument in all class member methods (functions) should be the self argument. The self
argument is used to represent the current instance of that class. You can see in the above example
when we call the print_me method of each instance, it prints the name stored in each separate
instance. So objects are containers that hold their own copies of data defined in their class definition.
35
www.chadvernon.com
# Inherit from Shape
class PolyCube(Shape):
def __init__(self, name, length, width, height):
# Call the constructor of the inherited class
super(PolyCube, self).__init__(name)
# Store the data associated with this inherited class
self.length = length
self.width = width
self.height = height
def print_me(self):
super(PolyCube, self).print_me()
# The .2f in the string format means use 2 decimal places
print 'I am also a cube with dimensions {0:.2f}, {1:.2f},
{2:.2f}.'.format(self.length, self.width, self.height)
class PolySphere(Shape):
def __init__(self, name, radius):
# Call the constructor of the inherited class
super(PolySphere, self).__init__(name)
# Store the data associated with this inherited class
self.radius = radius
def print_me(self):
super(PolySphere, self).print_me()
print 'I am also a sphere with a radius of {0:.2f}.'.format(self.radius)
>>> cube1 = PolyCube('firstCube', 2.0, 1.0, 3.0)
>>> cube2 = PolyCube('secondCube', 3.0, 3.0, 3.0)
>>> sphere1 = PolySphere('firstSphere', 2.2)
>>> sphere2 = PolySphere('secondSphere', 2.5)
>>> shape1 = Shape('myShape')
>>> cube1.print_me()
I am a shape named firstCube.
I am also a cube with dimensions 2.00, 1.00, 2.00.
>>> cube2.print_me()
I am a shape named secondCube.
I am also a cube with dimensions 3.00, 3.00, 3.00.
>>> sphere1.print_me()
I am a shape named firstSphere.
I am also a sphere with a radius of 2.20.
>>> sphere2.print_me()
I am a shape named secondSphere.
I am also a sphere with a radius of 2.50.
In the above example, we create two new classes, PolyCube and PolySphere, that inherit from the base
class, Shape. We tell a class to inherit from another class by placing the class to inherit from in
parentheses when we declare the derived class. The two new classes will have all the data and methods
associated with the shape base class. When we call the constructor method, __init__, of PolyCube and
PolySphere, we still want to use the functionality of the constructor of its super class, shape. We can
36
www.chadvernon.com
do this by calling the super class constructor explicitly with super(PolyCube,
self).__init__(name). This will set the name variable since the Shape constructor initializes
the name member variable. The second method, print_me, contains the added functionality of our
new class. The method name is the same as the method name in the super class, Shape. However,
when we call the method with cube1.print_me(), Python knows to use the method in PolyCube and
not the method from Shape. This is called polymorphism. It allows us to replace, change, or add
functionality to existing classes. By creating these hierarchies of objects, you can create pretty complex
systems in neat, reusable classes that will keep your code clean and maintainable.
The previous example is an extremely simplified version of Mayas architecture. Nodes inherit off of
other nodes to build a complex hierarchy. Below is part of Mayas object oriented node hierarchy:
37
www.chadvernon.com
When you type help(modulename), Python will print some nice documentation using the docstrings that
you specified in your module.
There is an official specification on how you should format your docstrings, called the PEP 0257 which
you can read about here: https://www.python.org/dev/peps/pep-0257/. Many people don't strictly
follow this format and use a format that is supported by documentation generation tools like Doxygen,
Epydoc, and Sphix. Formats include (taken from http://stackoverflow.com/questions/3898572/what-isthe-standard-python-docstring-format):
38
www.chadvernon.com
Epytext
"""
This is a javadoc style.
@param param1: this is a first param
@param param2: this is a second param
@return: this is a description of what is returned
@raise keyError: raises an exception
"""
reST
"""
This is a reST style.
:param param1: this is a first param
:param param2: this is a second param
:returns: this is a description of what is returned
:raises keyError: raises an exception
"""
Google
"""
This is an example of Google style.
Args:
param1: This is the first param.
param2: This is a second param.
Returns:
This is a description of what is returned.
Raises:
KeyError: Raises an exception.
"""
Numpydoc
39
www.chadvernon.com
"""
My numpydoc description of a kind
of very exhautive numpydoc format docstring.
Parameters
---------first : array_like
the 1st param name `first`
second :
the 2nd param
third : {'value', 'other'}, optional
the 3rd param, by default 'value'
Returns
------string
a value in a string
Raises
-----KeyError
when a key error
OtherError
when an other error
"""
I recommend using one of these widely used formats instead of making one up on your own. You
should try to put docstrings in your code as much as possible. It will save you and your coworkers a lot
of time down the road.
Coding Conventions
Unlike many other languages, Python has an official coding convention called the PEP8 standard. A
coding convention describes the style rules and format of your code. It is by no coincidence that all the
code examples listed so far use underscore_separated variable and function names, four spaces for
indentations, and CaptilizedCamelCase for class names. These conventions are all part of the PEP8
standard. You'll notice however that Maya does not follow the PEP8 standard. The main reason for this
is that the Python API is generated procedurally from the C++ API. When writing Python code, I highly
encourage you to follow the PEP8 standard even if you prefer or have used a different convention in the
past.
Most of the Python world uses the PEP8 standard. Do not do what Maya does, instead use the official
and accepted standard. The only time not to follow PEP8 conventions (as noted in PEP8 itself) is to
follow the standard of the existing codebase you are working with. You do not need to follow every
single little rule, just be consistent.
You can read more about the PEP8 standard here: https://www.python.org/dev/peps/pep-0008/
40
www.chadvernon.com
# Yes
# No
# Variable names
skin_joints = ['joint1', 'joint2'] # Yes
skinJoints = ['joint1', 'joint2'] # No
# Function names
def calculate_bounds(): # Yes
def calculateBounds(): # No
# Spacing
[1, 2, 3, 4] # Yes
[1,2,3,4] # No
if joint_height > 10.0: # Yes
if joint_height>10.0: # No
get_export_nodes(root='skeleton_grp') # Yes
get_export_nodes( root = 'skeleton_grp' ) # No
41
www.chadvernon.com
sphere = create_poly_sphere(name='left_eye')
assign_shader(sphere, 'blinn1')
parent_constrain(head, sphere)
sphere = create_poly_sphere(name='right_eye')
assign_shader(sphere, 'blinn2')
parent_constrain(head, sphere)
The second code example is easier to maintain. If an update needs to be implemented, we only have to
update code in one place rather than multiple places.
Can you tell what this code is doing? If you're familiar with writing pipeline environment tools, you
might recognize what it is trying to do. What is self.__curr? Why is it a list of 6 values? What do the
elements of cl repreprent? Why does the length of cl being greater than 3 differentiate one block of the
if statement from the other?
A cleaner implementation would look something like this:
42
www.chadvernon.com
context_type, sequence, subsequence, shot = self.get_current_context()
self.set_type(context_type)
self.set_sequence(sequence)
if subsequence:
self.set_subsequence(subsequence)
if shot:
self.set_shot(shot)
self.__work_area = work_area if work_area else self.get_current_work_area()
The above code cleans up all the list indices and string manipulations to make the code easier to read
and understand. The individual list elements have been assigned meaningful names. Also the
environment variable accesses have been extracted away into new methods. This makes the code
easier to maintain. What happens if we want to change the name of the environment variables or
maybe we want to read the values from a configuration file on disk? Extracting those values to
functions would let us update the code in one place rather than multiple direct accesses to the
environment variable.
Naming Classes
Class names should be a noun because they represent objects. The name should be as specific as
possible. If the name cannot be specific, it may be a sign that the class needs to be split into smaller
classes. Classes should have a single responsibility.
Bad class names include:
ShapeIE
Utility
Common
MyFunctions
DansUtils
ShapeClass
ShapeExporter
RigPublisher
Project
User
Naming Methods
Method names should be verbs because they perform actions. There should be no need to read the
contents of a method if the name accurately describes what the method does. If the function is doing
one thing (as it should) it should be easy to name. If not, split the code into smaller functions.
Sometimes explaining the code out loud and help you name the function. If you say "And", "If", or "Or"
it is a warning sign that the method is doing too much.
Bad method names include:
proc_new
pending
process1
process2
create_process
is_pending
send_notification
import_mesh
calculate_rivet_matrix
Methods should only perform the actions described by the name. Any other actions are called side
effects and they can confuse people using your code. For example, a method called validate_form
43
www.chadvernon.com
should not save the form. A method called publish_model should not smooth the normals. A method
called prune_weights should not remove unused influences.
Avoid Abbreviations
Abbreviated text may be easier to type, but code is read more than it is written. When people talk
about code or read it silently, it is harder to say the abbreviations. There are also no standards when
referring to abbreviations.
Bad names:
sjData
jid
sjid
nm
sjState
Good names:
subjob_data
job_id
subjob_id
name
subjob_state
Naming Booleans
Boolean values should be able to fit in an actual sentence of saying something is True or False.
Bad boolean names:
open
status
login
is_open
logged_in
is_valid
enabled
done
Symmetrical Names
When names have a corresponding opposite, be consistent and always use the same opposite.
Bad naming
on/disabled
quick/slow
lock/open
low/max
Good naming
on/off
fast/slow
lock/unlock
min/max
44
www.chadvernon.com
# Don't do this
if len(items) == 0:
remove_entry = True
else:
remove_entry = False
# Instead do this
remove_entry = len(items) == 0
Avoid using booleans that represent negative values. This leads to double negatives, which end up
confusing people:
# Don't do this
if not not_valid:
pass
# Instead do this
if valid:
pass
Use Ternaries
Ternaries are ways of assigning a value to a variable depending on if some condition is True or False. For
example:
# Don't do this
if height > height_threshold:
category = 'giant'
else:
category = 'hobbit'
# Instead do this
category = 'giant' if height > height_threshold else 'hobbit'
This is considered bad form for various reasons. If we end up wanting to change the value of one of
these types, we have to change it in all the places that it is used. It can also lead to typos and
inconsistencies. A better approach would be:
45
www.chadvernon.com
class ComponentType(object):
arm = 'arm'
leg = 'leg'
if component_type == ComponentType.arm:
# do something
elif component_type == ComponentType.leg:
# do something else
The new code provides one place to change and update values (the DRY principle). It also provides
auto-completion support and is more searchable if you are using an IDE like PyCharm or Eclipse.
What do these numbers mean? You would need to search all over code that could span multiple files to
figure out what these numbers represent. A better approach would be:
class JobStatus(object):
waiting = 1
starting = 2
running = 3
aborting = 4
done = 5
def __init__(self, value=JobStatus.waiting):
self.status = value
def not_yet_running(self):
return self.status < JobStatus.running
def abort(self):
if self.not_yet_running():
self.status = JobStatus.done
elif self.status == JobStatus.running:
self.status = JobStatus.aborting
# job_status is the new run_mode
job_status.abort()
46
www.chadvernon.com
# Instead of this
if (obj.component.partial_path.startswith('model') and
namespace == GEOM and
has_rigging_publish(obj.child) and
edits_path):
# Encapsulate the complex conditional in a function or variable
def is_model_only_publish(obj):
return (obj.component.partial_path.startswith('model') and
namespace == 'GEOM' and
has_rigging_publish(obj.child) and
edits_path)
if is_model_only_publish(obj):
Extracting a Method
If you find your code 3 or 4 indentation levels deep, it may be time to extract some of that code into a
separate function. For example:
# Instead of this
if something:
if something_else:
while some_condition:
# do something complicated
# Do this instead
if something:
if something_else:
do_complicated_things()
def do_complicated_things():
while some_condition:
# do something complicated
Return Early
People can usually only keep track of a handful of trains of thought at a time. Therefore, we should try
to organize our code is as many discrete independent chunks as possible. For example:
47
www.chadvernon.com
# Instead of this
def validate_mesh(mesh):
result = False
if has_uniform_scale(mesh):
if has_soft_normal(mesh):
if name_is_alphanumeric(mesh):
result = name_is_unique(mesh)
return result
# Do this
def validate_mesh(mesh):
if not has_uniform_scale(mesh):
return False
if not has_normal(mesh):
return False
if not name_is_alphanumeric(mesh):
return False
return name_is_unique(mesh)
This is not a strict rule. Like everything listed so far, use it when it enhances readability.
Chapter
o Heading 1
Paragraph 1
Paragraph 2
o Heading 2
Paragraph 1
Module
o Class 1
Method 1
Method 2
o Class 2
Method 1
High Cohesion
Cohesion is the fact of forming a united whole. When a class is said to have high cohesion, all of its
functionality is closely related. We should strive to create classes with high cohesion. High cohesion not
only enhances readability; it also increases the likelihood of reusing the class. Signs that a class does not
have high cohesion are:
The class has methods that don't interact with the rest of the class.
48
www.chadvernon.com
For example:
# Low cohesion class
class Vehicle(object):
def edit_options():
pass
def update_pricing():
pass
def schedule_maintenance():
pass
def send_maintenance_reminder():
pass
def select_financing():
pass
def calculate_monthly_payment():
pass
The Vehicle class contains many unrelated methods. This makes it harder to use and maintain because
parts of unrelated code are intertwined together. A better approach would be to split this class up into
smaller classes:
49
www.chadvernon.com
# Low cohesion class
class Vehicle(object):
def __init__(self)
def edit_options():
pass
def update_pricing():
pass
class VehicleMaintainer(object):
def schedule_maintenance():
pass
def send_maintenance_reminder():
pass
class VehicleFinancer(object)
def select_financing():
pass
def calculate_monthly_payment():
pass
Method Proximity
Code should read top to bottom and related methods should be kept together:
def add_take():
if not validate_take(): # First method referenced should be directly below
raise ValueError('Take is not valid')
save_take() # Second method referenced should be below first
def validate_take():
return take.endswith(.mov)
def save_take():
# save in database
Collapsed code should read like an outline. Strive for multiple layers of abstraction:
Class
o
Method 1
Method 1a
Method 1ai
Method 1aii
Method 1b
Method 1c
Method 2
Method 2a
50
www.chadvernon.com
Comments
Comments should only be used to explain ideas and assumptions not already apparent by reading the
code.
Redundant Comments
The comments in this code do not add anything the user could not have figured out by reading the code.
# Clear the node combo box then add items
self.node_combobox.clear()
if nodes:
# Sort the nodes
nodes.sort()
# Check to see if there is a shape controller associated with the node
self.find_shape_controllers(nodes)
# Now add the list of nodes to the combo box
self.node_combobox.addItems(nodes)
# If a node is specified set the combo box
if node:
# Find the node's index
index = self.node_combobox.findText(
node,
QtCore.Qt.MatchExactly | QtCore.Qt.MatchCaseSensitive)
self.node_combobox.setCurrentIndex(index)
Divider Comments
If you see divider comments, it's a sign you need to extract the code into its own function:
51
www.chadvernon.com
# Now create the new group object and insert it into the table
# ---------------------------------------------------------------------------------# Create the group object
group = slidergroup.SliderGroup(name)
self._slider_groups[name] = group
# Tell the group what its start row is
group.setRow(row)
# Apply color
if color:
group.setColor(color)
Zombie Comments
Zombie comments are large sections of commented out code. People often do this because they think
they might need the code in the future. This is unnecessary because version control systems like git,
svn, and perforce perform this exact functionality. People looking at code with large commented out
portions will be confused. Why is the code commented out? Is it important?
52
www.chadvernon.com
Python in Maya
Mayas scripting commands come in the module package maya.cmds.
import maya.cmds
You will notice that if you type in help(cmds), you do not get anything useful besides a list of function
names. This is because Autodesk converted all of their MEL commands to Python procedurally. To get
help with Mayas Python commands, you will need to refer to the Maya documentation.
53
www.chadvernon.com
what you are trying to accomplish with Mayas interface and look at the script editor. Most the actions
you do with the Maya interface output what commands were called in the script editor. The only caveat
is that the output is in MEL so well have to do a little bit of translating. Since all of the commands and
arguments are the same between the MEL and Python commands, translating between the two is pretty
easy.
MEL commands usually consist of the command name followed by several flags. In the above code,
polySphere is the command, and each group of letters following a - is a flag. The numbers after each
flag are arguments of their corresponding flag. For example, -r is a flag with an argument of 1, -sx
is a flag with an argument of 20, -ax is a flag with 3 arguments: 0, 1, 0. Using this information, we can
look up the command in the Maya Python command documentation and write its Python equivalent.
Like I said earlier, commands in MEL and Python have the exact same name, look up polySphere in
the Python documentation. It looks like this:
54
www.chadvernon.com
The documentation page contains all the information you need to work with the command. The
Synopsis shows the function and all the possible arguments that can be passed in along with a
description of what the command does. In this case, it creates a new polygonal sphere. The return
value tells us what the function returns. The polySphere command returns a list containing two strings.
The first element of the list will be the name of the transform of the new polygon sphere. The second
string in the list will be the name of the polySphere node, which controls how the sphere is constructed.
>>> x = cmds.polySphere()
>>> print x
[u'pSphere1', u'polySphere1']
Notice that each string has a u before it. This stands for Unicode string which is a type of string that
you can ignore for now. Unicode strings help with international languages so just assume they are the
same as normal strings.
Following the Return value section is a list of related Maya Commands. Following these links is a good
way to learn about other commands. After the related commands is the Flags section. This section
should really be called Arguments or Parameters; Flags are more of a MEL construct. The list of Flags
(arguments) contains all the arguments that can be passed into the documented function. Each
argument description contains the argument name, an abbreviated argument name, what kind of data
you can pass into the argument, in what context the argument is valid, and a description of the
argument. Take the radius argument for example. By passing this argument to the polySphere function,
we can control the radius of the created sphere.
>>> x = cmds.polySphere(radius=2.5)
Personally, I tend to avoid the abbreviated form as I can never remember what all the abbreviations
mean when I read my code. Using the full name is more typing and makes your code longer, but I find it
easier to read.
The documentation is not always clear about what type of data is expected with an argument. For
example, the documentation says that the radius argument expects some data of type linear. Usually
by looking at the equivalent MEL command, you can figure out what to pass into the Python command.
However, there are some cases where the documented format of the expected data is just really vague
or cryptic. In these cases, if you cant figure out how to format the command, ask on a forum or mailing
list.
After the list of arguments, there is usually an examples section that gives various usage examples of the
given command.
Going back to our example MEL command:
55
www.chadvernon.com
polySphere -r 1 -sx 20 -sy 20 -ax 0 1 0 -cuv 2 -ch 1;
we can see that each of the MEL flags corresponds to an argument in the Python function. By looking up
the flags in the Python documentation, we can write the equivalent Python command:
cmds.polySphere(r=1, sx=20, sy=20, ax=(0, 1, 0), cuv=2, ch=1)
Its up to you on whether to use the abbreviated flags or not. Note that I also swapped the ch=1 for
ch=True. Refering to the documentation, the constructionHistory argument expects a Boolean value.
Remember from the Booleans and Comparing Values section that all non-zero numbers are evaluated as
True. I like to actually pass in the value True (or False if you want False) to these types of arguments
just for my own preference.
You will also notice in the documentation the letters in the colored squares. The C, Q, M, and E stand
for Create, Query, Multiple, and Edit. These letters tell you in what context an argument is valid. Many
commands have different functionality depending on what context you are running the command in. In
the previous example, we were creating a sphere, so all the arguments marked C were valid.
When you run a command in query mode, you can find information about an object created with the
same command. You run a command in query mode by passing query=True as an argument.
>>> sphere = cmds.polySphere(radius=2.5)[0]
>>> print cmds.polySphere(sphere, query=True, radius=True)
2.5
When you query a command, you pass in the object you want to query first, followed by your
arguments. When in query mode, you pass a True or False to the argument you want to query
regardless of what the expected type for that argument is documented as. You should only query one
argument at a time. Youll notice that when you query a value, the value returned from the function
may not be the same as what the documentation says is returned. In create mode, the polySphere
command returns a list of 2 strings. In query mode, the return type depends on the value you are
querying.
>>> sphere = cmds.polySphere()[0]
>>> print cmds.polySphere(sphere, query=True, radius=True)
1.0
>>> print cmds.polySphere(sphere, query=True, axis=True)
[0.0, 1.0, 0.0]
Besides create and query modes, you can also run a command in edit mode. Edit mode lets you edit
values of an existing node created with the command. You run a command in edit mode by passing in
edit=True as an argument to the function.
56
www.chadvernon.com
>>> sphere = cmds.polySphere()[1]
>>> cmds.polySphere(sphere, edit=True, radius=5)
And finally, sometimes a flag is marked with being available multiple times. For example the curve
command has point argument (p) that we use to specify all the cv's of a curve.
# MEL
curve -p 0 0 0 -p 3 5 6 -p 5 6 7 -p 9 9 9 -p 0 0 0 -p 3 5 6;
# Python
cmds.curve(p=[(0, 0, 0), (3, 5, 6), (5, 6, 7), (9, 9, 9), (0, 0, 0), (3, 5, 6)])
You now know how to look up command syntax and decipher the documentation. You have just about
all the knowledge you need now to write your own Maya scripts in Python. All you need now is to learn
the various commands. A really good way to do that is to look at other peoples scripts.
Sample Scripts
In most of the sample scripts, you will notice that I always put the code in functions or classes. When
you import a module, all of the code in the module gets executed. However, code inside functions does
not get run until the function is called. When writing scripts for Maya, it is good practice to structure
your code as functions or classes to be called by users. Otherwise, you may surprise your users by
executing unwanted code when they import your modules.
lightintensity.py
57
www.chadvernon.com
import maya.cmds as cmds
def scale_light_intensity(percentage=1.0):
"""Scales the intensity of each light in the scene by a percentange.
@param percentage: Percentage to change each light's intensity.
"""
# The ls command is the list command. It is used to list various nodes
# in the current scene. You can also use it to list selected nodes.
# Notice the strange syntax 'ls() or []'. If no lights are in the scene, the ls
# command returns None which would prevent us from running the for loop. The
# strange syntax is a convenient way to make sure all_lights is a list if ls
# returns None.
all_lights = cmds.ls(type='light') or []
# Loop through each light
for light in all_lights:
# The getAttr command is used to get attribute values of a node
current_intensity = cmds.getAttr('{0}.intensity'.format(light))
# Calculate a new intensity
new_intensity = current_intensity * percentage
# Change the lights intensity to the new intensity
cmds.setAttr('{0}.intensity'.format(light), new_intensity)
# Report to the user what we just did
print 'Changed the intensity of light {0} from {1:.3f} to
{2:.3f}'.format(light, current_intensity, new_intensity)
Concepts used: functions, lists, for loops, conditional statements, string formatting
In the function docstring, I list the argument with "@param". This is a format compatible with
documentation generation tools such as Doxygen which can generate documentation based off of this
format. There are a few other docstring conventions out there, choose one and be consistent.
To run this script, in the script editor type:
import lightintensity
lightintensity.scale_light_intensity(1.2)
renamer.py
58
www.chadvernon.com
import maya.cmds as cmds
def rename(name, nodes=None):
"""Renames a hierarchy of transforms using a naming string with '#' characters.
If you select the root joint of a 3 joint chain and pass in 'C_spine##_JNT',
the joints will be named C_spine01_JNT, C_spine02_JNT, and C_spine03_JNT.
@param name: A renaming format string. The string must contain a consecutive
sequence of '#' characters
@param nodes: List of root nodes you want to rename. If this argument is
omitted, the script will use the currently selected nodes.
"""
# The variable "nodes" has a default value of None. If we do not specify a value
# for nodes, it will be None. If this is the case, we will store a list of the
# currently selected nodes in the variable nodes.
if nodes is None:
# The ls command is the list command. Get all selected nodes
# of type transform.
nodes = cmds.ls(sl=True, type='transform')
# If nothing is selected, nodes will be None so we don't need
# to continue with the script
if not nodes:
raise RuntimeError('Select a root node to rename.')
# Find out how many '#' characters are in the passed in name.
digit_count = name.count('#')
if digit_count == 0:
raise RuntimeError('Name has no # sequence.')
# We need to verify that all the '#' characters are in one sequence.
substring = '#' * digit_count
# '#' * 3 is the same as '###'
newsubstring = '0' * digit_count # '0' * 3 is the same as '000'
# The replace command of a string will replace all occurances of the first
# argument with the second argument. If the first argument is not found in
# the string, the original string is returned.
newname = name.replace(substring, newsubstring)
# If the string returned after the replace command is the same as
# the original string, it means that the sequence of '#' was not found in
# our specified name. This would happen if the '#' characters were not all
# consecutive (e.g. 'my##New##Name').
if newname == name:
raise RuntimeError('Pound signs must be consecutive..')
# Here we are creating a format string to use in our naming.
# The number of digits is determined by the number of consecutive '#' characters.
# Example 'C_spine##_JNT' has 2 '#' characters.
# In a chain of 3 joints, we would want to name the joints
# C_spine01_JNT, C_spine02_JNT, C_spine03_JNT.
# In a format string we would want to say 'C_spine%02d_JNT' % number.
# We are creating the '%02d' part here.
name = name.replace(substring, '%0{0}d'.format(digit_count))
59
www.chadvernon.com
# Start at number 1
number = 1
for node in nodes:
# Loop through each selected node and rename its child hierarchy.
number = rename_chain(node, name, number)
def rename_chain(node, name, number):
"""Recursive function that renames the passed in node to name % number.
@param node: The node to rename.
@param name: A renaming format string. The string must contain a
consecutive sequence of '#' characters
@param number: The number to use in the renaming.
@return The next number to use in the renaming chain.
"""
# Create the new name. The variable name is a string like 'C_spine%02d_JNT'
# so when we say "name % number", it is the same as 'C_spine%02d_JNT' % number
new_name = (name % number)
# The rename command renames a node. Sometimes you have to be careful.
# If you try to rename a node and there's already a node with the same name,
# Maya will add a number to the end of your new name. The returned string of
# the rename command is the name that Maya actually assigns the node.
node = cmds.rename(node, new_name)
# The listRelatives command is used to get a list of nodes in a dag
# hierarchy. You can get child nodes, parent nodes, shape nodes, or all
# nodes in a hierarchy. Here we are getting the child nodes.
children = cmds.listRelatives(node, children=True,
type='transform', path=True) or []
# Since we renamed the current node, we increment the number for the next
# node to be renamed.
number += 1
for child in children:
# We will call the rename_chain function for each child of this node.
number = rename_chain(child, name, number)
return number
Concepts used: functions, recursive functions, lists, for loops, conditionals, string formatting, exceptions.
To run this script, select the root joint of a joint chain and in the script editor type:
import renamer
renamer.rename('C_tail##_JNT')
The renamer script uses a concept called recursive functions. A recursive function is a function that calls
itself. Recursive functions are useful when you are performing operations on data in a hierarchical
graph, like Mayas DAG. In recursive functions, you must specify an ending condition or else the
function will call itself in an infinite loop. In the above example, the function calls itself for each child
node in the hierarchy. Since there are always a limited number of children in a Maya hierarchy, the
recursive function is guaranteed to stop at some point.
60
www.chadvernon.com
61
www.chadvernon.com
mayapy X:\file.py
You can also write scripts to run operations on Maya files in batch mode.
Example: Open a Maya file, assign the default shader to all meshes, and save the scene.
import maya.standalone
import maya.cmds as cmds
def assign_default_shader(file_path):
# Start Maya in batch mode
maya.standalone.initialize(name='python')
# Open the file with the file command
cmds.file(file_path, force=True, open=True)
# Get all meshes in the scene
meshes = cmds.ls(type="mesh", long=True)
for mesh in meshes:
# Assign the default shader to the mesh by adding the mesh to the
# default shader set.
cmds.sets(mesh, edit=True, forceElement='initialShadingGroup')
# Save the file
cmds.file(save=True, force=True)
# Starting Maya 2016, we have to call uninitialize to properly shutdown
if float(cmds.about(v=True)) >= 2016.0:
maya.standalone.uninitialize()
In order to run this script, you need to use the mayapy Python interpreter and not the normal Python
interpreter. In addition to stand alone scripts, I recommend reading about the argparse module which
would allow you to start the mayapy interpreter and call the script all in one line from the command
line.
Further Reading
While I've covered most of the basics, there are still many more aspects of Python and programming in
general.
The Beginner's Guide to Python contains many resources on Python:
https://wiki.python.org/moin/BeginnersGuide
Pluralsight has a great course on Unit Testing with Python: https://www.pluralsight.com/courses/unittesting-python
Clean Code: A Handbook of Agile Software Craftsmanship is a great book on reading clean and easy to
read code. While not Python specific (it's written for Java) it's concepts still apply.
http://www.amazon.com/Clean-Code-Handbook-Software-Craftsmanship/dp/0132350882
62
www.chadvernon.com
Conclusion
After reading these notes, you should have a good understanding of how to write and run Python scripts
inside and outside of Maya. There is still plenty to learn though. There are hundreds of Maya
commands and plenty of useful modules out there to experiment with. I recommend continuing your
Python education by looking through the references included with these notes, reading other peoples
scripts, and experimenting with your own scripts. Good luck!
63