Make j2y more resilient to differently-nested JSON input

This commit is contained in:
Danielle McLean 2023-10-05 18:28:39 +11:00
parent 1ac1dee58c
commit b24f37349a
Signed by: 00dani
GPG key ID: 52C059C3B22A753E

View file

@ -1,17 +1,20 @@
#!/usr/bin/env python3
from yaml import dump
try:
from yaml import CDumper as Dumper
except ImportError:
from yaml import Dumper
from collections.abc import Sized
import json, re, sys
def dict_representer(dumper, d):
node = dumper.represent_dict(d)
# Don't use YAML flow style for large dicts, because the flow style output
# only really looks good with a very small number of keys.
if len(d) > 5 or sum(len(v) for v in d.values()) > 30:
if len(d) > 5 or sum(len(v) for v in d.values() if isinstance(v, Sized)) > 30:
node.flow_style = False
return node
Dumper.add_representer(dict, dict_representer)