114 lines
3.7 KiB
Python
114 lines
3.7 KiB
Python
from numpy import arange, hstack
|
|
|
|
from flask import session, render_template, request, redirect, current_app
|
|
|
|
from . import main
|
|
from .forms import AKIForm
|
|
from .utils import getTemps, avg_air_temp, ann_air_indices, avg_air_indices, des_air_indices
|
|
from app.models import Community, Dataset, Temperature
|
|
|
|
|
|
@main.route('/', methods=['GET'])
|
|
def index():
|
|
form = AKIForm()
|
|
session['community_data'] = None
|
|
session['avg_temp'] = None
|
|
session['avg_indices'] = None
|
|
session['des_indices'] = None
|
|
|
|
if 'community' in session:
|
|
community_id = session['community']
|
|
if all(key in session for key in ('minyear', 'maxyear', 'datasets')):
|
|
community = Community.query.get_or_404(community_id)
|
|
|
|
# TODO: clean this up
|
|
session['community_data'] = dict()
|
|
session['community_data']['id'] = community_id
|
|
session['community_data']['name'] = community.name
|
|
session['community_data']['latitude'] = round(community.latitude, 5)
|
|
session['community_data']['longitude'] = round(community.longitude, 5)
|
|
|
|
session['ds_name'] = Dataset.query. \
|
|
with_entities(Dataset.modelname, Dataset.scenario). \
|
|
filter_by(id=session['datasets']).all()
|
|
temps_arr = getTemps(session['datasets'], community_id, session['minyear'], session['maxyear'])
|
|
|
|
session['avg_temp'] = avg_air_temp(temps_arr)
|
|
indices = ann_air_indices(temps_arr)
|
|
session['avg_indices'] = avg_air_indices(indices)
|
|
session['des_indices'] = des_air_indices(indices)
|
|
|
|
return render_template("index.html", form=form)
|
|
|
|
|
|
@main.route('/', methods=['POST'])
|
|
def index_submit():
|
|
form = AKIForm()
|
|
if form.validate():
|
|
session['community'] = request.form['community']
|
|
session['minyear'] = request.form['minyear']
|
|
session['maxyear'] = request.form['maxyear']
|
|
if session['minyear'] > session['maxyear']:
|
|
session['maxyear'] = session['minyear']
|
|
|
|
session['datasets'] = request.form['model']
|
|
return redirect('/')
|
|
else:
|
|
return render_template("index.html", form=form)
|
|
|
|
|
|
@main.route('/datatypes')
|
|
def datatypes():
|
|
return render_template("datatypes.html")
|
|
|
|
|
|
@main.route('/reset')
|
|
def reset():
|
|
session.clear()
|
|
return redirect('/')
|
|
|
|
|
|
@main.route('/details')
|
|
def details():
|
|
datasets = request.args.get('datasets', '')
|
|
community_id = request.args.get('community_id', '')
|
|
minyear = request.args.get('minyear', '')
|
|
maxyear = request.args.get('maxyear', '')
|
|
temps = getTemps(datasets, community_id, minyear, maxyear)
|
|
years = arange(int(minyear), int(maxyear)+1).reshape(int(maxyear)-int(minyear) + 1, 1)
|
|
temps = hstack((years, temps))
|
|
return render_template("details.html",
|
|
lat=request.args.get('lat', ''),
|
|
lon=request.args.get('lon', ''),
|
|
community_name=request.args.get('name', ''),
|
|
temps=temps)
|
|
|
|
|
|
@main.route('/save')
|
|
def save():
|
|
if 'save' in session:
|
|
i = len(session['save'])
|
|
save = session['save']
|
|
else:
|
|
save = dict()
|
|
i = 0
|
|
|
|
save[i] = dict()
|
|
save[i]['datasets'] = session['datasets']
|
|
save[i]['ds_name'] = session['ds_name']
|
|
save[i]['community_data'] = session['community_data']
|
|
save[i]['minyear'] = session['minyear']
|
|
save[i]['maxyear'] = session['maxyear']
|
|
save[i]['avg_temp'] = session['avg_temp']
|
|
save[i]['avg_indices'] = session['avg_indices']
|
|
save[i]['des_indices'] = session['des_indices']
|
|
session.clear()
|
|
session['save'] = save
|
|
return redirect('/')
|
|
|
|
|
|
@main.route('/delete')
|
|
def delete():
|
|
record = request.args.get('record', '')
|
|
session['save'].pop(record)
|
|
return redirect('/')
|