From: Philipp Spitzer <philipp@spitzer.priv.at>
Date: Wed, 18 Jul 2018 20:24:06 +0000 (+0200)
Subject: Move to_dict to JSONEncoder.
X-Git-Url: https://git.toastfreeware.priv.at/chrisu/seepark.git/commitdiff_plain/5a114b9646f8cd7a3972e3498deaf406905b3393

Move to_dict to JSONEncoder.
---

diff --git a/web/seepark_web.py b/web/seepark_web.py
index c966321..fc28fde 100644
--- a/web/seepark_web.py
+++ b/web/seepark_web.py
@@ -16,10 +16,18 @@ sys.path.append(lib_path)
 from seeparklib.openweathermap import openweathermap_json, OpenWeatherMapError
 
 
+# https://stackoverflow.com/a/37350445
+def sqlalchemy_model_to_dict(model):
+    return {c.key: getattr(model, c.key)
+        for c in inspect(model).mapper.column_attrs}
+
+
 class JSONEncoder(flask.json.JSONEncoder):
     def default(self, object):
         if isinstance(object, datetime.datetime):
             return object.isoformat()
+        elif isinstance(object, db.Model):
+            return sqlalchemy_model_to_dict(object)
         return super().default(object)
 
 
@@ -52,10 +60,6 @@ db.reflect(app=app)
 class Sensors(db.Model):
     __tablename__ = 'sensors'
 
-    # https://stackoverflow.com/a/37350445
-    def to_dict(self):
-        return {c.key: getattr(self, c.key)
-            for c in inspect(self).mapper.column_attrs}
 
 def select_sensordata(initial_where):
     query = Sensors.query.filter(initial_where)
@@ -101,7 +105,7 @@ def select_sensordata(initial_where):
             c3result[row.sensor_id + '_x'].append(dt)
         return c3result
     else:
-        return [row.to_dict() for row in result]
+        return result
 
 
 def currentairtemperature(apikey, cityid):