Add history route

3 files changed, 27 insertions, 0 deletions

diff --git a/poller/__init__.py b/poller/__init__.py
index c0c25c0..1e1580b 100644
--- a/poller/__init__.py
+++ b/poller/__init__.py
@@ -30,5 +30,6 @@ APP.logger.info('SQLAlchemy pointed at ' + repr(db.engine.url))
 #pylint: disable=wrong-import-position
 from . import models
 from . import commands
+from . import routes
 
 db.create_all()
diff --git a/poller/models.py b/poller/models.py
index a5dd2fa..15b223d 100644
--- a/poller/models.py
+++ b/poller/models.py
@@ -32,3 +32,22 @@ class Day(db.Model):
         Helper to get all values from the database
         """
         return cls.query.all()
+
+    def serialize(self):
+        """
+        used for json serialization
+        """
+        return {
+            'last_updated': self.last_updated,
+            'alert_level': self.alert_level,
+            'beds_available': self.beds_available,
+            'isolation_off_campus': self.isolation_off_campus,
+            'isolation_on_campus': self.isolation_on_campus,
+            'new_staff': self.new_staff,
+            'new_students': self.new_students,
+            'quarantine_off_campus': self.quarantine_off_campus,
+            'quarantine_on_campus': self.quarantine_on_campus,
+            'tests_administered': self.tests_administered,
+            'total_staff': self.total_staff,
+            'total_students': self.total_students,
+        }
diff --git a/poller/routes.py b/poller/routes.py
new file mode 100644
index 0000000..4c0010b
--- /dev/null
+++ b/poller/routes.py
@@ -0,0 +1,7 @@
+from flask import Flask, jsonify
+from . import APP
+from .models import Day
+
+@APP.route('/api/v0/history')
+def _get_api_v0_history():
+    return jsonify([day.serialize() for day in Day.get_all()])