def update_product(id): if session['role'] == 'seller': product = Product.query.get_or_404(id) if product.seller != current_user: abort(403) form = ProductForm() if form.validate_on_submit(): product.name = form.name.data product.description = form.description.data product.category = form.category.data product.price = form.price.data product.quantity = form.quantity.data if product.photo_name is None: product.photo_name = save_picture(form.photo.data) elif form.photo.data: pathp = app.root_path + '\static\product_pics\{}'.format( product.photo_name) os.remove(pathp) product.photo_name = save_picture(form.photo.data) db.session.commit() flash('Product has been updated!', 'check') return redirect(url_for('seller_dashboard')) elif request.method == 'GET': form.name.data = product.name form.category.data = product.category form.description.data = product.description form.price.data = product.price form.quantity.data = product.quantity return render_template('new_product.html', legend='Update Product', form=form) else: flash('Only Seller can access this page', 'warning') return redirect(url_for('customer_home'))
def new_product(): if session['role'] == 'seller': form = ProductForm() if form.validate_on_submit(): product = Product(name=form.name.data, description=form.description.data, category=form.category.data, price=form.price.data, quantity=form.quantity.data, photo_name=save_picture(form.photo.data), seller=current_user) db.session.add(product) db.session.commit() flash('Product has been added successfully!', 'check') return redirect(url_for('seller_dashboard')) return render_template('new_product.html', legend="Add Product", form=form) else: flash('Only Seller can access this page', 'warning') return redirect(url_for('customer_home'))