def add_item():
if request.method == 'GET':
if (login_session['username'] is not None):
categories = db.query(Category).all()
return render_template('add_item.html',
login_session=login_session,
categories=categories)
else:
return render_template('login.html',
error="You must be logged in to do that!")
elif request.method == 'POST':
if login_session['username'] is not None:
categories = db.query(Category).all()
item_name = request.form.get('item_name')
price = request.form.get('price')
description = request.form.get('description')
category = request.form.get('category')
new_category = request.form.get('new_category')
if item_name is not None and item_name != '':
item = Items(name=item_name)
if price is not None and item.price != '':
item.price = price
if description is not None and description != '':
item.description = description
item.author = getUserInfo(login_session['user_id'])
else:
flash("You need to give an item!")
return redirect(url_for('add_item'))
if category is not None and category != '':
c = db.query(Category).filter_by(name=category).first()
item.category = c
elif new_category is not None and new_category != '':
c = db.query(Category).filter_by(name=new_category).first()
if (c):
flash("You cannot add a category that already exists!")
return redirect(url_for('add_item'))
new_category = Category(name=new_category)
item.category = new_category
db.add(new_category)
else:
flash("You need to provide a category!")
return redirect(url_for('add_item'))
db.add(item)
db.commit()
return redirect(url_for('index'))
else:
flash("Please log in to add items")
return redirect(url_for('index'))
def addItem():
if request.method == 'GET':
# Make sure only logged in users can access this page
if ('username' in login_session):
return render_template('addItem.html',
login_session=login_session)
else:
flash('Please login in order to add key/value pairs')
return redirect(url_for('login'))
elif request.method == 'POST':
# Make sure only logged in users are adding key/value pairs
if ('username' in login_session):
key = request.form.get('key')
value = request.form.get('value')
item = db.query(Items).filter_by(key=key).first()
# Make sure key is unique/not already added
if item:
flash('"%s" has already been added' % item.key)
return redirect(url_for('addItem'))
if key is not None and key != '':
item = Items(key=key)
if value is not None and value != '':
item.value = value
item.author = getUserInfo(login_session['user_id'])
else:
flash('You need to provide a proper Key/Value pair')
return redirect(url_for('addItem'))
db.add(item)
db.commit()
flash('Item Added!')
return redirect(url_for('index'))
else:
flash('Please login in order to add key/value pairs')
return redirect(url_for('login'))
else:
return redirect(url_for('index'))
