slave/skia_slave_scripts/upload_skimage_results.py

#!/usr/bin/env python
# Copyright (c) 2013 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.

"""Upload results from running skimage."""

import build_step
import os
import posixpath
# Must be imported after build_step, which adds site_config to the python path.
import skia_vars
import sys

from utils import gs_utils
from utils import sync_bucket_subdir
from build_step import PLAYBACK_CANNED_ACL
from build_step import BuildStep

SKP_TIMEOUT_MULTIPLIER = 8

class UploadSKImageResults(BuildStep):

  def __init__(
      self,
      timeout=build_step.DEFAULT_TIMEOUT * SKP_TIMEOUT_MULTIPLIER,
      no_output_timeout=(
          build_step.DEFAULT_NO_OUTPUT_TIMEOUT * SKP_TIMEOUT_MULTIPLIER),
      **kwargs):
    """Constructs an UploadSKImageResults BuildStep instance.

    timeout: maximum time allowed for this BuildStep. The default value here is
             increased because there could be a lot of images
             to be copied over to Google Storage.
    no_output_timeout: maximum time allowed for this BuildStep to run without
        any output.
    """
    build_step.BuildStep.__init__(self, timeout=timeout,
                                  no_output_timeout=no_output_timeout,
                                  **kwargs)

    self._dest_gsbase = (self._args.get('dest_gsbase') or
                         sync_bucket_subdir.DEFAULT_PERFDATA_GS_BASE)

  def _Run(self):
    # Copy actual-results.json to skimage/actuals
    print '\n\n====Uploading skimage actual-results to Google Storage====\n\n'
    src_dir = os.path.abspath(os.path.join(self._skimage_out_dir,
                                           self._builder_name))
    dest_dir = posixpath.join(
        skia_vars.GetGlobalVariable('googlestorage_bucket'),
        'skimage', 'actuals')
    http_header_lines = ['Cache-Control:public,max-age=3600']
    gs_utils.copy_storage_directory(src_dir=src_dir,
                                    dest_dir=dest_dir,
                                    gs_acl='public-read',
                                    http_header_lines=http_header_lines)

    # Copy actual images to Google Storage at skimage/output. This will merge
    # with the existing files.
    print '\n\n========Uploading skimage results to Google Storage=======\n\n'
    src_dir = os.path.abspath(os.path.join(self._skimage_out_dir, 'images'))
    dest_dir = posixpath.join(
        skia_vars.GetGlobalVariable('googlestorage_bucket'),
        'skimage', 'output')
    if os.path.isdir(src_dir) and os.listdir(src_dir):
      gs_utils.copy_storage_directory(src_dir=src_dir,
                                      dest_dir=dest_dir,
                                      gs_acl=PLAYBACK_CANNED_ACL)

if '__main__' == __name__:
  sys.exit(BuildStep.RunBuildStep(UploadSKImageResults))