Added DB stuffs

2018-08-11 23:38:06 +10:00 · 2018-08-11 23:38:06 +10:00 · 59a6e45cf2
commit 59a6e45cf2
parent 0671803dfe
4 changed files with 89 additions and 0 deletions
--- a/.gitignore
+++ b/.gitignore
@ -1 +1,2 @@
 env/
+list.db
--- a/main.py
+++ b/main.py
@ -0,0 +1,63 @@
+#!/usr/bin/env python
+from flask import Flask, render_template, request
+from pprint import pprint
+import sqlite3
+import os.path
+
+
+def createDB():
+    if os.path.isfile('list.db'):
+        print("DB Exists")
+    else:
+        conn = sqlite3.connect('list.db')
+        c = conn.cursor()
+        c.execute('''CREATE TABLE LIST (
+                  requester text, item text, quatity real, gotten real)''')
+        conn.close()
+
+
+def insTestData(c, conn, data):
+    c.execute("INSERT INTO list VALUES ('Tim', 'Memes', 100, 0)")
+    conn.commit()
+
+
+def getData():
+    with sqlite3.connect('list.db') as conn:
+        c = conn.cursor()
+        res = c.execute("SELECT * FROM LIST")
+        data = res.fetchall()
+        pprint(data)
+        return data
+
+
+def updateDB(req):
+    with sqlite3.connect('list.db') as conn:
+        c = conn.cursor()
+        print(req)
+        c.executemany('INSERT INTO LIST VALUES (?,?,?,?)', req)
+        conn.commit()
+createDB()
+
+app = Flask(__name__)
+app.config['DEBUG'] = True
+
+
+@app.route('/', methods=['POST', 'GET'])
+def index():
+    if request.method == 'POST':
+        req = [(request.form['requester'], request.form['item'],
+               request.form['quantity'], request.form['gotten'])]
+        updateDB(req)
+        data = getData()
+        return render_template('index.html', data=data)
+    else:
+        data = getData()
+        return render_template('index.html', data=data)
+
+#insTestData(2)
+
+if __name__ == '__main__':
+
+    #ret = c.execute("SELECT * from LIST WHERE requester = 'Tim'")
+    #print(ret.fetchall())
+    app.run()
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1 @@
+flask
--- a/templates/index.html
+++ b/templates/index.html
@ -0,0 +1,24 @@
+<html>
+<body style="background: #123456">
+    <h1>Shopping list </h1>
+
+    {% for res in data %}
+        {% for field in res %}
+            {{ field }}
+        {% endfor %}
+        <br/>
+    {% endfor %}
+    <form action="/" method="post">
+        <label>Who Dat?</label>
+        <input type="text" name='requester'/><br/>
+        <label>Watchu Want?</label>
+        <input type="text" name='item'/><br/>
+        <label>How Many?</label>
+        <input type="text" name='quantity'/><br/>
+        <label>Have we gotten it?</label>
+        <input type="text" name='gotten'/><br/>
+        <input type="submit"/>
+    </form>
+
+</body>
+</html>