cdc7cff32754a13082daa205aafc98bdbf586e51
[blender-staging.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 branch = ''
37
38 if len(sys.argv) >= 3:
39         branch = sys.argv[2]
40
41 # scons does own packaging
42 if builder.find('scons') != -1:
43     os.chdir('../blender')
44     scons_options = ['BF_QUICK=slnt', 'BUILDBOT_BRANCH=' + branch, 'buildslave']
45
46     if builder.find('linux') != -1:
47         buildbot_dir = os.path.dirname(os.path.realpath(__file__))
48         config_dir = os.path.join(buildbot_dir, 'config')
49         build_dir = os.path.join('..', 'build', builder)
50         install_dir = os.path.join('..', 'install', builder)
51
52         scons_options += ['WITH_BF_NOBLENDER=True', 'WITH_BF_PLAYER=False',
53             'BF_BUILDDIR=' + build_dir,
54             'BF_INSTALLDIR=' + install_dir,
55             'WITHOUT_BF_INSTALL=True']
56
57         config = None
58
59         if builder.endswith('linux_x86_64_scons'):
60             config = 'user-config-x86_64.py'
61         elif builder.endswith('linux_i386_scons'):
62             config = 'user-config-x86_64.py'
63
64         if config is not None:
65             config_fpath = os.path.join(config_dir, config)
66             scons_options.append('BF_CONFIG=' + config_fpath)
67
68         blender = os.path.join(install_dir, 'blender')
69         blenderplayer = os.path.join(install_dir, 'blenderplayer')
70         subprocess.call(['strip', '--strip-all', blender, blenderplayer])
71
72         retcode = subprocess.call(['python', 'scons/scons.py'] + scons_options)
73         sys.exit(retcode)
74     else:
75         retcode = subprocess.call(['python', 'scons/scons.py'] + scons_options)
76         sys.exit(retcode)
77
78 # clean release directory if it already exists
79 dir = 'release'
80
81 if os.path.exists(dir):
82     for f in os.listdir(dir):
83         if os.path.isfile(os.path.join(dir, f)):
84             os.remove(os.path.join(dir, f))
85
86 # create release package
87 try:
88     subprocess.call(['make', 'package_archive'])
89 except Exception, ex:
90     sys.stderr.write('Make package release failed' + str(ex) + '\n')
91     sys.exit(1)
92
93 # find release directory, must exist this time
94 if not os.path.exists(dir):
95     sys.stderr.write("Failed to find release directory.\n")
96     sys.exit(1)
97
98 # find release package
99 file = None
100 filepath = None
101
102 for f in os.listdir(dir):
103     rf = os.path.join(dir, f)
104     if os.path.isfile(rf) and f.startswith('blender'):
105         file = f
106         filepath = rf
107
108 if not file:
109     sys.stderr.write("Failed to find release package.\n")
110     sys.exit(1)
111
112 # create zip file
113 try:
114     upload_zip = "buildbot_upload.zip"
115     if os.path.exists(upload_zip):
116         os.remove(upload_zip)
117     z = zipfile.ZipFile(upload_zip, "w", compression=zipfile.ZIP_STORED)
118     z.write(filepath, arcname=file)
119     z.close()
120 except Exception, ex:
121     sys.stderr.write('Create buildbot_upload.zip failed' + str(ex) + '\n')
122     sys.exit(1)