37 lines
981 B
Python
37 lines
981 B
Python
|
#!/usr/bin/env python3
|
||
|
|
# Copyright 2023 The Chromium Authors
|
||
|
|
# Use of this source code is governed by a BSD-style license that can be
|
||
|
|
# found in the LICENSE file.
|
||
|
|
"""This script is used to configure siso."""
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
THIS_DIR = os.path.abspath(os.path.dirname(__file__))
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser(description='configure siso')
|
||
|
|
parser.add_argument('--rbe_instance', help='RBE instance to use for Siso')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
project = None
|
||
|
|
if not args.rbe_instance:
|
||
|
|
return 0
|
||
|
|
rbe_instance = args.rbe_instance
|
||
|
|
elems = rbe_instance.split('/')
|
||
|
|
if len(elems) == 4 and elems[0] == 'projects':
|
||
|
|
project = elems[1]
|
||
|
|
rbe_instance = elems[-1]
|
||
|
|
siso_env_path = os.path.join(THIS_DIR, '.sisoenv')
|
||
|
|
with open(siso_env_path, 'w') as f:
|
||
|
|
if project:
|
||
|
|
f.write('SISO_PROJECT=%s\n' % project)
|
||
|
|
f.write('SISO_REAPI_INSTANCE=%s\n' % rbe_instance)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
sys.exit(main())
|