src/chrome/test/pyautolib/fetch_prebuilt_pyauto.py

   1 #!/usr/bin/env python
   2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Fetch prebuilt binaries to run PyAuto.
   7
   8 Sets up Chrome and PyAuto binaries using prebuilt binaries from the
   9 continuous build archives.  Works on mac, win, linux (32 & 64 bit).
  10
  11 Examples:
  12   On Mac:
  13   $ python fetch_prebuilt_pyauto.py -d xcodebuild/Release
  14       http://build.chromium.org/f/chromium/continuous/mac/LATEST
  15
  16   On Win:
  17   $ python fetch_prebuilt_pyauto.py -d chrome\Release
  18       http://build.chromium.org/f/chromium/continuous/win/LATEST
  19 """
  20
  21 import glob
  22 import httplib
  23 import optparse
  24 import os
  25 import platform
  26 import shutil
  27 import sys
  28 import urllib
  29 import urllib2
  30 import urlparse
  31
  32 import pyauto_utils
  33
  34
  35 class FetchPrebuilt(object):
  36   """Util class to fetch prebuilt binaries to run PyAuto."""
  37
  38   def _ParseArgs(self):
  39     parser = optparse.OptionParser()
  40     parser.add_option(
  41         '-d', '--outdir', type='string', default=None,
  42         help='Directory in which to setup. This is typically the directory '
  43         'where the binaries would go when compiled from source.')
  44     parser.add_option(
  45         '-p', '--platform', type='string',
  46         default=pyauto_utils.GetCurrentPlatform(),
  47         help='Platform. Valid options: win, mac, linux32, linux64. '
  48         'Default: current platform (%s)' % pyauto_utils.GetCurrentPlatform())
  49     parser.add_option(
  50         '-l', '--latest', action='store_true', default=False,
  51         help='Download the latest chromium build from commondatastorage. '
  52         '[default=False]')
  53     self._options, self._args = parser.parse_args()
  54     if self._options.latest:
  55       self._url = self._GetLastestDownloadURL(self._options.platform)
  56     elif not self._args:
  57       print >>sys.stderr, 'Need download url'
  58       sys.exit(2)
  59     else:
  60       self._url = self._args[0]
  61     if not self._options.outdir:
  62       print >>sys.stderr, 'Need output directory: -d/--outdir'
  63       sys.exit(1)
  64     self._outdir = self._options.outdir
  65     # Chromium continuous build archive has a non-standard format.
  66     if 'index.html?path=' in self._url:
  67       self._url = self._url.replace('index.html?path=', '')
  68     self._url = self._url.rstrip('/')
  69     # Determine name of zip.
  70     if not self._options.platform.startswith('linux'):
  71       self._chrome_zip_name = 'chrome-%s' % {'mac': 'mac',
  72                                              'win': 'win32'
  73                                             }[self._options.platform]
  74     else:
  75       linux_32_names = ['linux', 'lucid32bit']
  76       linux_64_names = ['linux64', 'lucid64bit']
  77       linux_names = {'linux': linux_32_names + linux_64_names,
  78                      'linux32': linux_32_names,
  79                      'linux64': linux_64_names
  80                     }[self._options.platform]
  81       for name in linux_names:
  82         zip_name = 'chrome-' + name
  83         if pyauto_utils.DoesUrlExist('%s/%s.zip' % (self._url, zip_name)):
  84           self._chrome_zip_name = zip_name
  85           break
  86       else:
  87         raise RuntimeError('Could not find chrome zip at ' + self._url)
  88
  89     # Setup urls to download.
  90     self._chrome_zip_url = '%s/%s.zip' % (self._url, self._chrome_zip_name)
  91     self._remoting_zip_url = self._url + '/' + 'remoting-webapp.zip'
  92     chrome_test_url = '%s/%s.test' % (self._url, self._chrome_zip_name)
  93     self._pyautolib_py_url = '%s/pyautolib.py' % chrome_test_url
  94     if self._options.platform == 'win':
  95       self._pyautolib_so_name = '_pyautolib.pyd'
  96       self._chromedriver_name = 'chromedriver.exe'
  97     else:
  98       self._pyautolib_so_name = '_pyautolib.so'
  99       self._chromedriver_name = 'chromedriver'
 100     if self._options.platform == 'mac':
 101       self._ffmpegsumo_so_name = 'ffmpegsumo.so'
 102       self._ffmpegsumo_so_url = chrome_test_url + '/' + self._ffmpegsumo_so_name
 103     self._pyautolib_so_url = chrome_test_url + '/' + self._pyautolib_so_name
 104     self._chromedriver_url = chrome_test_url + '/' + self._chromedriver_name
 105
 106   def _GetLastestDownloadURL(self, os_platform):
 107     os_type = {'win': 'Win',
 108                'mac': 'Mac',
 109                'linux': 'Linux',
 110                'linux32': 'Linux',
 111                'linux64': 'Linux_x64'}[os_platform]
 112     if os_type == 'Linux' and platform.architecture()[0] == '64bit':
 113       os_type = 'Linux_x64'
 114     last_change_url = ('http://commondatastorage.googleapis.com/'
 115                        'chromium-browser-continuous/%s/LAST_CHANGE' % os_type)
 116     response = urllib2.urlopen(last_change_url)
 117     last_change = response.read()
 118     if not last_change:
 119       print >>sys.stderr, ('Unable to get latest from %s' % last_change_url)
 120       sys.exit(2)
 121     last_change_url = ('http://commondatastorage.googleapis.com/'
 122                        'chromium-browser-continuous/%s/%s' % (os_type,
 123                                                               last_change))
 124     return last_change_url
 125
 126   def Cleanup(self):
 127     """Remove old binaries, if any."""
 128     pass
 129
 130   def Run(self):
 131     self._ParseArgs()
 132     if not os.path.isdir(self._outdir):
 133       os.makedirs(self._outdir)
 134     get_remoting = pyauto_utils.DoesUrlExist(self._remoting_zip_url)
 135
 136     # Fetch chrome & pyauto binaries
 137     print 'Fetching', self._chrome_zip_url
 138     chrome_zip = urllib.urlretrieve(self._chrome_zip_url)[0]
 139
 140     if get_remoting:
 141       print 'Fetching', self._remoting_zip_url
 142       remoting_zip = urllib.urlretrieve(self._remoting_zip_url)[0]
 143     else:
 144       print 'Warning: %s does not exist.' % self._remoting_zip_url
 145
 146     print 'Fetching', self._pyautolib_py_url
 147     pyautolib_py = urllib.urlretrieve(self._pyautolib_py_url)[0]
 148
 149     print 'Fetching', self._pyautolib_so_url
 150     pyautolib_so = urllib.urlretrieve(self._pyautolib_so_url)[0]
 151
 152     if self._options.platform == 'mac':
 153       print 'Fetching', self._ffmpegsumo_so_url
 154       ffmpegsumo_so = urllib.urlretrieve(self._ffmpegsumo_so_url)[0]
 155
 156     print 'Fetching', self._chromedriver_url
 157     chromedriver = urllib.urlretrieve(self._chromedriver_url)[0]
 158
 159     chrome_unzip_dir = os.path.join(self._outdir, self._chrome_zip_name)
 160     if os.path.exists(chrome_unzip_dir):
 161       print 'Cleaning', chrome_unzip_dir
 162       pyauto_utils.RemovePath(chrome_unzip_dir)
 163     print 'Unzipping'
 164     pyauto_utils.UnzipFilenameToDir(chrome_zip, self._outdir)
 165     if get_remoting:
 166       pyauto_utils.UnzipFilenameToDir(remoting_zip, self._outdir)
 167       shutil.move(self._outdir + '/remoting-webapp',
 168                   self._outdir + '/remoting/remoting.webapp')
 169
 170     # Copy over the binaries to outdir
 171     items_to_copy = {
 172       pyautolib_py: os.path.join(self._outdir, 'pyautolib.py'),
 173       pyautolib_so: os.path.join(self._outdir, self._pyautolib_so_name),
 174       chromedriver: os.path.join(self._outdir, self._chromedriver_name)
 175     }
 176     if self._options.platform == 'mac':
 177       items_to_copy[ffmpegsumo_so] = \
 178           os.path.join(self._outdir, self._ffmpegsumo_so_name)
 179
 180     unzip_dir_contents = glob.glob(os.path.join(chrome_unzip_dir, '*'))
 181     for item in unzip_dir_contents:
 182       name = os.path.basename(item)
 183       items_to_copy[item] = os.path.join(self._outdir, name)
 184
 185     for src, dest in items_to_copy.iteritems():
 186       pyauto_utils.RemovePath(dest)
 187       print '%s ==> %s' % (os.path.basename(src), dest)
 188       shutil.move(src, dest)
 189     pyauto_utils.RemovePath(chrome_unzip_dir)
 190
 191     # Final setup (if any)
 192     # Set executable bit on chromedriver binary.
 193     if not self._options.platform == 'win':
 194       os.chmod(items_to_copy[chromedriver], 0700)
 195
 196     # Create symlink to .framework on Mac
 197     if self._options.platform == 'mac':
 198       mac_app_name = os.path.basename([x for x in unzip_dir_contents
 199                                        if x.endswith('.app')][0])
 200       os.chdir(self._outdir)
 201       framework = glob.glob(os.path.join(
 202           mac_app_name, 'Contents', 'Versions', '*', '*.framework'))[0]
 203       print framework
 204       dest = os.path.basename(framework)
 205       os.path.lexists(dest) and os.remove(dest)
 206       print 'Creating symlink "%s"' % dest
 207       os.symlink(framework, dest)
 208
 209     print 'Prepared binaries in "%s"' % self._outdir
 210     return 0
 211
 212
 213 if __name__ == '__main__':
 214   sys.exit(FetchPrebuilt().Run())