/
server.py
47 lines (36 loc) · 1.11 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import peeweedbevolve # new; must be imported before models
from flask import Flask, render_template, request,redirect
from models import db, Store, Warehouse
app = Flask(__name__)
@app.before_request
def before_request():
db.connect()
@app.after_request
def after_request(response):
db.close()
return response
@app.cli.command() # new
def migrate(): # new
db.evolve(ignore_tables={'base_model'}) # new
@app.route("/")
def index():
return render_template('index.html')
@app.route("/store")
def store():
return render_template('store.html')
@app.route("/store", methods=["POST"])
def store_add():
Store.create(name=request.form.get("store_name"))
return redirect("/store")
@app.route("/warehouse")
def warehouse():
store = Store.select()
return render_template('warehouse.html',store=store)
@app.route("/warehouse", methods=["POST"])
def warehouse_add():
print(request.form["store_id"])
store = Store.get_by_id(request.form["store_id"])
Warehouse.create(location=request.form.get("warehouse_location"),store=store)
return redirect("/warehouse")
if __name__ == '__main__':
app.run()