summaryrefslogtreecommitdiff
path: root/app.py
blob: bcc134647d116f59cd5a0dd24b86115e5caccdd3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from flask import (
    Flask,
    render_template,
    request
)
import pymongo


class dbProxy(object):
    def __init__(self, mongouri):
        self.client = pymongo.MongoClient(mongouri)
        self.db = self.client['jokevotedb']

    def getJokes(self):
        jokes = []
        for joke in self.db.jokes.find():
            jokes.append(joke)
        return jokes

    def addJoke(self, text):
        self.db.jokes.insert_one({'text': text, 'votes': 0})


app = Flask(__name__)
db = dbProxy(
    'mongodb://jokevote:jokevote@127.0.0.1:27017/jokevotedb')


@app.route('/')
def root():
    jokes = db.getJokes()
    return render_template('index.html', jokes=jokes)


@app.route('/submit', methods=['POST'])
def submit():
    text = request.form['text']
    db.addJoke(text)  # TODO secure this
    return root()


if __name__ == '__main__':
    app.run(debug=True)