annotate libcxx/utils/cat_files.py @ 167:e037d1dd7369

...
author anatofuz
date Tue, 24 Mar 2020 17:35:46 +0900
parents 1d019706d866
children 1f2b6ac9f198
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
150
anatofuz
parents:
diff changeset
1 #!/usr/bin/env python
anatofuz
parents:
diff changeset
2 #===----------------------------------------------------------------------===##
anatofuz
parents:
diff changeset
3 #
anatofuz
parents:
diff changeset
4 # Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
anatofuz
parents:
diff changeset
5 # See https://llvm.org/LICENSE.txt for license information.
anatofuz
parents:
diff changeset
6 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
anatofuz
parents:
diff changeset
7 #
anatofuz
parents:
diff changeset
8 #===----------------------------------------------------------------------===##
anatofuz
parents:
diff changeset
9
anatofuz
parents:
diff changeset
10 from argparse import ArgumentParser
anatofuz
parents:
diff changeset
11 import sys
anatofuz
parents:
diff changeset
12
anatofuz
parents:
diff changeset
13 def print_and_exit(msg):
anatofuz
parents:
diff changeset
14 sys.stderr.write(msg + '\n')
anatofuz
parents:
diff changeset
15 sys.exit(1)
anatofuz
parents:
diff changeset
16
anatofuz
parents:
diff changeset
17 def main():
anatofuz
parents:
diff changeset
18 parser = ArgumentParser(
anatofuz
parents:
diff changeset
19 description="Concatenate two files into a single file")
anatofuz
parents:
diff changeset
20 parser.add_argument(
anatofuz
parents:
diff changeset
21 '-o', '--output', dest='output', required=True,
anatofuz
parents:
diff changeset
22 help='The output file. stdout is used if not given',
anatofuz
parents:
diff changeset
23 type=str, action='store')
anatofuz
parents:
diff changeset
24 parser.add_argument(
anatofuz
parents:
diff changeset
25 'files', metavar='files', nargs='+',
anatofuz
parents:
diff changeset
26 help='The files to concatenate')
anatofuz
parents:
diff changeset
27
anatofuz
parents:
diff changeset
28 args = parser.parse_args()
anatofuz
parents:
diff changeset
29
anatofuz
parents:
diff changeset
30 if len(args.files) < 2:
anatofuz
parents:
diff changeset
31 print_and_exit('fewer than 2 inputs provided')
anatofuz
parents:
diff changeset
32 data = ''
anatofuz
parents:
diff changeset
33 for filename in args.files:
anatofuz
parents:
diff changeset
34 with open(filename, 'r') as f:
anatofuz
parents:
diff changeset
35 data += f.read()
anatofuz
parents:
diff changeset
36 if len(data) != 0 and data[-1] != '\n':
anatofuz
parents:
diff changeset
37 data += '\n'
anatofuz
parents:
diff changeset
38 assert len(data) > 0 and "cannot cat empty files"
anatofuz
parents:
diff changeset
39 with open(args.output, 'w') as f:
anatofuz
parents:
diff changeset
40 f.write(data)
anatofuz
parents:
diff changeset
41
anatofuz
parents:
diff changeset
42
anatofuz
parents:
diff changeset
43 if __name__ == '__main__':
anatofuz
parents:
diff changeset
44 main()
anatofuz
parents:
diff changeset
45 sys.exit(0)