1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# 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
import six
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.
# pylint: disable=super-with-arguments
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)
# pylint: enable=super-with-arguments
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).decode('utf-8')
return 0, output
except subprocess.CalledProcessError as e:
output = e.output
if not isinstance(output, six.string_types):
output = output.decode('utf-8')
return e.returncode, output
|
