pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2025-08-14 09:27:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

this is much cleaner and might actually get all the coverage with out a bunch of work

Browse Source
This commit is contained in:
dmlb2000
2016-11-03 15:47:21 -07:00
parent 845a3d5bdf
commit d06a515ce1
1 changed files with 14 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

100
pre_commit_hooks/pretty_format_json.py
View File

@ -6,95 +6,23 @@ from collections import OrderedDict
import simplejson
class SortableOrderedDict(OrderedDict):
"""Performs an in-place sort of the keys if you want."""
def sort(*args, **kwds):
self = args[0]
args = args[1:]
if 'key' not in kwds:
kwds['key'] = lambda x: x[0]
if len(args):
raise TypeError('expected no positional arguments got {0}'.format(len(args)))
sorted_od = sorted([x for x in self.items()], **kwds)
self.clear()
self.update(sorted_od)
class TrackedSod(SortableOrderedDict):
"""Tracks instances of the SortableOrderedDict."""
_instances = []
def __init__(self, *args, **kwds):
super(TrackedSod, self).__init__(*args, **kwds)
self.__track(self)
@classmethod
def __track(cls, obj):
cls._instances.append(obj)
def _get_pretty_format(contents, indent, sort_keys=True, top_keys=[]):
class KeyToCmp(object):
def __init__(self, obj, *args):
self.obj = obj[0]
def __lt__(self, other):
if self.obj in top_keys and other.obj in top_keys:
return top_keys.index(self.obj) < top_keys.index(other.obj)
elif self.obj in top_keys and other.obj not in top_keys:
return True
elif self.obj not in top_keys and other.obj in top_keys:
return False
else:
return self.obj < other.obj
def __gt__(self, other):
if self.obj in top_keys and other.obj in top_keys:
return top_keys.index(self.obj) > top_keys.index(other.obj)
elif self.obj in top_keys and other.obj not in top_keys:
return False
elif self.obj not in top_keys and other.obj in top_keys:
return True
else:
return self.obj > other.obj
def __eq__(self, other):
if self.obj in top_keys and other.obj in top_keys:
return top_keys.index(self.obj) == top_keys.index(other.obj)
elif self.obj in top_keys and other.obj not in top_keys:
return False
elif self.obj not in top_keys and other.obj in top_keys:
return False
else:
return self.obj == other.obj
def __le__(self, other):
if self.obj in top_keys and other.obj in top_keys:
return top_keys.index(self.obj) <= top_keys.index(other.obj)
elif self.obj in top_keys and other.obj not in top_keys:
return True
elif self.obj not in top_keys and other.obj in top_keys:
return False
else:
return self.obj <= other.obj
def __ge__(self, other):
if self.obj in top_keys and other.obj in top_keys:
return top_keys.index(self.obj) >= top_keys.index(other.obj)
elif self.obj in top_keys and other.obj not in top_keys:
return False
elif self.obj not in top_keys and other.obj in top_keys:
return True
else:
return self.obj >= other.obj
def __ne__(self, other):
if self.obj in top_keys and other.obj in top_keys:
return top_keys.index(self.obj) != top_keys.index(other.obj)
elif self.obj in top_keys and other.obj not in top_keys:
return False
elif self.obj not in top_keys and other.obj in top_keys:
return False
else:
return self.obj != other.obj
py_obj = simplejson.loads(contents, object_pairs_hook=TrackedSod)
if sort_keys:
for tsod in TrackedSod._instances:
tsod.sort(key=KeyToCmp)
# dumps don't end with a newline
return simplejson.dumps(py_obj, indent=indent) + "\n"
def pairs_first(pairs):
before = [pair for pair in pairs if pair[0] in top_keys]
before = sorted(before, key=lambda x: top_keys.index(x[0]))
after = [pair for pair in pairs if pair[0] not in top_keys]
if sort_keys:
after = sorted(after, key=lambda x: x[0])
return OrderedDict(before + after)
return simplejson.dumps(
simplejson.loads(
contents,
object_pairs_hook=pairs_first,
),
indent=indent
) + "\n" # dumps don't end with a newline
def _autofix(filename, new_contents):
print("Fixing file {0}".format(filename))