diff --git a/squawker/schema.sql b/squawker/schema.sql index 5e67ffb..968329d 100644 --- a/squawker/schema.sql +++ b/squawker/schema.sql @@ -1,3 +1,2 @@ --- TODO change this DROP TABLE IF EXISTS mytable; -CREATE TABLE mytable (id integer); +CREATE TABLE mytable (id INTEGER PRIMARY KEY AUTOINCREMENT, squawk VARCHAR(140) DEFAULT NULL); diff --git a/squawker/server.py b/squawker/server.py index 6ff24ba..2665a8e 100644 --- a/squawker/server.py +++ b/squawker/server.py @@ -1,4 +1,4 @@ -from flask import Flask, g +from flask import Flask, g, request, render_template import sqlite3 @@ -37,11 +37,20 @@ def close_connection(exception): # ------------------------------ -@app.route('/') +@app.route('/', methods=["POST", "GET"]) def root(): conn = get_db() - # TODO change this - return "Hello World!" + if request.method == "POST": + post = request.form["squawk_post"] + if len(post) > 140: + return "Message is too long", 400 + else: + conn.execute("INSERT INTO mytable (squawk) VALUES(?)", [post]) + conn.commit() + c = conn.cursor() + c.execute("SELECT * FROM mytable ORDER BY id DESC") + squawks = c.fetchall() + return render_template("index.html", squawks=squawks) if __name__ == '__main__': diff --git a/squawker/templates/index.html b/squawker/templates/index.html new file mode 100644 index 0000000..cace0bc --- /dev/null +++ b/squawker/templates/index.html @@ -0,0 +1,25 @@ + + + + My Squawker + + +
+
+ + +
+
+ +
+
+
+ +
+ + +