mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-09-19 22:07:28 +02:00
33 lines
992 B
Python
33 lines
992 B
Python
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 obj.__dict__
|
|
#elif hasattr(obj, "callback"):
|
|
# return obj()
|
|
else:
|
|
raise TypeError()
|
|
except TypeError:
|
|
pass
|
|
return None
|