file_getter.py

# Copyright 2017 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Module for fetching files used by suite scheduler."""

import base64
import logging
import os
import subprocess
import urllib2

# The path to save credentials.
CREDENTIALS_PATH = os.path.join(os.path.dirname(__file__), 'credentials')

# The path to save configs.
CONFIG_PATH = os.path.join(os.path.dirname(__file__), 'configs')

# The suite scheduler config file for testing.
TEST_SUITE_SCHEDULER_CONFIG_FILE = os.path.join(CONFIG_PATH,
                                                'fake_suite_scheduler.ini')

# Test lab config file
TEST_LAB_CONFIG_FILE = os.path.join(CONFIG_PATH, 'fake_lab_config.ini')

# Test rubik boards config file
TEST_RUBIK_CONFIG_FILE = os.path.join(CONFIG_PATH, 'fake_rubik_config.ini')

# Suite scheduler config repo
NEW_SUITE_SCHEDULER_CONFIG_FILE = None
NEW_LAB_CONFIG_FILE = None
NEW_RUBIK_CONFIG_FILE = None

def _write_config(config_name):
  tot_response = urllib2.urlopen(
      'https://chromium.googlesource.com/chromiumos/infra/suite_scheduler/+/refs/heads/main/generated_configs/{}?format=text'.format(config_name)
  ).read()
  config_text = base64.b64decode(tot_response)
  logging.info('Writing %s as: %s', config_name, config_text)
  filename = '/tmp/{}'.format(config_name)
  with open(filename, 'w') as f:
    f.write(config_text)
  return filename

def clone_config():
  """Function to clone the config for suite scheduler from the current head."""
  global NEW_SUITE_SCHEDULER_CONFIG_FILE
  global NEW_LAB_CONFIG_FILE
  global NEW_RUBIK_CONFIG_FILE
  logging.info('Fetching config from suite scheduler repo.')
  try:
    NEW_SUITE_SCHEDULER_CONFIG_FILE = _write_config('suite_scheduler.ini')
    NEW_LAB_CONFIG_FILE = _write_config('lab_config.ini')
    NEW_RUBIK_CONFIG_FILE = _write_config('rubik_config.ini')
  except Exception as e:
    logging.error(str(e))

# Service account secret json file used for staging project.
STAGING_CLIENT_SECRETS_FILE = os.path.join(
    CREDENTIALS_PATH,
    'suite-scheduler-staging_client_secret_service_account.json')

# Service account secret json file used for prod project.
PROD_CLIENT_SECRETS_FILE = os.path.join(
    CREDENTIALS_PATH, 'suite-scheduler_client_secret_service_account.json')