from inspect import isclass

from json_to_models.dynamic_typing import (

ComplexType,

ModelMeta,

ModelPtr,

SingleType,

StringSerializable,

)

from json_to_models.generator import MetadataGenerator

from json_to_models.registry import ModelRegistry

from testing_tools.data import test_data

def _pprint_gen(value, key=None, lvl=0, empty_line=True, ignore_ptr=False):

if empty_line:

yield "\n"

if lvl > 0:

yield "\t" * lvl

else:

print(end=" ")

if key is not None:

yield f"{key} -> "

if isinstance(value, dict):

yield "object:"

for k, v in value.items():

yield from _pprint_gen(v, k, lvl=lvl + 1, ignore_ptr=ignore_ptr)

elif isinstance(value, list):

for t in value:

yield from _pprint_gen(t, lvl=lvl + 1, ignore_ptr=ignore_ptr)

# raise ValueError(value)

elif ignore_ptr and isinstance(value, ModelPtr):

yield f"Model#{value.type.index}: ..."

elif isinstance(value, ModelMeta):

yield str(value) + ":"

for key, sub_value in value.type.items():

yield from _pprint_gen(

sub_value, key, lvl=lvl + 1, ignore_ptr=ignore_ptr

)

if not value.type:

yield " <empty>"

elif isinstance(value, SingleType):

yield f"{type(value).__name__}:"

yield from _pprint_gen(

value.type, lvl=lvl, empty_line=False, ignore_ptr=ignore_ptr

)

elif isinstance(value, ComplexType):

yield f"{type(value).__name__}:"

for t in value.types:

yield from _pprint_gen(t, lvl=lvl + 1, ignore_ptr=ignore_ptr)

elif isclass(value) and issubclass(value, StringSerializable):

yield f"(type=<class '{value.__name__}'>)"

else:

yield f"(type={getattr(value, 'type', value)})"

def pretty_format_meta(value, ignore_ptr=False):

"""

Formatter function to prettify metadata on printout/STDOUT

Args:

value:

ignore_ptr:

Returns:

str

"""

return "".join(_pprint_gen(value, ignore_ptr=ignore_ptr))

if __name__ == "__main__":

gen = MetadataGenerator()

reg = ModelRegistry()

fields = gen.generate(*test_data)

model = reg.process_meta_data(fields)

print(pretty_format_meta(model))

print("\n" + "-" * 10, end="")

reg.merge_models(generator=gen)

for model in reg.models:

print(pretty_format_meta(model, ignore_ptr=True))