aTools_python3/aToolsInstall.py at master · MKlimenko/aTools_python3 · 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
frommayaimportcmds, mel
importos, shutil, urllib.request, urllib.error, urllib.parse, shutil, zipfile, importlib

defformatPath(path):
path=path.replace('/', os.sep)
path=path.replace('\\', os.sep)
returnpath

defdownload(downloadUrl, saveFile):

try: response=urllib.request.urlopen(downloadUrl, timeout=60)
except: pass

ifresponseisNone:
cmds.warning('Error trying to install.')
return

fileSize=int(response.info().get('Content-Length')[0])
fileSizeDl=0
blockSize=128
output=open(saveFile,'wb')
progBar=mel.eval('$tmp = $gMainProgressBar')

cmds.progressBar( progBar,
edit=True,
beginProgress=True,
status='Downloading aTools...',
progress=0,
maxValue=100 )

whileTrue:
buffer=response.read(blockSize)
ifnotbuffer:
output.close()
cmds.progressBar(progBar, edit=True, progress=100)
cmds.progressBar(progBar, edit=True, endProgress=True)
break

fileSizeDl+=len(buffer)
output.write(buffer)
p=float(fileSizeDl) /fileSize*100

cmds.progressBar(progBar, edit=True, progress=p)

returnoutput


defaToolsInstall():

mayaAppDir=mel.eval('getenv MAYA_APP_DIR')
aToolsPath=mayaAppDir+os.sep+"scripts"
aToolsFolder=aToolsPath+os.sep+"aTools"+os.sep
tmpZipFile="%s%stmp.zip"%(aToolsPath, os.sep)
DOWNLOAD_URL="https://github.com/MKlimenko/aTools_python3/releases/download/v2.04/aTools.zip"

ifos.path.isfile(tmpZipFile): os.remove(tmpZipFile)
ifos.path.isdir(aToolsFolder): shutil.rmtree(aToolsFolder)

output=download(DOWNLOAD_URL, tmpZipFile)

zfobj=zipfile.ZipFile(tmpZipFile)
fornameinzfobj.namelist():
uncompressed=zfobj.read(name)

filename=formatPath('%s%s%s'%(aToolsPath, os.sep, name))
d=os.path.dirname(filename)

ifnotos.path.exists(d): os.makedirs(d)
iffilename.endswith(os.sep): continue

output=open(filename,'wb')
output.write(uncompressed)
output.close()

#delete temp
zfobj.close()
ifos.path.isfile(tmpZipFile): os.remove(tmpZipFile)

fromaToolsimportsetup; setup.install()

#refresh
cmds.evalDeferred("from aTools.animTools.animBar import animBarUI; importlib.reload(animBarUI); animBarUI.show('refresh')")

aToolsInstall()