def edit_address(aid): address = Address.query.filter_by(hashkey=aid).first() form = AddressForm(obj=address) if form.validate_on_submit(): if address is None: address=Address() address.user = g.user db.session.add(address) form.populate_obj(address) db.session.commit() return redirect(url_for('confirm_payment')) return render_template('address.html.jinja2', user=g.user, addresses=Address.query.filter_by(user=g.user), form=form, title='Select Billing Address')
def select_address(invoice, workspace): """ Select an Address or enter a new one. Session must contain invoice id if it exists """ form = AddressForm() if form.validate_on_submit(): address = Address(user=g.user) db.session.add(address) form.populate_obj(address) db.session.commit() session['workspace'] = workspace.name session['address'] = getattr(address, 'hashkey', None) session['invoice'] = invoice.id print "SESSION WORKSPACE:", session['workspace'] print "SESSION ADDRESS:", session['address'] print "SESSION INVOICE:", session['invoice'] return redirect(url_for('confirm_payment')) addresses = Address.get(g.user) return render_template('address.html', form=form, invoice=invoice, workspace=workspace, addresses=addresses)