app.py

import datetime
import os
import psycopg2

from flask import Flask, render_template

app = Flask(__name__)
app.secret_key = os.environ['APP_SECRET_KEY']

@app.route("/", methods=('GET', 'POST'))
def index():
    # Connect to database
    conn = psycopg2.connect(host='db', database=os.environ['POSTGRES_DB'], user=os.environ['POSTGRES_USER'], password=os.environ['POSTGRES_PASSWORD'])
    cur = conn.cursor()

    # Get number of all GET requests
    sql_all = """SELECT COUNT(*) FROM weblogs;"""
    cur.execute(sql_all)
    all = cur.fetchone()[0]

    # Get number of all local GET requests
    sql_local_all = """SELECT COUNT(*) FROM weblogs WHERE source = \'local\';"""
    cur.execute(sql_local_all)
    local_all = cur.fetchone()[0]

    # Get number of all remote GET requests
    sql_remote_all = """SELECT COUNT(*) FROM weblogs WHERE source = \'remote\';"""
    cur.execute(sql_remote_all)
    remote_all = cur.fetchone()[0]

    # Get number of all succesful requests
    sql_success = """SELECT COUNT(*) FROM weblogs WHERE status LIKE \'2__\';"""
    cur.execute(sql_success)
    success = cur.fetchone()[0]

    # Get number of all local requests
    sql_local = """SELECT COUNT(*) FROM weblogs WHERE status LIKE \'2__\' AND source = \'local\';"""
    cur.execute(sql_local)
    local = cur.fetchone()[0]

    # Get number of all remote requests
    sql_remote = """SELECT COUNT(*) FROM weblogs WHERE status LIKE \'2__\' AND source = \'remote\';"""
    cur.execute(sql_remote)
    remote = cur.fetchone()[0]

    # Determine rate if there was at least one request
    rate = "No entries yet!"
    localrate = "No entries yet!"
    remoterate = "No entries yet!"
    if all != 0:
        rate = str(success / all)
        localrate = str(local / local_all)
        remoterate = str(remote / remote_all)

    return render_template('index.html', rate = rate, localrate = localrate, remoterate = remoterate)

if __name__ == '__main__':
    app.run(host='0.0.0.0')