예제 #1
0
 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)
예제 #2
0
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)