Skip to content

Commit

Permalink
tools: fix Python 3 issues in inspector_protocol
Browse files Browse the repository at this point in the history
PR-URL: #29296
Reviewed-By: Jiawen Geng <technicalcute@gmail.com>
Reviewed-By: Trivikram Kamat <trivikr.dev@gmail.com>
  • Loading branch information
cclauss authored and BridgeAR committed Sep 4, 2019
1 parent 9a15002 commit 783c8ee
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 12 deletions.
14 changes: 8 additions & 6 deletions tools/compress_json.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,14 +7,16 @@

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
Expand All @@ -24,8 +26,8 @@

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)
10 changes: 4 additions & 6 deletions tools/inspector_protocol/convert_protocol_to_json.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__':
Expand Down

0 comments on commit 783c8ee

Please sign in to comment.