Now the generated PDF contains only 1 page and in A4 landscape mode.

[chrisu/seepark.git] / web / seepark_web.py

diff --git a/web/seepark_web.py b/web/seepark_web.py
index 59e0c8b0264d8d0a0a9067afea2e2f9c1dc82e9c..b899a1730ebdfaea506fd652c7993f0d3b2d2779 100644 (file)
--- a/web/seepark_web.py
+++ b/web/seepark_web.py
@@ -4,11 +4,41 @@ import configparser
 import os
 import sys
 from collections import defaultdict
 import os
 import sys
 from collections import defaultdict

-from flask import Flask, render_template, jsonify, request, abort, Response
+import io
+import numpy as np
+import matplotlib
+matplotlib.use('pdf')
+import matplotlib.pyplot as plt
+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
 
 import flask.json
 from flask_sqlalchemy import SQLAlchemy, inspect
 from sqlalchemy import func
 

+MONTH_DE = [
+    'Jänner',
+    'Februar',
+    'März',
+    'April',
+    'Mai',
+    'Juni',
+    'Juli',
+    'August',
+    'September',
+    'Oktober',
+    'November',
+    'Dezember']
+
+DAY_OF_WEEK_DE = [
+    'Montag',
+    'Dienstag',
+    'Mittwoch',
+    'Donnerstag',
+    'Freitag',
+    'Samstag',
+    'Sonntag']
+

 
 # https://stackoverflow.com/a/37350445
 def sqlalchemy_model_to_dict(model):
 
 # https://stackoverflow.com/a/37350445
 def sqlalchemy_model_to_dict(model):


@@ -62,8 +92,9 @@ class OpenWeatherMap(db.Model):
 def calc_grouping_resolution(begin, end):
     """How many data points should be between the timestamps begin and end?"""
     # copied from munin/master/_bin/munin-cgi-graph.in
 def calc_grouping_resolution(begin, end):
     """How many data points should be between the timestamps begin and end?"""
     # copied from munin/master/_bin/munin-cgi-graph.in

+    # except day: 300 -> 600

     resolutions = dict(
     resolutions = dict(

-        day   =   300,
+        day   =   600,

         week  =  1800,
         month =  7200,
         year  = 86400,
         week  =  1800,
         month =  7200,
         year  = 86400,


@@ -213,6 +244,10 @@ def currentwatertemperature(sensorid):
     return result.value, result.timestamp
 
 
     return result.value, result.timestamp
 
 

+def add_month(date):
+    return (date + datetime.timedelta(days=42)).replace(day=1)
+
+

 @app.route('/api/<version>/sensors/')
 def sensors(version):
     """List all sensors found in the database"""
 @app.route('/api/<version>/sensors/')
 def sensors(version):
     """List all sensors found in the database"""


@@ -262,6 +297,89 @@ def openweathermap_city(version, cityid):
     return jsonify(result)
 
 
     return jsonify(result)
 
 

+@app.route('/api/<version>/currentairtemperature')
+def currentair(version):
+    value, timestamp = currentairtemperature(cityid)
+    return jsonify({"value": value, "timestamp": timestamp})
+
+
+@app.route('/api/<version>/currentwatertemperature')
+def currentwater(version):
+    value, timestamp = currentwatertemperature(mainsensor)
+    return jsonify({"value": value, "timestamp": timestamp})
+
+
+@app.route('/report/<int:year>-<int:month>')
+def report(year, month):
+
+    begin = datetime.datetime(year, month, 1)
+    end = add_month(begin)
+    data = list(select_sensordata(mainsensor, 'Wassertemperatur', begin, end))
+    x = np.array([d.timestamp for d in data])
+    y = np.array([d.value for d in data])
+
+    days_datetime = []
+    d = begin
+    while d < end:
+        days_datetime.append(d)
+        d = d + datetime.timedelta(1)
+
+    binary_pdf = io.BytesIO()
+    with PdfPages(binary_pdf) as pdf:
+        a4 = (29.7/2.54, 21./2.54)
+        title = 'Seepark Wassertemperatur {} {}'.format(MONTH_DE[begin.month-1], begin.year)
+        report_times = [datetime.time(10), datetime.time(15)]
+
+        # graphic
+        plt.figure(figsize=a4)
+        plt.plot(x, y)
+        plt.xticks(days_datetime, [''] * len(days_datetime))
+        plt.ylabel('Temperatur in °C')
+        plt.axis(xmin=begin, xmax=end)
+        plt.grid()
+        plt.title(title)
+
+        # table
+        columns = []
+        for d in days_datetime:
+            columns.append('{}.'.format(d.day))
+        rows = []
+        for t in report_times:
+            rows.append('Wasser {:02d}:{:02d} °C'.format(t.hour, t.minute))
+        cells = []
+        for t in report_times:
+            columns.append('{}.'.format(d.day))
+            row_cells = []
+            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):
+                    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')
+        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
+        pdf.savefig()
+
+        pdf_info = pdf.infodict()
+        pdf_info['Title'] = title
+        pdf_info['Author'] = 'Chrisu Jähnl'
+        pdf_info['Subject'] = 'Wassertemperatur'
+        pdf_info['Keywords'] = 'Seepark Wassertemperatur'
+        pdf_info['CreationDate'] = datetime.datetime.now()
+        pdf_info['ModDate'] = datetime.datetime.today()
+
+    response = make_response(binary_pdf.getvalue())
+    response.headers['Content-Type'] = 'application/pdf'
+    response.headers['Content-Disposition'] = 'attachment; filename=seepark_{:04d}-{:02d}.pdf'.format(year, month)
+    return response
+
+

 @app.route("/")
 def index():
     airvalue, airtime     = currentairtemperature(cityid)
 @app.route("/")
 def index():
     airvalue, airtime     = currentairtemperature(cityid)