src/tests/benchmarks/fidl/benchmark_suite/copy_replace_fidl_library.py

#!/usr/bin/env python3
# Copyright 2018 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# Generates a fidl file that is a copy of a fidl file but with the library
# name replaced with the name passed to this script.
#
# This exists because benchmark FIDL files are sometimes used in a single
# fidl() GN build rule and sometimes in multiple GN build rules and the
# library name can't be reused across build rules.

import sys

if len(sys.argv) != 4:
    print("expected 3 arguments: [input file] [output file] [new fidl library]")
    exit(1)

input_file = sys.argv[1]
output_file = sys.argv[2]
new_fidl_library = sys.argv[3]

with open(input_file) as fi:
    with open(output_file, "w") as fo:
        for line in fi:
            if 'library benchmarkfidl;\n' in line:
                fo.write('library %s;\n' % new_fidl_library)
            else:
                fo.write(line)