startup-systems · botingli · Oct 25, 2016 · Oct 25, 2016 · Oct 25, 2016 · Oct 25, 2016
diff --git a/squawker/schema.sql b/squawker/schema.sql
@@ -1,3 +1,5 @@
--- TODO change this
-DROP TABLE IF EXISTS mytable;
-CREATE TABLE mytable (id integer);
+DROP TABLE IF EXISTS squawks;
+CREATE TABLE squawks (
+  id INTEGER PRIMARY KEY AUTOINCREMENT,
+  squawk VARCHAR(140),
+  time_stamp DATETIME);
diff --git a/squawker/server.py b/squawker/server.py
@@ -1,6 +1,7 @@
-from flask import Flask, g
+from flask import Flask, request, g, abort, render_template
 import sqlite3
-
+import os
+import datetime
 
 # -- leave these lines intact --
 app = Flask(__name__)
@@ -37,11 +38,22 @@ def close_connection(exception):
 # ------------------------------
 
 
-@app.route('/')
+@app.route('/', methods=['POST', 'GET'])
 def root():
     conn = get_db()
-    # TODO change this
-    return "Hello World!"
+    cursor = conn.cursor()
+    if request.method == "POST":
+        msg = request.form["content"]
+        if len(msg) <= 140:
+            query = "INSERT INTO squawks (squawk, time_stamp) VALUES (?, ?)"
+            time = datetime.datetime.now()
+            cursor.execute(query, (msg, time))
+            conn.commit()
+        else:
+            abort(400)
+    cursor.execute("SELECT squawk FROM squawks ORDER BY time_stamp DESC")
+    res = cursor.fetchall()
+    return render_template("index.html", squawks=res)
 
 
 if __name__ == '__main__':

diff --git a/squawker/templates/index.html b/squawker/templates/index.html
@@ -0,0 +1,18 @@
+<!DOCTYPE HTML>
+<html>
+    <head>
+        <title>Squawker</title>
+    </head>
+
+    <body>
+        <h1>Enter Squawkers Here</h1>
+        <form method="post">
+            <textarea name="content" maxlength="140" placeholder="Say something..."></textarea>
+            <input type="submit">
+            <h2>Your squawks:</h2>
+            {% for s in squawks %}
+            <p>{{s[0]}}</p>
+            {% endfor %}
+        </form>
+    </body>
+</html>
diff --git a/tests/test_squawker.py b/tests/test_squawker.py
@@ -7,7 +7,6 @@
 import string
 import tempfile
 import time
-from . import flaskclient_fix
 
 
 URL = '/'