output_managerless_skia_gold_session.py

Enable keyboard shortcuts

# Copyright 2020 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.

"""Implementation of skia_gold_session.py without output managers.

Diff output is instead stored in a directory and pointed to with file:// URLs.

"""

import os

import subprocess

import time

from skia_gold_common import skia_gold_session

class OutputManagerlessSkiaGoldSession(skia_gold_session.SkiaGoldSession):

  def RunComparison(  # pylint: disable=too-many-arguments

      self,

      name,

      png_file,

      output_manager=True,

      inexact_matching_args=None,

      use_luci=True,

      optional_keys=None,

      force_dryrun=False):

    # Passing True for the output manager is a bit of a hack, as we don't

    # actually need an output manager and just need to get past the truthy

    # check.

    return super(OutputManagerlessSkiaGoldSession, self).RunComparison(

        name=name,

        png_file=png_file,

        output_manager=output_manager,

        inexact_matching_args=inexact_matching_args,

        use_luci=use_luci,

        optional_keys=optional_keys,

        force_dryrun=force_dryrun)

  def _CreateDiffOutputDir(self, name):

    # Do this instead of just making a temporary directory so that it's easier

    # for users to look through multiple results. We intentionally do not clean

    # this directory up since the user might need to look at it later.

    timestamp = int(time.time())

    name = '%s_%d' % (name, timestamp)

    filepath = os.path.join(self._local_png_directory, name)

    os.makedirs(filepath)

    return filepath

  def _StoreDiffLinks(self, image_name, _, output_dir):

    results = self._comparison_results.setdefault(image_name,

                                                  self.ComparisonResults())

    # The directory should contain "input-<hash>.png", "closest-<hash>.png",

    # and "diff.png".

    for f in os.listdir(output_dir):

      file_url = 'file://%s' % os.path.join(output_dir, f)

      if f.startswith('input-'):

        results.local_diff_given_image = file_url

      elif f.startswith('closest-'):

        results.local_diff_closest_image = file_url

      elif f == 'diff.png':

        results.local_diff_diff_image = file_url

  @staticmethod

  def _RunCmdForRcAndOutput(cmd):

    try:

      output = subprocess.check_output(cmd, stderr=subprocess.STDOUT)

      return 0, output

    except subprocess.CalledProcessError as e:

      return e.returncode, e.output