infra/bots/upload_skps.py

# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Create a CL to update the SKP version."""


import argparse
import os
import subprocess
import sys
import urllib2

import git_utils

SKIA_COMMITTER_EMAIL = 'update-skps@skia.org'
SKIA_COMMITTER_NAME = 'UpdateSKPs'
COMMIT_MSG = '''Update SKP version

Automatic commit by the RecreateSKPs bot.

TBR=%s
NO_MERGE_BUILDS
''' % SKIA_COMMITTER_EMAIL
SKIA_REPO = 'https://skia.googlesource.com/skia.git'


def main(target_dir, gitcookies):
  with git_utils.NewGitCheckout(repository=SKIA_REPO):
    # Download CIPD.
    cipd_sha1 = os.path.join(os.getcwd(), 'infra', 'bots', 'tools', 'luci-go',
                             'linux64', 'cipd.sha1')
    subprocess.check_call(['download_from_google_storage', '-s', cipd_sha1,
                           '--bucket', 'chromium-luci'])

    # First verify that there are no gen_tasks diffs.
    gen_tasks = os.path.join(os.getcwd(), 'infra', 'bots', 'gen_tasks.go')
    try:
      subprocess.check_call(['go', 'run', gen_tasks, '--test'])
    except subprocess.CalledProcessError as e:
      print >> sys.stderr, (
         'gen_tasks.go failed, not uploading SKP update:\n\n%s' % e.output)
      sys.exit(1)

    # Skip GCE Auth in depot_tools/gerrit_utils.py. Use gitcookies instead.
    os.environ['SKIP_GCE_AUTH_FOR_GIT'] = 'True'
    os.environ['GIT_COOKIES_PATH'] = gitcookies
    # Upload the new version, land the update CL as the update-skps user.
    config_dict = {
      'user.name': SKIA_COMMITTER_NAME,
      'user.email': SKIA_COMMITTER_EMAIL,
      'http.cookiefile': gitcookies,
    }
    with git_utils.GitLocalConfig(config_dict):
      with git_utils.GitBranch(branch_name='update_skp_version',
                               commit_msg=COMMIT_MSG,
                               commit_queue=True):
        upload_script = os.path.join(
            os.getcwd(), 'infra', 'bots', 'assets', 'skp', 'upload.py')
        subprocess.check_call(['python', upload_script, '-t', target_dir])
        subprocess.check_call(['go', 'run', gen_tasks])
        subprocess.check_call([
            'git', 'add', os.path.join('infra', 'bots', 'tasks.json')])


if '__main__' == __name__:
  parser = argparse.ArgumentParser()
  parser.add_argument("--target_dir")
  parser.add_argument("--gitcookies")
  args = parser.parse_args()
  main(args.target_dir, args.gitcookies)