craftbeerpi4-pione/cbpi/utils/encoder.py

from json import JSONEncoder

from cbpi.database.model import ActorModel, SensorModel


class ComplexEncoder(JSONEncoder):

    def default(self, obj):

        from cbpi.database.orm_framework import DBModel

        try:

            if isinstance(obj, ActorModel):
                data =  dict(**obj.__dict__, state=obj.instance.get_state())
                del data["instance"]
                return data

            elif isinstance(obj, SensorModel):
                data =  dict(**obj.__dict__, state=obj.instance.get_state(), value=obj.instance.get_value())
                del data["instance"]
                return data
            #elif callable(getattr(obj, "reprJSON")):
            #    return obj.reprJSON()
            elif isinstance(obj, DBModel):
                return None
            #elif hasattr(obj, "callback"):
            #    return obj()
            else:
                raise TypeError()
        except TypeError:
            pass
        return None