From 59a6e45cf2d2b3d012f9d031ec7acb495473ca00 Mon Sep 17 00:00:00 2001 From: Benjamyn Love Date: Sat, 11 Aug 2018 23:38:06 +1000 Subject: [PATCH] Added DB stuffs --- .gitignore | 1 + main.py | 63 ++++++++++++++++++++++++++++++++++++++++++++ requirements.txt | 1 + templates/index.html | 24 +++++++++++++++++ 4 files changed, 89 insertions(+) create mode 100755 main.py create mode 100644 requirements.txt create mode 100644 templates/index.html diff --git a/.gitignore b/.gitignore index bdaab25..8dbe2ec 100644 --- a/.gitignore +++ b/.gitignore @@ -1 +1,2 @@ env/ +list.db diff --git a/main.py b/main.py new file mode 100755 index 0000000..2026c1a --- /dev/null +++ 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() diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..7e10602 --- /dev/null +++ b/requirements.txt @@ -0,0 +1 @@ +flask diff --git a/templates/index.html b/templates/index.html new file mode 100644 index 0000000..4807f39 --- /dev/null +++ b/templates/index.html @@ -0,0 +1,24 @@ + + +

Shopping list

+ + {% for res in data %} + {% for field in res %} + {{ field }} + {% endfor %} +
+ {% endfor %} +
+ +
+ +
+ +
+ +
+ +
+ + + \ No newline at end of file