def edit_material_page(name):
if not is_current_user_admin():
abort(403)
material = item_db.get(name)
if not material:
abort(404)
return template('edit_material', material=material)
def edit_material_submit():
if not is_current_user_admin():
abort(403)
name = request.forms.get('name')
try:
buy_price = float(request.forms.get('buy_price', ''))
except ValueError:
buy_price = None
try:
desired_quantity = int(request.forms.get('desired_quantity', ''))
except ValueError:
desired_quantity = 0
material = item_db.get(name)
if not material:
abort(404)
material.buy_price = buy_price
material.desired_quantity = desired_quantity
if isinstance(material, item_db.Ship):
redirect('/ships')
elif isinstance(material, item_db.Module):
redirect('/modules')
else:
redirect('/materials')
