def deletebusiness(business_id): qry = db.session.query(Business).filter(Business.id == id) business = qry.first() if Business: form = BusinessForm(formdata=request.form, obj=Business) if request.method == 'POST' and form.validate(): business = Business.query.get_or_404(business_id) db.session.delete(business) db.session.commit() flash('Your business has been deleted', 'success') return redirect(url_for('available')) return render_template('delete.html', form=form) else: return 'Error deleting #{id}'.format(id=id)
def update_business(business_id): business = Business.query.get_or_404(business_id) form = BusinessForm() if form.validate_on_submit(): business.name = form.name.data business.location = form.location.data business.business_description = form.business_description.data db.session.commit() flash(f'Your business has been updated', 'success') return redirect(url_for('single_business', business_id=business.id)) elif request.method == 'GET': form.name.data = business.name form.location.data = business.location form.business_description.data = business.business_description return render_template('register_biz.html', title='update', form=form)
def businesses(): form = BusinessForm() if form.validate_on_submit(): business = Business( name=form.name.data, location=form.location.data, started=form.date.data, business_description=form.business_description.data) db.session.add(business) db.session.commit() flash(f'Successfully registered {business.name} business', 'success') return redirect(url_for('businesses_present')) else: flash( 'Your business was not registered. Please check your details and try again ', 'danger') return render_template('register_biz.html', title='Register-biz', form=form)
def update_business(business_id): business = Business.query.get_or_404(business_id) if business.owner != current_user: abort(403) form = BusinessForm() if form.validate_on_submit(): business.BusinessName = form.BusinessName.data business.BusinessLocation = form.BusinessLocation.data business.business_description = form.business_description.data db.session.commit() flash(f'Your business has been updated', 'success') return redirect(url_for('single_business', business_id=business.id)) elif request.method == 'GET': form.BusinessName.data = business.BusinessName form.BusinessLocation.data = business.BusinessLocation form.business_description.data = business.business_description return render_template('business.html', title='update', form=form, legend='Update Business', btn='Update')
def BusinessFunction(): form = BusinessForm() if form.validate_on_submit(): business = Business( BusinessName=form.BusinessName.data, BusinessLocation=form.BusinessLocation.data, date_established=form.date_established.data, business_description=form.business_description.data) db.session.add(business) db.session.commit() flash( f'Your Business {business.BusinessName} has successfully been registered', 'success') return redirect(url_for('available')) else: flash( 'Your business not registered please check on your details and try again', 'danger') return render_template('business.html', title='Business', form=form, legend='Register Business', btn='Register')
def setUp(self): self.login = LoginForm() self.register = RegisterForm() self.business = BusinessForm()