mirror of
https://github.com/google/brotli.git
synced 2024-11-21 19:20:09 +00:00
Python: Format bro.py with yapf (#480)
This commit is contained in:
Alex Nicksay
Eugene Kliuchnikov
parent
0ee416139f
commit
4651f7c000
132
python/bro.py
132
python/bro.py
@ -1,13 +1,13 @@
|
||||
#! /usr/bin/env python
|
||||
"""bro %s -- compression/decompression utility using the Brotli algorithm."""
|
||||
"""Compression/decompression utility using the Brotli algorithm."""
|
||||
|
||||
from __future__ import print_function
|
||||
import argparse
|
||||
import sys
|
||||
import os
|
||||
import brotli
|
||||
import platform
|
||||
|
||||
import brotli
|
||||
|
||||
# default values of encoder parameters
|
||||
DEFAULT_PARAMS = {
|
||||
@ -22,15 +22,15 @@ def get_binary_stdio(stream):
|
||||
""" Return the specified standard input, output or errors stream as a
|
||||
'raw' buffer object suitable for reading/writing binary data from/to it.
|
||||
"""
|
||||
assert stream in ['stdin', 'stdout', 'stderr'], "invalid stream name"
|
||||
assert stream in ['stdin', 'stdout', 'stderr'], 'invalid stream name'
|
||||
stdio = getattr(sys, stream)
|
||||
if sys.version_info[0] < 3:
|
||||
if sys.platform == 'win32':
|
||||
# set I/O stream binary flag on python2.x (Windows)
|
||||
runtime = platform.python_implementation()
|
||||
if runtime == "PyPy":
|
||||
if runtime == 'PyPy':
|
||||
# the msvcrt trick doesn't work in pypy, so I use fdopen
|
||||
mode = "rb" if stream == "stdin" else "wb"
|
||||
mode = 'rb' if stream == 'stdin' else 'wb'
|
||||
stdio = os.fdopen(stdio.fileno(), mode, 0)
|
||||
else:
|
||||
# this works with CPython -- untested on other implementations
|
||||
@ -42,45 +42,85 @@ def get_binary_stdio(stream):
|
||||
if hasattr(stdio, 'buffer'):
|
||||
return stdio.buffer
|
||||
else:
|
||||
orig_stdio = getattr(sys, "__%s__" % stream)
|
||||
orig_stdio = getattr(sys, '__%s__' % stream)
|
||||
return orig_stdio.buffer
|
||||
|
||||
|
||||
def main(args=None):
|
||||
|
||||
parser = argparse.ArgumentParser(
|
||||
prog='bro.py',
|
||||
description="Compression/decompression utility using the Brotli algorithm.")
|
||||
parser.add_argument('--version', action='version', version=brotli.__version__)
|
||||
parser.add_argument('-i', '--input', metavar='FILE', type=str, dest='infile',
|
||||
help='Input file', default=None)
|
||||
parser.add_argument('-o', '--output', metavar='FILE', type=str, dest='outfile',
|
||||
help='Output file', default=None)
|
||||
parser.add_argument('-f', '--force', action='store_true',
|
||||
help='Overwrite existing output file', default=False)
|
||||
parser.add_argument('-d', '--decompress', action='store_true',
|
||||
help='Decompress input file', default=False)
|
||||
prog=os.path.basename(__file__), description=__doc__)
|
||||
parser.add_argument(
|
||||
'--version', action='version', version=brotli.__version__)
|
||||
parser.add_argument(
|
||||
'-i',
|
||||
'--input',
|
||||
metavar='FILE',
|
||||
type=str,
|
||||
dest='infile',
|
||||
help='Input file',
|
||||
default=None)
|
||||
parser.add_argument(
|
||||
'-o',
|
||||
'--output',
|
||||
metavar='FILE',
|
||||
type=str,
|
||||
dest='outfile',
|
||||
help='Output file',
|
||||
default=None)
|
||||
parser.add_argument(
|
||||
'-f',
|
||||
'--force',
|
||||
action='store_true',
|
||||
help='Overwrite existing output file',
|
||||
default=False)
|
||||
parser.add_argument(
|
||||
'-d',
|
||||
'--decompress',
|
||||
action='store_true',
|
||||
help='Decompress input file',
|
||||
default=False)
|
||||
params = parser.add_argument_group('optional encoder parameters')
|
||||
params.add_argument('-m', '--mode', metavar="MODE", type=int, choices=[0, 1, 2],
|
||||
help='The compression mode can be 0 for generic input, '
|
||||
'1 for UTF-8 encoded text, or 2 for WOFF 2.0 font data. '
|
||||
'Defaults to 0.')
|
||||
params.add_argument('-q', '--quality', metavar="QUALITY", type=int,
|
||||
choices=list(range(0, 12)),
|
||||
help='Controls the compression-speed vs compression-density '
|
||||
'tradeoff. The higher the quality, the slower the '
|
||||
'compression. Range is 0 to 11. Defaults to 11.')
|
||||
params.add_argument('--lgwin', metavar="LGWIN", type=int,
|
||||
choices=list(range(10, 25)),
|
||||
help='Base 2 logarithm of the sliding window size. Range is '
|
||||
'10 to 24. Defaults to 22.')
|
||||
params.add_argument('--lgblock', metavar="LGBLOCK", type=int,
|
||||
choices=[0] + list(range(16, 25)),
|
||||
help='Base 2 logarithm of the maximum input block size. '
|
||||
'Range is 16 to 24. If set to 0, the value will be set based '
|
||||
'on the quality. Defaults to 0.')
|
||||
params.add_argument('--custom-dictionary', metavar="FILE", type=str, dest='dictfile',
|
||||
help='Custom dictionary file.', default = None)
|
||||
params.add_argument(
|
||||
'-m',
|
||||
'--mode',
|
||||
metavar='MODE',
|
||||
type=int,
|
||||
choices=[0, 1, 2],
|
||||
help='The compression mode can be 0 for generic input, '
|
||||
'1 for UTF-8 encoded text, or 2 for WOFF 2.0 font data. '
|
||||
'Defaults to 0.')
|
||||
params.add_argument(
|
||||
'-q',
|
||||
'--quality',
|
||||
metavar='QUALITY',
|
||||
type=int,
|
||||
choices=list(range(0, 12)),
|
||||
help='Controls the compression-speed vs compression-density '
|
||||
'tradeoff. The higher the quality, the slower the '
|
||||
'compression. Range is 0 to 11. Defaults to 11.')
|
||||
params.add_argument(
|
||||
'--lgwin',
|
||||
metavar='LGWIN',
|
||||
type=int,
|
||||
choices=list(range(10, 25)),
|
||||
help='Base 2 logarithm of the sliding window size. Range is '
|
||||
'10 to 24. Defaults to 22.')
|
||||
params.add_argument(
|
||||
'--lgblock',
|
||||
metavar='LGBLOCK',
|
||||
type=int,
|
||||
choices=[0] + list(range(16, 25)),
|
||||
help='Base 2 logarithm of the maximum input block size. '
|
||||
'Range is 16 to 24. If set to 0, the value will be set based '
|
||||
'on the quality. Defaults to 0.')
|
||||
params.add_argument(
|
||||
'--custom-dictionary',
|
||||
metavar='FILE',
|
||||
type=str,
|
||||
dest='dictfile',
|
||||
help='Custom dictionary file.',
|
||||
default=None)
|
||||
# set default values using global DEFAULT_PARAMS dictionary
|
||||
parser.set_defaults(**DEFAULT_PARAMS)
|
||||
|
||||
@ -89,7 +129,7 @@ def main(args=None):
|
||||
if options.infile:
|
||||
if not os.path.isfile(options.infile):
|
||||
parser.error('file "%s" not found' % options.infile)
|
||||
with open(options.infile, "rb") as infile:
|
||||
with open(options.infile, 'rb') as infile:
|
||||
data = infile.read()
|
||||
else:
|
||||
if sys.stdin.isatty():
|
||||
@ -101,28 +141,32 @@ def main(args=None):
|
||||
if options.outfile:
|
||||
if os.path.isfile(options.outfile) and not options.force:
|
||||
parser.error('output file exists')
|
||||
outfile = open(options.outfile, "wb")
|
||||
outfile = open(options.outfile, 'wb')
|
||||
else:
|
||||
outfile = get_binary_stdio('stdout')
|
||||
|
||||
if options.dictfile:
|
||||
if not os.path.isfile(options.dictfile):
|
||||
parser.error('file "%s" not found' % options.dictfile)
|
||||
with open(options.dictfile, "rb") as dictfile:
|
||||
with open(options.dictfile, 'rb') as dictfile:
|
||||
custom_dictionary = dictfile.read()
|
||||
else:
|
||||
custom_dictionary = ''
|
||||
|
||||
|
||||
try:
|
||||
if options.decompress:
|
||||
data = brotli.decompress(data, dictionary=custom_dictionary)
|
||||
else:
|
||||
data = brotli.compress(
|
||||
data, mode=options.mode, quality=options.quality,
|
||||
lgwin=options.lgwin, lgblock=options.lgblock, dictionary=custom_dictionary)
|
||||
data,
|
||||
mode=options.mode,
|
||||
quality=options.quality,
|
||||
lgwin=options.lgwin,
|
||||
lgblock=options.lgblock,
|
||||
dictionary=custom_dictionary)
|
||||
except brotli.error as e:
|
||||
parser.exit(1,'bro: error: %s: %s' % (e, options.infile or 'sys.stdin'))
|
||||
parser.exit(1,
|
||||
'bro: error: %s: %s' % (e, options.infile or 'sys.stdin'))
|
||||
|
||||
outfile.write(data)
|
||||
outfile.close()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user