diff --git a/squawker/server.py b/squawker/server.py index 6ff24ba..928c7e9 100644 --- a/squawker/server.py +++ b/squawker/server.py @@ -1,6 +1,12 @@ -from flask import Flask, g +from flask import Flask +from flask import g +from flask import render_template +from flask import request import sqlite3 - +import time +import datetime +import random +import webbrowser # -- leave these lines intact -- app = Flask(__name__) @@ -37,11 +43,37 @@ def close_connection(exception): # ------------------------------ -@app.route('/') +@app.route('/', methods=['GET', 'POST']) def root(): conn = get_db() - # TODO change this - return "Hello World!" + c = conn.cursor() + c.execute('CREATE TABLE IF NOT EXISTS allSquawksIEverMade(datestamp TEXT, squawk TEXT)') + s = [] + c.execute('SELECT * FROM allSquawksIEverMade') + for row in c.fetchall(): + s.append(row[1]) + t = [] + for i in reversed(s): + t.append(i) + return render_template('Homepage.html', s=t) + + +@app.route('/send', methods=['GET', 'POST']) +def send(): + conn = get_db() + c = conn.cursor() + if request.method == 'POST': + data = str(request.form['squawk']) + if len(data) > 140: + s = [] + s.append('Error Code 400: Input correct form details') + return render_template('Homepage.html', s=s) + unix = time.time() + timestamp = str(datetime.datetime.fromtimestamp(unix).strftime('%Y-%m-%d %H:%M:%S')) + c.execute("INSERT INTO allSquawksIEverMade (datestamp, squawk) VALUES (?, ?)", (timestamp, data)) + conn.commit() + webbrowser.open('http://localhost:5000/') + return 'all Ok' if __name__ == '__main__': diff --git a/squawker/templates/Homepage.html b/squawker/templates/Homepage.html new file mode 100644 index 0000000..00f8420 --- /dev/null +++ b/squawker/templates/Homepage.html @@ -0,0 +1,35 @@ + + +
+