pep8 cleanup
[blender.git] / build_files / buildbot / slave_pack.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20
21 # Runs on buildbot slave, creating a release package using the build
22 # system and zipping it into buildbot_upload.zip. This is then uploaded
23 # to the master in the next buildbot step.
24
25 import os
26 import subprocess
27 import sys
28 import zipfile
29
30 # get builder name
31 if len(sys.argv) < 2:
32     sys.stderr.write("Not enough arguments, expecting builder name\n")
33     sys.exit(1)
34
35 builder = sys.argv[1]
36
37 # scons does own packaging
38 if builder.find('scons') != -1:
39     os.chdir('../blender')
40     retcode = subprocess.call(['python', 'scons/scons.py', 'BF_QUICK=slnt', 'buildslave'])
41     sys.exit(retcode)
42
43 # clean release directory if it already exists
44 dir = 'release'
45
46 if os.path.exists(dir):
47     for f in os.listdir(dir):
48         if os.path.isfile(os.path.join(dir, f)):
49             os.remove(os.path.join(dir, f))
50
51 # create release package
52 try:
53     subprocess.call(['make', 'package_archive'])
54 except Exception, ex:
55     sys.stderr.write('Make package release failed' + str(ex) + '\n')
56     sys.exit(1)
57
58 # find release directory, must exist this time
59 if not os.path.exists(dir):
60     sys.stderr.write("Failed to find release directory.\n")
61     sys.exit(1)
62
63 # find release package
64 file = None
65 filepath = None
66
67 for f in os.listdir(dir):
68     rf = os.path.join(dir, f)
69     if os.path.isfile(rf) and f.startswith('blender'):
70         file = f
71         filepath = rf
72
73 if not file:
74     sys.stderr.write("Failed to find release package.\n")
75     sys.exit(1)
76
77 # create zip file
78 try:
79     upload_zip = "buildbot_upload.zip"
80     if os.path.exists(upload_zip):
81         os.remove(upload_zip)
82     z = zipfile.ZipFile(upload_zip, "w", compression=zipfile.ZIP_STORED)
83     z.write(filepath, arcname=file)
84     z.close()
85 except Exception, ex:
86     sys.stderr.write('Create buildbot_upload.zip failed' + str(ex) + '\n')
87     sys.exit(1)