enotcode / facemash

app/views.py

import random

There seems to be a cyclic import (app -> app.views).

from flask import render_template, url_for, redirect
from pymongo import MongoClient

from . import app

base = 'mongodb://localhost:21017'

client = MongoClient(base)
db = client["facemash"]


@app.route('/')
def index():
    f = db.data.find_one({"id": random.randint(1, db.data.count())})
    s = db.data.find_one({"id": random.randint(1, db.data.count())})
    if f == s:
        s = db.data.find_one({"id": random.randint(1, db.data.count())})
    return render_template('index.html', first=str(f["id"]), second=str(s["id"]))


@app.route('/vote/<v>')
def vote(v):
    vote = db.data.find_one({"id": int(v)})["vote"] + 1
    db.data.update({"id": int(v)}, {"$set": {"vote": vote}})
    return redirect(url_for('index'), 302)