File: compress_json.py

package info (click to toggle)
nodejs 22.14.0%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 246,928 kB
  • sloc: cpp: 1,582,349; javascript: 582,017; ansic: 82,400; python: 60,561; sh: 4,009; makefile: 2,263; asm: 1,732; pascal: 1,565; perl: 248; lisp: 222; xml: 42
file content (33 lines) | stat: -rw-r--r-- 900 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#!/usr/bin/env python

import json
import struct
import sys
import zlib

try:
    xrange          # Python 2
    PY2 = True
except NameError:
    PY2 = False
    xrange = range  # Python 3


if __name__ == '__main__':
  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
  # with the size of the uncompressed data as a 24 bits BE unsigned integer.
  assert len(text) < 1 << 24, 'Uncompressed JSON must be < 16 MiB.'
  data = struct.pack('>I', len(text))[1:4] + data

  step = 20
  slices = (data[i:i+step] for i in xrange(0, len(data), step))
  slices = [','.join(str(ord(c) if PY2 else c) for c in s) for s in slices]
  text = ',\n'.join(slices)

  with open(sys.argv[2], 'w') as fp:
    fp.write(text)