]> ToastFreeware Gitweb - chrisu/seepark.git/blobdiff - web/seepark_web.py
Fix error in PDF report generation when no water data is available.
[chrisu/seepark.git] / web / seepark_web.py
index 0ba8f57e5132d8ddf368538e0a37afe4dbc8297b..a2d77063d44384a6049c039f77099826b015b7a5 100644 (file)
@@ -1,4 +1,6 @@
+import collections
 import datetime
+import itertools
 import time
 import configparser
 import os
@@ -13,8 +15,8 @@ from matplotlib.backends.backend_pdf import PdfPages
 
 from flask import Flask, render_template, jsonify, request, abort, Response, make_response
 import flask.json
-from flask_sqlalchemy import SQLAlchemy, inspect
-from sqlalchemy import func
+from flask_sqlalchemy import SQLAlchemy
+from sqlalchemy import func, inspect
 
 MONTH_DE = [
     'Jänner',
@@ -59,6 +61,11 @@ def parse_datetime(date_str):
     return datetime.datetime.strptime(date_str, '%Y-%m-%dT%H:%M:%S')
 
 
+def ntimes(it, n):
+    for v in it:
+        yield from itertools.repeat(v, n)
+
+
 def get_sqlalchemy_database_uri(config):
     user = config.get('database', 'user')
     pwd = config.get('database', 'password')
@@ -78,7 +85,8 @@ app.json_encoder = JSONEncoder
 app.config['SQLALCHEMY_DATABASE_URI'] = get_sqlalchemy_database_uri(config)
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
 db = SQLAlchemy(app)
-db.reflect(app=app)
+with app.app_context():
+    db.reflect()
 
 
 class Sensors(db.Model):
@@ -258,8 +266,15 @@ def currentwatertemperature(sensorid):
     return result.value, result.timestamp
 
 
-def add_month(date):
-    return (date + datetime.timedelta(days=42)).replace(day=1)
+def first_of_month(date, month):
+    date = date.replace(day=1)
+    if month == 0:
+        return date
+    if month == 1:
+        return (date + datetime.timedelta(days=42)).replace(day=1)
+    if month == -1:
+        return (date - datetime.timedelta(days=1)).replace(day=1)
+    assert False
 
 
 @app.route('/api/<version>/sensors/')
@@ -274,10 +289,11 @@ def sensorid(version, sensor_id):
     """Return all data for a specific sensor
 
     URL parameters:
-    begin=<datetime>, optional, format like "2018-05-19T21:07:53"
-    end=<datetime>, optional, format like "2018-05-19T21:07:53"
-    mode=<full|consolidated>, optional. return all rows (default) or with lower resolution (for charts)
-    format=<default|c3>, optional. return result as returned by sqlalchemy (default) or formatted for c3.js
+
+    * ``begin=<datetime>``, optional, format like ``2018-05-19T21:07:53``
+    * ``end=<datetime>``, optional, format like ``2018-05-19T21:07:53``
+    * ``mode=<full|consolidated>``, optional. return all rows (default) or with lower resolution (for charts)
+    * ``format=<default|c3>``, optional. return result as returned by sqlalchemy (default) or formatted for c3.js
     """
     result = sensordata(sensor_id=sensor_id)
     return jsonify(result)
@@ -288,10 +304,11 @@ def sensortype(version, sensor_type):
     """Return all data for a specific sensor type
 
     URL parameters:
-    begin=<datetime>, optional, format like "2018-05-19T21:07:53"
-    end=<datetime>, optional, format like "2018-05-19T21:07:53"
-    mode=<full|consolidated>, optional. return all rows (default) or with lower resolution (for charts)
-    format=<default|c3>, optional. return result as returned by sqlalchemy (default) or formatted for c3.js
+
+    * ``begin=<datetime>``, optional, format like ``2018-05-19T21:07:53``
+    * ``end=<datetime>``, optional, format like ``2018-05-19T21:07:53``
+    * ``mode=<full|consolidated>``, optional. return all rows (default) or with lower resolution (for charts)
+    * ``format=<default|c3>``, optional. return result as returned by sqlalchemy (default) or formatted for c3.js
     """
     result = sensordata(sensor_type=sensor_type)
     return jsonify(result)
@@ -323,12 +340,14 @@ def currentwater(version):
     return jsonify({"value": value, "timestamp": timestamp})
 
 
-@app.route('/report/<int:year>-<int:month>')
+@app.route('/report/<int(fixed_digits=4):year>/<int(fixed_digits=2):month>')
 def report(year, month):
+    """Report for given year (4 digits) and month (2 digits)
+    """
     paper_size = (29.7 / 2.54, 21. / 2.54)  # A4
 
     begin = datetime.datetime(year, month, 1)
-    end = add_month(begin)
+    end = first_of_month(begin, 1)
 
     water_data = sensordata_to_xy(select_sensordata(mainsensor, 'Wassertemperatur', begin, end))
     air_data = openweatherdata_to_xy(select_openweatherdata(cityid, begin, end))
@@ -348,9 +367,11 @@ def report(year, month):
 
         # graphic
         plt.figure(figsize=paper_size)
+        report_colors = []
         for label, data in sorted(report_data.items(), reverse=True):
             x, y = data
-            plt.plot(x, y, label=label)
+            lines = plt.plot(x, y, label=label)
+            report_colors.append(lines[0].get_color())
         plt.xticks(days_datetime, [''] * len(days_datetime))
         plt.ylabel('Temperatur in °C')
         plt.axis(xmin=begin, xmax=end)
@@ -373,15 +394,19 @@ def report(year, month):
                 x, y = data
                 for d in days_datetime:
                     report_datetime = datetime.datetime.combine(d.date(), t)
-                    closest_index = np.argmin(np.abs(x - report_datetime))
-                    if abs(x[closest_index] - report_datetime) > datetime.timedelta(hours=1):
+                    if len(x) == 0:
                         cell = 'N/A'
                     else:
-                        value = y[closest_index]
-                        cell = '{:.1f}'.format(value)
+                        closest_index = np.argmin(np.abs(x - report_datetime))
+                        if abs(x[closest_index] - report_datetime) > datetime.timedelta(hours=1):
+                            cell = 'N/A'
+                        else:
+                            value = y[closest_index]
+                            cell = '{:.1f}'.format(value)
                     row_cells.append(cell)
                 cells.append(row_cells)
-        table = plt.table(cellText=cells, colLabels=columns, rowLabels=rows, loc='bottom')
+        row_colors = list(ntimes(report_colors + ['w'], len(report_times)))
+        table = plt.table(cellText=cells, colLabels=columns, rowLabels=rows, rowColours=row_colors, loc='bottom')
         table.scale(xscale=1, yscale=2)
         plt.title(title)
         plt.subplots_adjust(left=0.15, right=0.97, bottom=0.3)  # do not cut row labels
@@ -405,6 +430,8 @@ def report(year, month):
 def index():
     airvalue, airtime     = currentairtemperature(cityid)
     watervalue, watertime = currentwatertemperature(mainsensor)
+    this_month = first_of_month(datetime.date.today(), 0)
+    last_month = first_of_month(this_month, -1)
 
     return render_template(
         'seepark_web.html',
@@ -413,4 +440,6 @@ def index():
         watertime=watertime,
         airvalue=airvalue,
         airtime=airtime,
+        this_month=this_month,
+        last_month=last_month
     )