maya-scripts-zyf/maya_python_scripts/maya_export_expressions.py at master · walkoncross/maya-scripts-zyf · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# coding=utf-8
# """
# Export all expression in a scene/project in Maya.

# Author: Zhao Yafei (zhaoyafei0210@gmail.com)
# """
import os
import os.path as osp
import maya.cmds as cmds
from pprint import pprint


def get_current_scene_name():
    """
    Get current scene name.

    Args:
        None.

    Returns:
        str
            Scene name.
    """

    scene_name = cmds.file(query=True, sceneName=True, shortName=True)
    scene_name = osp.splitext(scene_name)[0]

    return scene_name


def get_all_expression_nodes():
    """
    Get name list of expression nodes.

    Args:
        None.

    Returns:
        list of str
            Name list of all expression nodes.
    """

    expression_list = cmds.ls(type='expression')
    pprint(expression_list)

    return expression_list


def export_expression(expression_node_name, save_dir='./'):
    """
    Export the content (string) of a expression to save_dir.

    Args:
        expression_node_name: str
            Expression name;
        save_dir: str
            Save directory;

    Returns:
        None.
    """
    print("===> save_dir: {}".format(save_dir))

    if not osp.exists(save_dir):
        os.makedirs(save_dir)

    scene_name = get_current_scene_name()
    # save_dir = osp.join(save_dir, scene_name+'_expressions')
    # os.mkdir(save_dir)
    pprint('===> export expression: ' + expression_node_name)
    # save_filename = osp.join(save_dir, expression_node_name+'.txt')
    save_filename = osp.join(save_dir, '{}.expression.{}.txt'.format(
        scene_name, expression_node_name))

    expression = cmds.getAttr(expression_node_name+'.expression')
    # pprint(expression)

    pprint('---> save expression into file {}'.format(save_filename))
    fp = open(save_filename, 'w')
    fp.write(expression)
    fp.close()


def export_all_expressions(save_dir):
    """
    Export the content (string) of all expression to save_dir, one .txt file for each expression.

    Args:
        save_dir: str
            Directory to save expression content file.

    Returns:
        list of str
            Name list of all expressions.

    """
    print("===> save_dir: {}".format(save_dir))

    if not osp.exists(save_dir):
        os.makedirs(save_dir)

    # scene_name = get_current_scene_name()
    # save_dir = osp.join(save_dir, scene_name+'_expressions')
    # os.mkdir(save_dir)

    expression_list = get_all_expression_nodes()

    for exp_name in expression_list:
        export_expression(exp_name, save_dir)


if __name__ == '__main__':
    save_dir = '/Users/zhaoyafei/work/maya-scripts-zyf/maya_exports/'
    export_all_expressions(save_dir)