blob: 4019d0226847f15754eb11239802fc97b6127b42 [file] [log] [blame]
# Copyright 2021 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.
from recipe_engine import recipe_api
import calendar
DM_JSON = 'dm.json'
class GoldUploadApi(recipe_api.RecipeApi):
def upload(self):
"""Attempt to upload files to Gold.
This module assumes setup has occurred for the vars and flavor modules.
revision =['revision']
results_dir = self.m.flavor.host_dirs.dm_dir
# Upload the images. It is preferred that the images are uploaded first
# so they exist whenever the json is processed.
image_dest_path = 'gs://%s/dm-images-v1' %['gs_bucket']
for ext in ['.png']:
files_to_upload = self.m.file.glob_paths(
'find %s images' % ext,
'*%s' % ext,
# For some reason, glob returns results_dir when it should return nothing.
files_to_upload = [f for f in files_to_upload if str(f).endswith(ext)]
if len(files_to_upload) > 0:
self.m.gsutil.cp('%s images' % ext, results_dir.join('*%s' % ext),
image_dest_path, multithread=True)
summary_dest_path = 'gs://%s' %['gs_bucket']
ref = revision
# Trybot results are siloed by issue/patchset.
if self.m.vars.is_trybot:
summary_dest_path = '/'.join([summary_dest_path, 'trybot'])
ref = '%s_%s' % (str(self.m.vars.issue), str(self.m.vars.patchset))
# Compute the directory to upload results to
now = self.m.time.utcnow()
summary_dest_path = '/'.join([
str(now.year ).zfill(4),
str( ).zfill(2),
str(now.hour ).zfill(2),
# Directly upload dm.json if it exists.
json_file = results_dir.join(DM_JSON)
# -Z compresses the json file at rest with gzip.
self.m.gsutil.cp('dm.json', json_file,
summary_dest_path + '/' + DM_JSON, extra_args=['-Z'])