<reponame>salesforce/threatshell
##############################################################################
# Author: <NAME> <<EMAIL>>
# Copyright (c) 2016, Salesforce.com, Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#     * Redistributions of source code must retain the above copyright notice,
#       this list of conditions and the following disclaimer.
#
#     * Redistributions in binary form must reproduce the above copyright
#       notice, this list of conditions and the following disclaimer in the
#       documentation and/or other materials provided with the distribution.
#
#     * Neither the name of Salesforce.com nor the names of its contributors
#       may be used to endorse or promote products derived from this software
#       without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
##############################################################################

import json
import logging

log = logging.getLogger(__name__)


class JSONEntry(dict):

    def __setattr__(self, key, value):
        self.__dict__[key] = value
        dict.__setitem__(self, key, value)

    def __getattr__(self, key):
        return self.__dict__.get(key)
        dict.__getitem__(self, key)

    def __init__(self, jdata):

        if jdata is None:
            return

        for key in jdata.keys():

            if not hasattr(self, key):
                log.warn(
                    (
                        "API change detected! %s object has no " +
                        "attribute '%s'"
                    ) % (type(self).__name__, key)
                )

            if(
                isinstance(getattr(self, key), list) and
                not isinstance(jdata[key], list)
            ):
                setattr(self, key, [jdata[key]])
            else:
                setattr(self, key, jdata[key])

    def __str__(self):
        return json.dumps(self, indent=4)

    def to_json(self):
        return str(self)
