eson/googletest
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

pump.py: add support for Python 3

Browse Source
This commit is contained in:
Krystian Kuzniarek 2019-11-02 01:13:24 +01:00
parent e8a82dc7ed
commit 442f45b376
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
googlemock/scripts/pump.py
View File

@ -1,4 +1,4 @@
#!/usr/bin/env python2.7 #!/usr/bin/env python
# #
# Copyright 2008, Google Inc. # Copyright 2008, Google Inc.
# All rights reserved. # All rights reserved.
@ -64,6 +64,7 @@ GRAMMAR:
from __future__ import print_function from __future__ import print_function
import io
import os import os
import re import re
import sys import sys
@ -834,7 +835,7 @@ def main(argv):
sys.exit(1) sys.exit(1)
file_path = argv[-1] file_path = argv[-1]
output_str = ConvertFromPumpSource(file(file_path, 'r').read()) output_str = ConvertFromPumpSource(io.open(file_path, 'r').read())
if file_path.endswith('.pump'): if file_path.endswith('.pump'):
output_file_path = file_path[:-5] output_file_path = file_path[:-5]
else: else:
@ -842,11 +843,11 @@ def main(argv):
if output_file_path == '-': if output_file_path == '-':
print(output_str,) print(output_str,)
else: else:
output_file = file(output_file_path, 'w') output_file = io.open(output_file_path, 'w')
output_file.write('// This file was GENERATED by command:\n') output_file.write(u'// This file was GENERATED by command:\n')
output_file.write('// %s %s\n' % output_file.write(u'// %s %s\n' %
(os.path.basename(__file__), os.path.basename(file_path))) (os.path.basename(__file__), os.path.basename(file_path)))
output_file.write('// DO NOT EDIT BY HAND!!!\n\n') output_file.write(u'// DO NOT EDIT BY HAND!!!\n\n')
output_file.write(output_str) output_file.write(output_str)
output_file.close() output_file.close()