def edit_vendor(id): """ Edit a vendor """ check_admin() add_vendor = False vendor = Vendor.query.get_or_404(id) form = VendorForm(obj=vendor) if form.validate_on_submit(): vendor.name = form.name.data vendor.description = form.description.data db.session.commit() flash('You have successfully edited the vendor.') # redirect to the vendors page return redirect(url_for('admin.list_vendors')) form.description.data = vendor.description form.name.data = vendor.name return render_template('admin/vendors/vendor.html', action="Edit", add_vendor=add_vendor, form=form, vendor=vendor, title="Edit Vendor")
def add_vendor(): """ Add a vendor to the database """ check_admin() add_vendor = True form = VendorForm() if form.validate_on_submit(): vendor = Vendor(name=form.name.data, description=form.description.data) try: # add vendor to the database db.session.add(vendor) db.session.commit() flash('You have successfully added a new vendor.') except: # in case vendor name already exists flash('Error: vendor name already exists.') # redirect to vendors page return redirect(url_for('admin.list_vendors')) # load vendor template return render_template('admin/vendors/vendor.html', action="Add", add_vendor=add_vendor, form=form, title="Add Vendor")