def post(self): address_all = current_user.address_user_set phone = request.form.get('phone', None) site = request.form.get('sit', None) name = request.form.get('name', None) if phone and site and name: address = Address() address.phone = phone address.site = site address.name = name address.user_address = current_user.id db.session.add(address) db.session.commit() flash('添加完成') return redirect(url_for('address.address')) flash('检查输入完整') return render_template('./user/address.html', address_all=address_all)
def address(): if not current_user.is_authenticated: return redirect(url_for('signup')) form = AddresForm() if form.validate_on_submit(): address = Address.query.filter_by(user_id=current_user.id).first() if not address: address = Address(name=form.name.data, user_id=current_user.id) db.session.add(addres) db.session.commit() flash('Address added') else: address.name = form.name.data db.session.commit() flash('Edit Address') return render_template('address.html', form=form)