| #!/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()) |
