from gluon.fileutils import read_file
response.menu = [['Register User', False, URL(r=request,
f='register_user')], ['Register Dog', False,
URL('register_dog')], ['Register Product'
, False, URL('register_product')],
['Buy product', False, URL('buy')]]
def register_user():
""" simple user registration form with validation and database.insert()
also lists all records currently in the table"""
# ## create an insert form from the table
form = SQLFORM(db.users)
# ## if form correct perform the insert
if form.accepts(request.vars, session):
response.flash = 'new record inserted'
# ## and get a list of all users
records = SQLTABLE(db().select(db.users.ALL))
return dict(form=form, records=records)
def register_dog():
""" simple user registration form with validation and database.insert()
also lists all records currently in the table"""
form = SQLFORM(db.dogs)
if form.accepts(request.vars, session):
response.flash = 'new record inserted'
download = URL('download') # to see the picture
records = SQLTABLE(db().select(db.dogs.ALL), upload=download)
return dict(form=form, records=records)
def register_product():
""" simple user registration form with validation and database.insert()
also lists all records currently in the table"""
form = SQLFORM(db.products)
if form.accepts(request.vars, session):
response.flash = 'new record inserted'
records = SQLTABLE(db().select(db.products.ALL))
return dict(form=form, records=records)
def buy():
""" uses a form to query who is buying what. validates form and
updates existing record or inserts new record in purchases """
buyerRecords = db().select(db.users.ALL)
buyerOptions = []
for row in buyerRecords:
buyerOptions.append(OPTION(row.name, _value=row.id))
productRecords = db().select(db.products.ALL)
productOptions = []
for row in productRecords:
productOptions.append(OPTION(row.name, _value=row.id))
form = FORM(TABLE(
TR('Buyer id:',
SELECT(buyerOptions,_name='buyer_id')),
TR('Product id:',
SELECT(productOptions,_name='product_id')),
TR('Quantity:',
INPUT(_type='text', _name='quantity',
requires=IS_INT_IN_RANGE(1, 100))),
TR('',
INPUT(_type='submit', _value='Order'))
))
if form.accepts(request.vars, session, keepvalues=True):
# ## check if user is in the database
if len(db(db.users.id == form.vars.buyer_id).select()) == 0:
form.errors.buyer_id = 'buyer not in database'
# ## check if product is the database
if len(db(db.products.id == form.vars.product_id).select())\
== 0:
form.errors.product_id = 'product not in database'
# ## if no errors
if len(form.errors) == 0:
# ## get a list of same purchases by same user
purchases = db((db.purchases.buyer_id == form.vars.buyer_id)
& (db.purchases.product_id
== form.vars.product_id)).select()
# ## if list contains a record, update that record
if len(purchases) > 0:
purchases[0].update_record(quantity=purchases[0].quantity
+ form.vars.quantity)
else:
# ## or insert a new record in table
db.purchases.insert(buyer_id=form.vars.buyer_id,
product_id=form.vars.product_id,
quantity=form.vars.quantity)
response.flash = 'product purchased!'
if len(form.errors):
response.flash = 'invalid values in form!'
# ## now get a list of all purchases
# quick fix to make it runnable on gae
if purchased:
records = db(purchased).select(db.users.name,
db.purchases.quantity,
db.products.name)
else:
records = db().select(db.purchases.ALL)
return dict(form=form, records=SQLTABLE(records), vars=form.vars,
vars2=request.vars)
def delete_purchased():
""" delete all records in purchases """
db(db.purchases.id > 0).delete()
redirect(URL('buy'))
def reset_purchased():
""" set quantity=0 for all records in purchases """
db(db.purchases.id > 0).update(quantity=0)
redirect(URL('buy'))
def download():
""" used to download uploaded files """
import gluon.contenttype
app = request.application
filename = request.args[0]
response.headers['Content-Type'] = \
gluon.contenttype.contenttype(filename)
return read_file('applications/%s/uploads/%s' % (app, filename), 'rb')