Merge pull request #186 from pre-commit/attempt_to_fix_json_hook

Attempt to fix json hook
This commit is contained in:
Anthony Sottile 2017-03-20 11:53:31 -07:00 committed by GitHub
commit 5bf6c09bfa
3 changed files with 9 additions and 12 deletions

View file

@ -1,5 +1,5 @@
- repo: https://github.com/pre-commit/pre-commit-hooks - repo: https://github.com/pre-commit/pre-commit-hooks
sha: 4c3cc8f3edc3c673b3a4abe2054e64f21d26772c sha: dc50b7f09c9612624c97e7f11fa346937c37f444
hooks: hooks:
- id: trailing-whitespace - id: trailing-whitespace
- id: end-of-file-fixer - id: end-of-file-fixer

View file

@ -1,10 +1,10 @@
from __future__ import print_function from __future__ import print_function
import argparse import argparse
import io
import json
import sys import sys
import simplejson
def check_json(argv=None): def check_json(argv=None):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
@ -14,8 +14,8 @@ def check_json(argv=None):
retval = 0 retval = 0
for filename in args.filenames: for filename in args.filenames:
try: try:
simplejson.load(open(filename)) json.load(io.open(filename, encoding='UTF-8'))
except (simplejson.JSONDecodeError, UnicodeDecodeError) as exc: except (ValueError, UnicodeDecodeError) as exc:
print('{}: Failed to json decode ({})'.format(filename, exc)) print('{}: Failed to json decode ({})'.format(filename, exc))
retval = 1 retval = 1
return retval return retval

View file

@ -27,9 +27,9 @@ def _get_pretty_format(contents, indent, ensure_ascii=True, sort_keys=True, top_
)) + "\n" # dumps does not end with a newline )) + "\n" # dumps does not end with a newline
def _autofix(filename, new_contents, encoding=None): def _autofix(filename, new_contents):
print("Fixing file {}".format(filename)) print("Fixing file {}".format(filename))
with io.open(filename, 'w', encoding=encoding) as f: with io.open(filename, 'w', encoding='UTF-8') as f:
f.write(new_contents) f.write(new_contents)
@ -100,7 +100,7 @@ def pretty_format_json(argv=None):
status = 0 status = 0
for json_file in args.filenames: for json_file in args.filenames:
with io.open(json_file, encoding='utf-8') as f: with io.open(json_file, encoding='UTF-8') as f:
contents = f.read() contents = f.read()
try: try:
@ -113,10 +113,7 @@ def pretty_format_json(argv=None):
print("File {} is not pretty-formatted".format(json_file)) print("File {} is not pretty-formatted".format(json_file))
if args.autofix: if args.autofix:
_autofix( _autofix(json_file, pretty_contents)
json_file, pretty_contents,
encoding='utf-8' if args.no_ensure_ascii else None,
)
status = 1 status = 1