Implement a script to download changed screenshots from CI
This should streamline updating the screenshots when they have to be
changed. Screenshot diff report generated on CI shows a command line
one has to enter, and provides a button to copy it.
Example: https://storage.googleapis.com/perfetto-ci-artifacts/20220711123401--cls-2149676-1--ui-clang-x86_64-release/ui-test-artifacts/index.html
Change-Id: I5147ffc9ab85a968ec893b45c82a737f1e2f8a12
diff --git a/tools/download_changed_screenshots.py b/tools/download_changed_screenshots.py
new file mode 100755
index 0000000..326b130
--- /dev/null
+++ b/tools/download_changed_screenshots.py
@@ -0,0 +1,54 @@
+#!/usr/bin/env python3
+# Copyright (C) 2022 The Android Open Source Project
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import argparse
+import sys
+import urllib.request
+from os import path
+
+
+def get_artifact_url(run, name):
+ return f'https://storage.googleapis.com/perfetto-ci-artifacts/{run}/ui-test-artifacts/{name}'
+
+
+def main():
+ parser = argparse.ArgumentParser()
+ parser.add_argument('run', metavar='RUN', help='CI run identifier')
+ args = parser.parse_args()
+
+ with urllib.request.urlopen(get_artifact_url(args.run, 'report.txt')) as resp:
+ handle_report(resp.read().decode('utf-8'), args.run)
+
+
+def handle_report(report: str, run: str):
+ for line in report.split('\n'):
+ if len(line) == 0:
+ continue
+
+ parts = line.split(';')
+ if len(parts) != 2:
+ print('Erroneous report line!')
+ sys.exit(1)
+
+ screenshot_name = parts[0]
+ url = get_artifact_url(run, screenshot_name)
+ output_path = path.join('test', 'data', 'ui-screenshots', screenshot_name)
+ print(f'Downloading {url}')
+ urllib.request.urlretrieve(url, output_path)
+ print('All done!')
+
+
+if __name__ == "__main__":
+ main()
