tools: fix Python 3 issues in inspector_protocol

PR-URL: https://github.com/nodejs/node/pull/29296
Reviewed-By: Jiawen Geng <technicalcute@gmail.com>
Reviewed-By: Trivikram Kamat <trivikr.dev@gmail.com>
This commit is contained in:
cclauss 2019-08-24 10:58:58 +02:00 committed by Rich Trott
parent b2cce432f3
commit 4662f67e38
2 changed files with 12 additions and 12 deletions

View File

@ -7,14 +7,16 @@ import zlib
try:
xrange # Python 2
PY2 = True
except NameError:
PY2 = False
xrange = range # Python 3
if __name__ == '__main__':
fp = open(sys.argv[1])
obj = json.load(fp)
text = json.dumps(obj, separators=(',', ':'))
with open(sys.argv[1]) as fp:
obj = json.load(fp)
text = json.dumps(obj, separators=(',', ':')).encode('utf-8')
data = zlib.compress(text, zlib.Z_BEST_COMPRESSION)
# To make decompression a little easier, we prepend the compressed data
@ -24,8 +26,8 @@ if __name__ == '__main__':
step = 20
slices = (data[i:i+step] for i in xrange(0, len(data), step))
slices = [','.join(str(ord(c)) for c in s) for s in slices]
slices = [','.join(str(ord(c) if PY2 else c) for c in s) for s in slices]
text = ',\n'.join(slices)
fp = open(sys.argv[2], 'w')
fp.write(text)
with open(sys.argv[2], 'w') as fp:
fp.write(text)

View File

@ -21,14 +21,12 @@ def main(argv):
parser.add_argument("json_file", help="The .json output file write.")
args = parser.parse_args(argv)
file_name = os.path.normpath(args.pdl_file)
input_file = open(file_name, "r")
pdl_string = input_file.read()
with open(file_name, "r") as input_file:
pdl_string = input_file.read()
protocol = pdl.loads(pdl_string, file_name, args.map_binary_to_string)
input_file.close()
output_file = open(os.path.normpath(args.json_file), 'wb')
json.dump(protocol, output_file, indent=4, separators=(',', ': '))
output_file.close()
with open(os.path.normpath(args.json_file), 'w') as output_file:
json.dump(protocol, output_file, indent=4, separators=(',', ': '))
if __name__ == '__main__':