c0fec0de/anytree

# -*- coding: utf-8 -*-
import json

from .dictimporter import DictImporter


class JsonImporter:
    """
    Import Tree from JSON.

    The JSON is read and converted to a dictionary via `dictimporter`.

    Keyword Arguments:
        dictimporter: Dictionary Importer used (see :any:`DictImporter`).
        kwargs: All other arguments are passed to
                :any:`json.load`/:any:`json.loads`.
                See documentation for reference.

    >>> from anytree.importer import JsonImporter
    >>> from anytree import RenderTree
    >>> importer = JsonImporter()
    >>> data = '''
    ... {
    ...   "a": "root",
    ...   "children": [
    ...     {
    ...       "a": "sub0",
    ...       "children": [
    ...         {
    ...           "a": "sub0A",
    ...           "b": "foo"
    ...         },
    ...         {
    ...           "a": "sub0B"
    ...         }
    ...       ]
    ...     },
    ...     {
    ...       "a": "sub1"
    ...     }
    ...   ]
    ... }'''
    >>> root = importer.import_(data)
    >>> print(RenderTree(root))
    AnyNode(a='root')
    ├── AnyNode(a='sub0')
    │   ├── AnyNode(a='sub0A', b='foo')
    │   └── AnyNode(a='sub0B')
    └── AnyNode(a='sub1')
    """

    def __init__(self, dictimporter=None, **kwargs):
        self.dictimporter = dictimporter
        self.kwargs = kwargs

    def __import(self, data):
        dictimporter = self.dictimporter or DictImporter()
        return dictimporter.import_(data)

    def import_(self, data):
        """Read JSON from `data`."""
        return self.__import(json.loads(data, **self.kwargs))

    def read(self, filehandle):
        """Read JSON from `filehandle`."""
        return self.__import(json.load(filehandle, **self.kwargs))