import os
import sys
from collections import defaultdict
-from flask import Flask, render_template, jsonify, request
+from flask import Flask, render_template, jsonify, request, abort, Response
import flask.json
-from flask_sqlalchemy import SQLAlchemy
+from flask_sqlalchemy import SQLAlchemy, inspect
app_path = os.path.dirname(os.path.realpath(__file__))
-lib_path = os.path.join(cur_path, '..')
+lib_path = os.path.join(app_path, '..')
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)
__tablename__ = 'sensors'
-def select_sensordata(initial_where):
- query = Sensors.query.filter(initial_where)
- begin = request.args.get('begin', None, parse_datetime)
- end = request.args.get('end', None, parse_datetime)
+class OpenWeatherMap(db.Model):
+ __tablename__ = 'openweathermap'
+
+
+def select_sensordata(sensor_id, sensor_type, begin, end, mode):
+ query = Sensors.query
+ if sensor_id is not None:
+ query = query.filter(Sensors.sensor_id == sensor_id)
+ if sensor_type is not None:
+ query = query.filter(Sensors.value_type == sensor_type)
if begin is not None:
query = query.filter(Sensors.timestamp >= begin)
if end is not None:
query = query.filter(Sensors.timestamp <= end)
- result = query.all()
+ if mode == 'consolidated' and begin is not None and end is not None:
+ # copied from munin/master/_bin/munin-cgi-graph.in
+ # interval in seconds for data points
+ resolutions = dict(
+ day = 300,
+ week = 1800,
+ month = 7200,
+ year = 86400,
+ )
+ duration = (end - begin).total_seconds()
+ day = 60 * 60 * 24
+ if duration < day:
+ resolution = resolutions['day']
+ elif duration < 7 * day:
+ resolution = resolutions['week']
+ elif duration < 31 * day:
+ resolution = resolutions['month']
+ else:
+ resolution = resolutions['year']
+ # TODO: filter out samples from 'result'
+ # something like
+ # select to_seconds(datetime) DIV (60*60*24) as interval_id, min(datetime), max(datetime), min(temp), avg(temp), max(temp), count(temp) from openweathermap group by interval_id order by interval_id;
+ return query.all()
- mode = request.args.get('mode', 'full')
- if mode == 'consolidated':
- if begin is None or end is None:
- pass
+
+def select_openweatherdata(cityid, begin, end, mode):
+ query = OpenWeatherMap.query.filter(OpenWeatherMap.cityid == cityid)
+ if begin is not None:
+ query = query.filter(OpenWeatherMap.datetime >= begin)
+ if end is not None:
+ query = query.filter(OpenWeatherMap.datetime <= end)
+ if mode == 'consolidated' and begin is not None and end is not None:
+ # copied from munin/master/_bin/munin-cgi-graph.in
+ # interval in seconds for data points
+ resolutions = dict(
+ day = 300,
+ week = 1800,
+ month = 7200,
+ year = 86400,
+ )
+ duration = (end - begin).total_seconds()
+ day = 60 * 60 * 24
+ if duration < day:
+ resolution = resolutions['day']
+ elif duration < 7 * day:
+ resolution = resolutions['week']
+ elif duration < 31 * day:
+ resolution = resolutions['month']
else:
- # copied from munin/master/_bin/munin-cgi-graph.in
- resolutions = dict(
- day = 300,
- week = 1800,
- month = 7200,
- year = 86400,
- )
- duration = (end - begin).total_seconds()
- day = 60 * 60 * 24
- if duration < day:
- resolution = resolutions['day']
- elif duration < 7 * day:
- resolution = resolutions['week']
- elif duration < 31 * day:
- resolution = resolutions['month']
- else:
- resolution = resolutions['year']
- # TODO: filter out samples from 'result'
- # like loop over results and skip if timestamp(n+1)-timestamp(n)<resolution
+ resolution = resolutions['year']
+ # TODO: filter out samples from 'result'
+ # something like
+ # select to_seconds(datetime) DIV (60*60*24) as interval_id, min(datetime), max(datetime), min(temp), avg(temp), max(temp), count(temp) from openweathermap group by interval_id order by interval_id;
+ return query.all()
+
+
+def convert_to_c3(result, id, field_x, field_y):
+ c3result = defaultdict(list)
+ for row in result:
+ c3result[getattr(row, id)].append(getattr(row, field_y))
+ dt = getattr(row, field_x).strftime('%Y-%m-%d %H:%M:%S')
+ c3result[str(getattr(row, id)) + '_x'].append(dt)
+ return c3result
+
+
+def request_arg(key, type, default=None):
+ """Returns the key from the request if available, otherwise the default value.
+ In case type is provided and the key is present, the value is converted by calling type.
+ In other words: Reimplement request.args.get but don't return default value if
+ type raises a ValueError."""
+ if key in request.args:
+ try:
+ return type(request.args[key])
+ except ValueError as e:
+ abort(Response(str(e), 400))
+ else:
+ return default
+
+
+def sensordata(sensor_id=None, sensor_type=None):
+ begin = request_arg('begin', parse_datetime)
+ end = request_arg('end', parse_datetime)
+ mode = request.args.get('mode', 'full')
+ format = request.args.get('format', 'default')
+
+ result = select_sensordata(sensor_id, sensor_type, begin, end, mode)
+
+ if format == 'c3':
+ return convert_to_c3(result, 'sensor_id', 'timestamp', 'value')
+ return result
+
+def openweathermapdata(cityid):
+ begin = request_arg('begin', parse_datetime)
+ end = request_arg('end', parse_datetime)
+ mode = request.args.get('mode', 'full')
format = request.args.get('format', 'default')
+
+ result = select_openweatherdata(cityid, begin, end, mode)
+
if format == 'c3':
- c3result = defaultdict(list)
- for row in result:
- c3result[row.sensor_id].append(row.value)
- dt = row.timestamp.strftime('%Y-%m-%d %H:%M:%S')
- c3result[row.sensor_id + '_x'].append(dt)
- result = c3result
+ return convert_to_c3(result, 'cityid', 'datetime', 'temp')
return result
-def currentairtemperature(apikey, cityid):
- """Retruns the tuple temperature, datetime (as float, datetime) in case of success, otherwise None, None."""
- try:
- weatherdata = openweathermap_json(apikey, cityid)
- return weatherdata['main']['temp'], datetime.datetime.fromtimestamp(weatherdata['dt'])
- except OpenWeatherMapError:
- return None, None
+def currentairtemperature(cityid):
+ result = OpenWeatherMap.query.filter_by(cityid=cityid).order_by(OpenWeatherMap.datetime.desc()).first()
+ return result.temp, result.datetime
def currentwatertemperature(sensorid):
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 = select_sensordata(Sensors.sensor_id == sensor_id)
- return jsonify([row._asdict() for row in result])
+ result = sensordata(sensor_id=sensor_id)
+ return jsonify(result)
@app.route('/api/<version>/sensor/type/<sensor_type>')
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 = select_sensordata(Sensors.value_type == sensor_type)
+ result = sensordata(sensor_type=sensor_type)
+ return jsonify(result)
+
+
+@app.route('/api/<version>/openweathermap/cities')
+def openweathermap_cities(version):
+ """List all city IDs found in the database"""
+ result = db.session.query(OpenWeatherMap.cityid).distinct().all()
+ return jsonify(result)
+
+
+@app.route('/api/<version>/openweathermap/city/<cityid>')
+def openweathermap_city(version, cityid):
+ """List all data found for a city"""
+ result = openweathermapdata(cityid=cityid)
return jsonify(result)
@app.route("/")
def index():
- airvalue, airtime = currentairtemperature(apikey, cityid)
+ airvalue, airtime = currentairtemperature(cityid)
watervalue, watertime = currentwatertemperature(mainsensor)
return render_template(
ToastFreeware / chrisu / seepark.git / blobdiff