aqi_monitor/web/views.py

""" main entry page to handle all the routes """

import os
from datetime import datetime, timedelta

from flask import Flask, render_template, request, app
from flask import url_for  # pylint: disable=unused-import
from flask_httpauth import HTTPBasicAuth
from apscheduler.schedulers.background import BackgroundScheduler


from src.helper import get_config, get_table
from src.db import get_current, insert_data
from src.graph_current import main as current_graph
from src.graph_nightly import main as nightly_graph
from src.graph_monthly import main as monthly_graph

import matplotlib
matplotlib.use('Agg')

# start up
app = Flask(__name__)

# maintenance mode
is_maintenance_mode = bool(os.environ.get("MAINTENANCE"))
@app.before_request
def check_for_maintenance():
    print(request.path)
    ALLOW = ["/static/css/style.css", "/static/favicon.ico", "/static/font/Rubik-Bold.ttf", "/static/font/Rubik-Regular.ttf"]
    if is_maintenance_mode and request.path not in ALLOW: 
        # return redirect(url_for('maintenance'))
        # Or alternatively, dont redirect 
        # return 'Sorry, off for maintenance!', 503
        return render_template('maintenance.html'), 503


# @app.route('/maintenance')
# def maintenance():
#     return 'Sorry, off for maintenance!', 503


CONFIG = get_config()
auth = HTTPBasicAuth()
aqi_user = CONFIG['aqi_monitor']
USER_DATA = {
    aqi_user['authUsername']: aqi_user['authPassword']
}

if not is_maintenance_mode:
    # initial export
    print('initial export')
    current_graph()
    nightly_graph()

    # start scheduler
    scheduler = BackgroundScheduler(timezone=os.environ.get("TZ", "UTC"))
    scheduler.add_job(
        current_graph, trigger="cron", minute='*/5', name='current_graph'
    )
    scheduler.add_job(
        nightly_graph, trigger="cron", day='*', hour='1', minute='1', name='night'
    )
    scheduler.add_job(
        monthly_graph, trigger="cron", day='*', hour='1', minute='2', name='month'
    )
    scheduler.add_job(
        monthly_graph, next_run_time=datetime.now() + timedelta(seconds=10)
    )
    scheduler.start()


@auth.verify_password
def verify(username, password):
    """ get password """
    if not (username and password):
        return False
    return USER_DATA.get(username) == password


@app.route("/")
def home():
    """ home page """
    return render_template('home.html')


@app.route("/about")
def about():
    """ about page """
    return render_template('about.html', title='About')


@app.route("/graphs")
def graphs():
    """ graphs page """
    table = get_table("static/dyn/year-table.json")
    return render_template('graphs.html', title='Graphs', table=table)


@app.route("/monthly")
def monthly():
    """ monthly statistics page """
    months = [i for i in os.listdir('static/dyn/monthly') if '.json' in i]
    months.sort(reverse=True)

    month_dicts = []
    for month in months:
        month_clean = os.path.splitext(month)[0]
        month_graph = os.path.join('static/dyn/monthly', month_clean + '.png')
        month_name = datetime.strptime(month_clean, "%Y-%m").strftime('%B %Y')
        month_json = os.path.join('static/dyn/monthly', month)
        table = get_table(month_json)
        month_dict = {
            'month_graph': month_graph,
            'month_name': month_name,
            'table': table
        }
        month_dicts.append(month_dict)

    return render_template('monthly.html', title='Monthly', months=month_dicts)


@app.route("/data/in", methods=['POST'])
@auth.login_required
def ingest():
    """ handle post request from monitor """
    post_data = request.json
    print(post_data)
    insert_data(post_data)
    return 'ingest'


@app.route("/data/out")
def data():
    """ return data from db """
    json_data = get_current()
    return json_data