app.py

import os
from flask import Flask, redirect, url_for, request, render_template
from pymongo import MongoClient

from tools.dummy_db import create_dummies

app = Flask(__name__)

client = MongoClient(os.environ['MONGODB_HOST'],27017)
db = client.profiles

#TODO: automate this at build stage
if "jacs" not in db.list_collection_names():
    import json
    with open(os.path.join("tools", "jacs.json"), "r") as f:
        db.jacs.insert_one(json.load(f))

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


@app.route('/register', methods=['GET', 'POST'])
def register():
    """ route that handles registration form submission
        by inserting new document in database
        #TODO: add security!
    """
    if request.method == 'POST':
        _jacs = db.jacs.find_one()
        profile_doc = {
            'first_name': request.form['first_name'],
            'surname': request.form['surname'],
            'email': request.form['email'],
            'orcid': request.form['orcid'],
            'github': request.form['github'],
            'institution_url': request.form['institution_url'],
            'disciplines': request.form.getlist('disciplines'),
            'languages': request.form.getlist('languages')
        }
        db.profiles.insert_one(profile_doc)
        return redirect(url_for('confirm'))
    else:
        _jacs = db.jacs.find_one()
        areas = _jacs['area']
        discs = _jacs['discipline']
        return render_template('register.html', disciplines=discs, areas=areas, page='register')

@app.route("/dummy", methods=["GET", "POST"])
def register_dummies():
    """ Insert dummy records in the database
    """
    if request.method == "POST":
        number = request.form["number"]
        prof = create_dummies(int(number))
        db.profiles.insert_many(prof)
        return redirect(url_for("confirm"))
    else:
        return render_template("dummy.html")

@app.route('/confirm')
def confirm():
    """ route that outputs confirmation of registration
        #TODO: mechanism to determine if registration successful!
    """
    return render_template('confirm.html', page='confirmation')

@app.route('/search')
def search():
    """ A route to the search page.
    """
    _jacs = db.jacs.find_one()
    areas = _jacs['area']
    discs = _jacs['discipline']
    return render_template('search.html', areas=areas, disciplines=discs, page='search')

@app.route('/results', methods=['GET', 'POST'])
def results():
    _profiles = db.profiles.find()
    profiles = [profile for profile in _profiles]
    for profile in profiles:
        try:
            profile['disciplines'] = disc_names_from_codes(profile['disciplines'])
        except KeyError:
            profile['disciplines'] = ""
    return render_template('results.html', profiles=profiles)

def disc_names_from_codes(codes):
    """ gets the discipline names from a list of discipline JACS codes
        #TODO: this belongs somewhere else
    """
    _jacs = db.jacs.find_one()
    all_discs = _jacs['discipline']
    discs = [all_discs[code] for code in codes]
    return discs

if __name__ == "__main__":
    app.run(host='0.0.0.0', debug=True)