web_whiteblocker_api.py

import logging
import os
import json
import whiteblocker_api
from common.DatabaseUtils import select_all
from common.ApiUtils import unblock, case_3 as block
from flask import Flask, flash, redirect, render_template, request, session, send_from_directory
from threading import Thread


# Credentials for web login
with open("config.json") as config_file:
    data = json.load(config_file)
    credentials = data['web_credentials']
    username = credentials['username']
    password = credentials['password']

# Create the application instance
app = Flask(__name__)
logging.basicConfig(filename="api.log", filemode="w", format="%(asctime)s %(name)s - %(levelname)s - %(message)s",
                    level=logging.INFO)


@app.route('/')
def root():
    return redirect('/home')


@app.route('/home')
def home():
    if not session.get('logged_in'):
        return render_template('login.html')
    else:
        data = select_all()
        ips = []
        tries = []
        location = []
        status = []
        last_view = []
        for i in data:
            ips.append(i[0])
            tries.append(i[1])
            location.append(i[2])
            status.append(i[3])
            last_view.append(i[7])

        bar_labels = ips
        bar_values = tries
        return render_template('home.html', max=17000, labels=bar_labels, values=bar_values)


@app.route('/login', methods=['GET', 'POST'])
def login():
    error = None
    if request.method == 'POST':
        if request.form['username'] != username or request.form['password'] != password:
            error = "Invalid Credentials. Please, try again."
            flash(error)
            return render_template('login.html')
        else:
            session['logged_in'] = True
            return redirect('/home')
    return render_template('login.html', error=error)


@app.route('/logout')
def logout():
    session.clear()
    info = "Logged Out Successfully"
    flash(info)
    return render_template('login.html', error=info)


@app.route('/favicon.ico')
def favicon():
    return send_from_directory(os.path.join(app.root_path, 'static'),
                               'favicon.ico', mimetype='static/favicon.ico')


@app.route('/list')
def list_whiteblocker():
    if not session.get('logged_in'):
        return render_template('login.html')
    else:
        rows = select_all()
        return render_template('list.html', rows=rows)


@app.route('/manual', methods=['GET', 'POST'])
def manual_whiteblocker():
    if not session.get('logged_in'):
        return render_template('login.html')
    else:
        if request.method == 'POST':
            action = request.form['action']
            ip = request.form['ip']
            if action == "Block":
                tier = request.form['tier']
                process = block(ip=ip, tier=tier)
                if process == "ok":
                    msg = "Correctly blocked IP {0}".format(ip)
                    flash(msg)
                    return render_template('manual.html', msg=msg)
                else:
                    msg = "Failed to block IP {0}".format(ip)
                    flash(msg)
                    return render_template('manual.html', msg=msg)
            elif action == "Unblock":
                process = unblock(ip=ip)
                if process == "ok":
                    msg = "Correctly unblocked IP {0}".format(ip)
                    flash(msg)
                    return render_template('manual.html', msg=msg)
                else:
                    msg = "Failed to unblocked IP {0}".format(ip)
                    flash(msg)
                    return render_template('manual.html', msg=msg)
        return render_template('manual.html')


@app.route('/help')
def help_whiteblocker():
    if not session.get('logged_in'):
        return render_template('login.html')
    else:
        return redirect("https://github.com/fernandocastrovilar/whiteblocker", code=302)


if __name__ == "__main__":
    try:
        app.secret_key = os.urandom(12)
        Thread(target=whiteblocker_api.main()).start()
        Thread(target=app.run(host="0.0.0.0", debug=True)).start()
    except KeyboardInterrupt:
        print("^C received, shutting down the web server")
        logging.error("^C received, shutting down the web server")
        exit("Exit")
    except Exception as e:
        print(e)
        logging.error(e)