def add_stall() :
stall_form = StallForm(request.form)
# get a list of hawkers for this user. If admin, get all hawkers
stalls = None
users = None
if g.user.is_admin :
stalls = Hawker.query.all()
for stall in stalls :
owner = User.query.get(stall.owner)
stall.user = owner
users = User.query.all()
stall_form.owner.choices = [(user.id, user.login) for user in users]
else :
stalls = Hawker.query.filter_by(owner=g.user.id).all()
stall_form.owner.choices = [(g.user.id, g.user.login)]
if request.method == 'POST' and stall_form.validate() :
# add a stall for this hawker
h = Hawker()
h.owner = stall_form.owner.data
h.name = stall_form.name.data
h.address = stall_form.address.data
h.pincode = stall_form.pincode.data
h.email = stall_form.email.data
h.contact_number = stall_form.contact_number.data
db.session.add(h)
db.session.commit()
return redirect(url_for('vendor_page.index'))
return render_template('addstall.html', form=stall_form)
def populatedata() :
from hawkers.models import db, Hawker, Food, User
populatevendor()
# Empty the tables first
print(Hawker.query.delete(), " rows deleted from Hawker")
print(Food.query.delete(), " rows deleted from Food")
# add hawker
h = Hawker()
h.name = 'Indian Stall'
h.address = 'Golden Palace Eating House, 5 Kallang Sector, Singapore'
h.pincode = 349279
h.contact_number = 99991111
h.email = '*****@*****.**'
h.owner = User.query.filter_by(login='******').first().id
db.session.add(h)
db.session.commit()
# add foods for this menu
f1 = Food()
f1.name = 'Dosa'
f1.description = 'Indian style lentil pancakes'
f1.price = 1.00
f1.is_available = True
f1.image = 'dosa.jpg'
f1.hawker_id = h.id
f2 = Food()
f2.name = 'South Indian Lunch'
f2.description = 'Complete Platter with variety rice and side dish'
f2.price = 3.20
f2.is_available = False
f2.image = '2.jpg'
f2.hawker_id = h.id
f3 = Food()
f3.name = 'Idli'
f3.description = 'Indian Style steamed rice cakes'
f3.is_available = True
f3.price = 1.00
f3.image = 'idli.jpg'
f3.hawker_id = h.id
db.session.add_all([f1, f2, f3])
db.session.commit()
