bazarr/libs/ftfy/cli.py

151 lines
4.1 KiB
Python

"""
A command-line utility for fixing text found in a file.
"""
import os
import sys
from ftfy import __version__, fix_file, TextFixerConfig
ENCODE_ERROR_TEXT_UNIX = """ftfy error:
Unfortunately, this output stream does not support Unicode.
Your system locale may be very old or misconfigured. You should use a locale
that supports UTF-8. One way to do this is to `export LANG=C.UTF-8`.
"""
ENCODE_ERROR_TEXT_WINDOWS = """ftfy error:
Unfortunately, this output stream does not support Unicode.
You might be trying to output to the Windows Command Prompt (cmd.exe), which
does not fully support Unicode for historical reasons. In general, we recommend
finding a way to run Python without using cmd.exe.
You can work around this problem by using the '-o filename' option in ftfy to
output to a file instead.
"""
DECODE_ERROR_TEXT = """ftfy error:
This input couldn't be decoded as %r. We got the following error:
%s
ftfy works best when its input is in a known encoding. You can use `ftfy -g`
to guess, if you're desperate. Otherwise, give the encoding name with the
`-e` option, such as `ftfy -e latin-1`.
"""
SAME_FILE_ERROR_TEXT = """ftfy error:
Can't read and write the same file. Please output to a new file instead.
"""
def main():
"""
Run ftfy as a command-line utility.
"""
import argparse
parser = argparse.ArgumentParser(
description="ftfy (fixes text for you), version %s" % __version__
)
parser.add_argument(
'filename',
default='-',
nargs='?',
help='The file whose Unicode is to be fixed. Defaults '
'to -, meaning standard input.',
)
parser.add_argument(
'-o',
'--output',
type=str,
default='-',
help='The file to output to. Defaults to -, meaning ' 'standard output.',
)
parser.add_argument(
'-g',
'--guess',
action='store_true',
help="Ask ftfy to guess the encoding of your input. "
"This is risky. Overrides -e.",
)
parser.add_argument(
'-e',
'--encoding',
type=str,
default='utf-8',
help='The encoding of the input. Defaults to UTF-8.',
)
parser.add_argument(
'-n',
'--normalization',
type=str,
default='NFC',
help='The normalization of Unicode to apply. '
'Defaults to NFC. Can be "none".',
)
parser.add_argument(
'--preserve-entities',
action='store_true',
help="Leave HTML entities as they are. The default "
"is to decode them, as long as no HTML tags "
"have appeared in the file.",
)
args = parser.parse_args()
encoding = args.encoding
if args.guess:
encoding = None
if args.filename == '-':
# Get a standard input stream made of bytes, so we can decode it as
# whatever encoding is necessary.
file = sys.stdin.buffer
else:
file = open(args.filename, 'rb')
if args.output == '-':
outfile = sys.stdout
else:
if os.path.realpath(args.output) == os.path.realpath(args.filename):
sys.stderr.write(SAME_FILE_ERROR_TEXT)
sys.exit(1)
outfile = open(args.output, 'w', encoding='utf-8')
normalization = args.normalization
if normalization.lower() == 'none':
normalization = None
if args.preserve_entities:
unescape_html = False
else:
unescape_html = 'auto'
config = TextFixerConfig(
unescape_html=unescape_html,
normalization=normalization
)
try:
for line in fix_file(
file,
encoding=encoding,
config=config
):
try:
outfile.write(line)
except UnicodeEncodeError:
if sys.platform == 'win32':
sys.stderr.write(ENCODE_ERROR_TEXT_WINDOWS)
else:
sys.stderr.write(ENCODE_ERROR_TEXT_UNIX)
sys.exit(1)
except UnicodeDecodeError as err:
sys.stderr.write(DECODE_ERROR_TEXT % (encoding, err))
sys.exit(1)
if __name__ == '__main__':
main()