| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2011 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """Litify a .proto file. | 6 """Litify a .proto file. |
| 7 | 7 |
| 8 This program add a line | 8 This program add a line |
| 9 "option optimize_for = LITE_RUNTIME;" | 9 "option optimize_for = LITE_RUNTIME;" |
| 10 to the input .proto file. | 10 to the input .proto file. |
| 11 | 11 |
| 12 Run it like: | 12 Run it like: |
| 13 litify_proto_file.py input.proto output.proto | 13 litify_proto_file.py input.proto output.proto |
| 14 | |
| 15 """ | 14 """ |
| 16 | 15 |
| 17 import fileinput | 16 import fileinput |
| 18 import sys | 17 import sys |
| 19 | 18 |
| 19 |
| 20 def main(argv): | 20 def main(argv): |
| 21 if len(argv) != 3: | 21 if len(argv) != 3: |
| 22 print 'Usage: litify_proto_file.py [input] [output]' | 22 print 'Usage: litify_proto_file.py [input] [output]' |
| 23 sys.exit(1) | 23 return 1 |
| 24 output_file = open(sys.argv[2], 'w') | 24 output_file = open(sys.argv[2], 'w') |
| 25 for line in fileinput.input(sys.argv[1]): | 25 for line in fileinput.input(sys.argv[1]): |
| 26 output_file.write(line) | 26 output_file.write(line) |
| 27 | 27 |
| 28 output_file.write("\noption optimize_for = LITE_RUNTIME;\n") | 28 output_file.write("\noption optimize_for = LITE_RUNTIME;\n") |
| 29 return 0 |
| 29 | 30 |
| 30 | 31 |
| 31 if __name__ == '__main__': | 32 if __name__ == '__main__': |
| 32 main(sys.argv) | 33 sys.exit(main(sys.argv)) |
| OLD | NEW |