-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
125 lines (107 loc) · 4.04 KB
/
app.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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
import os
from flask import Flask, render_template, request
from flask.ext.wtf import Form, TextField, BooleanField
from flask.ext.wtf import Required
import stripe
from auth import requires_auth
from sqlalchemy import *
from sqlalchemy.orm import *
from sqlalchemy.ext.declarative import *
from models import User, Operator, Item, Order, OrderItems
stripe_keys = {
'secret_key': os.environ['SECRET_KEY'],
'publishable_key': os.environ['PUBLISHABLE_KEY']
}
stripe.api_key = stripe_keys['secret_key']
engine = create_engine("mysql://root:@localhost/Truckate")
SessionMkr = sessionmaker()
SessionMkr.configure(bind=engine)
session = SessionMkr()
app = Flask(__name__)
# Robots.txt
@app.route('/robots.txt')
def robots():
return render_template('robots.txt')
# Checkout Stuff
@app.route('/checkout')
def checkout(amount):
return render_template('checkout.html', key=stripe_keys['publishable_key'], amount = amount)
@app.route('/charge', methods=['POST'])
def charge():
order = session.query(Order).filter(Order.id == int(request.form.get('order'))).first()
user = session.query(User).filter(User.id == int(request.form.get('user'))).first()
# Amount in cents
customer = stripe.Customer.create(
email=user.email,
card=request.form['stripeToken']
)
try:
charge = stripe.Charge.create(
customer=customer.id,
amount=order.total_amount(),
currency='usd',
description='Truckate Order'
)
order.status = 'paid'
session.commit()
return render_template('charge.html', amount=order.total_amount(),user = user)
except stripe.CardError, e:
render_template('/404.html')
@app.route('/truck/<truck>',methods=['GET','POST'])
def truck_page(truck):
items = session.query(Item).filter(Item.operator_id==truck)
if request.method == 'POST':
user = User()
user.firstName = request.form.get('firstName')
user.lastName = request.form.get('lastName')
user.email = request.form.get('email')
session.add(user)
session.commit()
order = Order()
order.operator_id = truck
order.user = user
order.status = 'created'
session.add(order)
session.commit()
for item_id in request.form.getlist('item'):
orderitem = OrderItems()
orderitem.order_id = order.id
orderitem.item_id = item_id
item = items.filter(Item.id == item_id).first()
session.add(orderitem)
session.commit()
return render_template('checkout.html',key=stripe_keys['publishable_key'],order=order,user=user)
else:
items = items.all()
items = filter(lambda x: x.count > 0, items)
return render_template('truckPage.html', items = items)
#Truck Admin Stuff
@app.route('/orders/<truck>', methods=['GET','POST'])
@requires_auth
def order_page(truck):
open_orders = session.query(Order).filter(Order.status=='paid').all()
if request.method == 'POST':
served = request.form.getlist('served')
for serve in served:
dbobject = session.query(Order).filter(Order.id == int(serve)).first()
dbobject.status = 'complete'
session.commit()
open_orders = session.query(Order).filter(Order.status=='paid').all()
return render_template('line.html', orders = open_orders)
else:
return render_template('line.html', orders = open_orders)
@app.route('/admin/<truck>',methods=['GET','POST'])
def admin_page(truck):
items = session.query(Item).filter(Item.operator_id==truck).all()
operator = session.query(Operator).filter(Operator.id == truck).first()
return render_template('truckAdmin.html',items = items,truck = operator)
#index
@app.route('/')
def index():
operators = [instance for instance in session.query(Operator) if instance.open == True]
return render_template('index.html',trucks = operators)
@app.route('/help')
def help():
return render_template('help.html')
if __name__ == '__main__':
app.run(debug=True)