Source code for osaca.parser.attr_dict

#!/usr/bin/env python3
"""Attribute Dictionary to access dictionary entries as attributes."""


class AttrDict(dict):
    def __init__(self, *args, **kwargs):
        super(AttrDict, self).__init__(*args, **kwargs)
        self.__dict__ = self

    @staticmethod
    def convert_dict(dictionary):
        """
        Convert given dictionary to `AttrDict`.

        :param dictionary: `dict` to be converted
        :type dictionary: `dict`
        :returns: `AttrDict` representation of ``dictionary``
        """
        if isinstance(dictionary, type(list())):
            return [AttrDict.convert_dict(x) for x in dictionary]
        if isinstance(dictionary, type(dict())):
            for key in list(dictionary.keys()):
                entry = dictionary[key]
                if isinstance(entry, type(dict())) or isinstance(entry, type(AttrDict())):
                    dictionary[key] = AttrDict.convert_dict(dictionary[key])
                if isinstance(entry, type(list())):
                    dictionary[key] = [AttrDict.convert_dict(x) for x in entry]
            return AttrDict(dictionary)
        return dictionary

    @staticmethod
    def get_dict(attrdict):
        """
        Convert given `AttrDict` to a standard dictionary.

        :param attrdict: `AttrDict` to be converted
        :type attrdict: `AttrDict`
        :returns: `dict` representation of ``AttrDict``
        """
        if isinstance(attrdict, type(list())):
            return [AttrDict.get_dict(x) for x in attrdict]
        if isinstance(attrdict, type(AttrDict())):
            newdict = {}
            for key in list(attrdict.keys()):
                entry = attrdict[key]
                if isinstance(entry, type(dict())) or isinstance(entry, type(AttrDict())):
                    newdict[key] = AttrDict.get_dict(attrdict[key])
                elif isinstance(entry, type(list())):
                    newdict[key] = [AttrDict.get_dict(x) for x in entry]
                else:
                    newdict[key] = entry
            return newdict
        return attrdict