aaa

# -*- coding: utf-8 -*-
from PySide2 import QtCore
from PySide2 import QtGui
from shiboken2 import wrapInstance
import pyside2uic
import xml.etree.ElementTree as xml
from cStringIO import StringIO
import maya.cmds as cmds
import maya.OpenMayaUI as mui
import maya.mel as mel
import os, glob, math, re, sys, time



def setEnvPath():
    aboutValue = cmds.about(v=True)
    if '2014' in aboutValue:
        aboutValueName = '2014'
    elif '2015' in aboutValue:
        aboutValueName = '2015'
    elif '2016' in aboutValue:
        aboutValueName = '2016'
    elif '2013' in aboutValue:
        aboutValueName = '2013'
    elif '2012' in aboutValue:
        aboutValueName = '2012'
    else:
        aboutValueName = aboutValue
    ################################################################################
    pathNameList = {'MAYA_SCRIPT_PATH':'scriptPathLsit', 'MAYA_SHELF_PATH':'shelfPathList',
                    'XBMLANGPATH':'XBMLANGPathList', 'PYTHONPATH':'pythonPathList',
                    'MAYA_PLUG_IN_PATH':'pluginPathLsit'}
    scriptDir = cmds.internalVar(usd=True)
    aboutDir = os.path.abspath(os.path.join(os.path.dirname(scriptDir), os.path.pardir))
    scriptPathLsit = [scriptDir[:-1], r'F:/Reference/TD/maya2017/tool_scripts', r'F:/Reference/TD/maya2017/scripts']
    shelfPathList = [aboutDir+r'\prefs\shelves', r'F:/Reference/TD/maya2017/tool_scripts', r'F:/Reference/TD/maya2017/scripts']
    XBMLANGPathList = [aboutDir+r'\prefs\icons', r'F:/Reference/TD/maya2017/tool_scripts/icons', r'F:/Reference/TD/maya2017/icons']
    pythonPathList = [r'F:/Reference/TD/maya2017/tool_scripts', r'F:/Reference/TD/maya2017', r'F:/Reference/TD/maya2017/imagesQrc']
    pluginPathLsit = [r'F:/Reference/TD/maya2017/plugins/'+aboutValueName]
    ################################################################################
    if os.path.exists(aboutDir+'/Maya.env'):
        fileName = open(aboutDir+'/Maya.env', 'r')
        try:
            all_lines = fileName.readlines()
            fileName.close()
            if all_lines:
                ##################
                pathKeyList = pathNameList.keys()
                editPathList = []
                for index in range(len(all_lines)):
                    oldString = all_lines[index].strip()
                    linePath = oldString.split('=')[0].strip()
                    if linePath in pathKeyList:
                        addString = ''
                        for pathValue in eval(pathNameList[linePath]):
                            if not (pathValue in oldString):
                                addString = addString + ';' + pathValue
                        if addString:
                            if oldString.endswith(';'):
                                all_lines[index] = oldString[:-1] + addString + ';\r\n'
                            else:
                                all_lines[index] = oldString + addString + ';\r\n'
                        editPathList.append(linePath)
                ##################
                for pathKey in pathKeyList:
                    if not (pathKey in editPathList):
                        all_lines.append('\r\n' + pathKey + ' = ' + ';'.join(eval(pathNameList[pathKey])) + ';')
                ##################
                fileName = open(aboutDir+'/Maya.env', 'w')
                fileName.truncate()
                fileName.writelines(all_lines)
                fileName.close()
                ##################
            else:
                fileName = open(aboutDir+'/Maya.env', 'w')
                fileName.write('MAYA_SCRIPT_PATH = ')
                for a in range(len(scriptPathLsit)):
                    fileName.write(scriptPathLsit[a]+';')
                fileName.write('\r\n')
                #############################
                fileName.write('MAYA_SHELF_PATH = ')
                for a in range(len(shelfPathList)):
                    fileName.write(shelfPathList[a]+';')
                fileName.write('\r\n')
                #############################
                fileName.write('XBMLANGPATH = ')
                for a in range(len(XBMLANGPathList)):
                    fileName.write(XBMLANGPathList[a]+';')
                fileName.write('\r\n')
                #############################
                fileName.write('PYTHONPATH = ')
                for a in range(len(pythonPathList)):
                    fileName.write(pythonPathList[a]+';')
                fileName.write('\r\n')
                #############################
                fileName.write('MAYA_PLUG_IN_PATH = ')
                for a in range(len(pluginPathLsit)):
                    fileName.write(pluginPathLsit[a]+';')
                fileName.write('\r\n')
                #############################
                fileName.close()
                #############################
            cmds.confirmDialog(t=u'提示！', m=u'插件已经安装完成，重启maya后即可使用！', b='OK', db='OK', cb='OK', ds='OK')
        finally:
            fileName.close()
    else:
        fileName = open(aboutDir+'/Maya.env', 'w')
        #############################
        try:
            fileName.write('MAYA_SCRIPT_PATH = ')
            for a in range(len(scriptPathLsit)):
                fileName.write(scriptPathLsit[a]+';')
            fileName.write('\r\n')
            #############################
            fileName.write('MAYA_SHELF_PATH = ')
            for a in range(len(shelfPathList)):
                fileName.write(shelfPathList[a]+';')
            fileName.write('\r\n')
            #############################
            fileName.write('XBMLANGPATH = ')
            for a in range(len(XBMLANGPathList)):
                fileName.write(XBMLANGPathList[a]+';')
            fileName.write('\r\n')
            #############################
            fileName.write('PYTHONPATH = ')
            for a in range(len(pythonPathList)):
                fileName.write(pythonPathList[a]+';')
            fileName.write('\r\n')
            #############################
            fileName.write('MAYA_PLUG_IN_PATH = ')
            for a in range(len(pluginPathLsit)):
                fileName.write(pluginPathLsit[a]+';')
            fileName.write('\r\n')
            #############################
            fileName.close()
            #############################
            cmds.confirmDialog(t=u'提示！', m=u'插件已经安装完成，重启maya后即可使用！', b='OK', db='OK', cb='OK', ds='OK')
        finally:
            fileName.close()
############################################################
setEnvPath()