app.py

#!/usr/bin/env python
import argparse
import datetime
import json
import os
import random

from flask import Flask, make_response, render_template
from pymongo import MongoClient

import models
import settings
import utils

app = Flask(__name__)

def tally():
    models.Session.tally()
    models.User.tally()
    with open('/tmp/log.log', 'w') as writefile:
        writefile.write("%s" % datetime.datetime.now())

@app.route('/')
def index():
    # return open('static/2019.html').read()
    if settings.VOTING:

        sessions = utils.connect('session').find({})
        payload = []

        for s in sessions:
            s = dict(s)
            try:
                user = utils.connect('user').find_one({"_id": s['user']})
                s['user_obj'] = dict(utils.connect('user').find_one({"_id": s['user']}))
                s['username'] = user['name']
                payload.append(s)
            except:
                pass

        random.shuffle(payload)

        return render_template('session_list.html', sessions=payload, VOTING=settings.VOTING, ACTIVE=settings.ACTIVE)

    else:
        return render_template('create_session.html', VOTING=settings.VOTING, ACTIVE=settings.ACTIVE)

@app.route('/<path:path>')
def static_proxy(path):
  return app.send_static_file(path)

@app.route('/api/dashboard/<secret_key>')
def dashboard(secret_key, methods=['GET']):
    if secret_key == os.environ.get('API_KEY', None):
        sessions = utils.connect('session').find({})
        payload = []

        for s in sessions:
            try:
                s = dict(s)
                s['user_obj'] = dict(utils.connect('user').find_one({"_id": s['user']}))
                payload.append(s)
            except:
                pass

        payload = sorted(payload, key=lambda x: x['votes'], reverse=True)[:100]

        for s in payload:
            s['all_votes'] = []
            votes = utils.connect('vote').find({"session": s["_id"]})
            for v in votes:
                vote = dict(v)
                user = dict(utils.connect('user').find_one({"_id": vote['user']}))
                for x in ['login_hash', 'updated', 'password']:
                    del user[x]
                vote['user'] = user
                vote['vote_time'] = datetime.datetime.fromtimestamp(vote['created'])
                vote['user_time'] = datetime.datetime.fromtimestamp(user['created'])
                s['all_votes'].append(vote)

        return render_template('dashboard.html', sessions=payload, VOTING=True)
    return json.dumps({})

@app.route('/api/user/action/')
def user_action(methods=['GET']):
    from flask import request
    email = request.args.get('email', None)
    password = request.args.get('password', None)

    not_found = json.dumps({"success": False, "text": "Username or password is incorrect."})

    user = utils.connect('user').find_one({ "email": email })

    if not user:
        name = request.args.get('name', None)
        fingerprint = request.args.get('fingerprint', None)
        if not name or not fingerprint:
            return not_found

        u = models.User(email=email, name=name, password=password, fingerprint=fingerprint)
        u.save()
        return json.dumps({"success": True, "_id": u._id, "name": u.name, "votes": "|".join(u.sessions_voted_for), "action": "register"})

    else:
        user_dict = dict(user)
        u = models.User(**user_dict)

        if u.auth_user(password):
            return json.dumps({"success": True, "_id": u._id, "name": u.name, "votes": "|".join(u.sessions_voted_for), "action": "login"})

    return not_found

@app.route('/api/session/action/')
def session_action(methods=['GET']):
    from flask import request
    _id = request.args.get('user', None)
    user = None

    session_dict = {}
    session_dict['title'] = request.args.get('title', None)
    session_dict['description'] = request.args.get('description', None)
    session_dict['votes'] = 0
    session_dict['accepted'] = False

    error = json.dumps({"success": False, "text": "Please send a valid user ID and a session title and description."})

    if not _id:
        return json.dumps(error)

    if _id:
        user = dict(utils.connect('user').find_one({"_id": _id}))
        session_dict['user'] = _id
        s = models.Session(**session_dict).save()

        tally()

        return json.dumps({"success": True, "action": "create", "session": s['_id']})

@app.route('/api/vote/action/')
def vote_action(methods=['GET']):
    from flask import request
    session = request.args.get('session')
    user = request.args.get('user')

    error = json.dumps({"success": False, "text": "Please send a session ID and a user ID."})

    if not session or not user:
        return error

    u = utils.connect('user').find_one({"_id": user})
    if not u:
        return error

    s = utils.connect('session').find_one({"_id": session})
    if not s:
        return error

    votes = [vote for vote in utils.connect('vote').find({"user": user, "session": session})]

    # Create a new vote.
    if len(votes) == 0:
        models.Vote(user=u['_id'], session=s['_id']).save()
        sesh = models.Session(s)
        sesh.update_records()
        # tally()
        return json.dumps({"success": True, "action": "create vote"})

    # Delete existing votes.
    if len(votes) > 0:
        utils.connect('vote').remove({"user": user, "session": session})
        sesh = models.Session(s)
        sesh.update_records()
        # tally()
        return json.dumps({"success": True, "action": "delete vote"})

    return error

@app.route('/api/vote/')
def api_vote(methods=['GET']):
    from flask import request
    _id = request.args.get('_id', None)
    if not _id:
        return json.dumps(list(utils.connect('vote').find({})))
    vote = dict(utils.connect('vote').find_one({"_id": _id}))
    return json.dumps(vote)

@app.route('/api/user/')
def api_user(methods=['GET']):
    from flask import request
    _id = request.args.get('_id', None)
    if not _id:
        return json.dumps(list(utils.connect('user').find({})))
    user = dict(utils.connect('user').find_one({"_id": _id}))
    for x in ['login_hash', 'updated', 'created', 'password', 'fingerprint']:
        del user[x]
    return json.dumps(user)

# @app.route('/api/session/')
# def api_session(methods=['GET']):
#     from flask import request
#     _id = request.args.get('_id', None)
#     if not _id:
#         return json.dumps(list(utils.connect('session').find({})))

#     session = dict(utils.connect('session').find_one({"_id": _id}))
#     return json.dumps(session)

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-p', '--port')
    args = parser.parse_args()
    server_port = 8005

    if args.port:
        server_port = int(args.port)

    app.run(host='0.0.0.0', port=server_port, debug=True)