def regidter():
form = CustomerRegForm()
if request.method == 'POST':
if form.validate():
# 从表单上取出数据添加到Customer数据模型对象
new_customer = Customer()
new_customer.id = form.userid.data
new_customer.name = form.name.data
new_customer.password = form.password.data
new_customer.address = form.address.data
new_customer.birthday = form.birthday.data
new_customer.phone = form.phone.data
db.session.add(new_customer)
db.session.commit()
print('注册成功')
return render_template('customer_reg_success.html', form=form)
# else:
# render_template('customer_reg.html', form=form)
return render_template('customer_reg.html', form=form)
def customers_save_process(request):
if request.session.is_empty():
return redirect('/login/')
user = User.objects.get(username=request.session.get('username'))
name = request.GET['name']
owner_first_name = request.GET['owner_first_name']
owner_last_name = request.GET['owner_last_name']
address = request.GET['address']
landline = request.GET['landline']
email = request.GET['email']
mobile = request.GET['mobile']
bank = request.GET['bank']
bank_number = request.GET['bank_number']
customer = Customer()
customer.name = name
customer.owner_first_name = owner_first_name
customer.owner_last_name = owner_last_name
customer.address = address
customer.landline = landline
customer.email = email
customer.mobile = mobile
customer.bank = bank
customer.bank_number = bank_number
try:
customer.save()
user.branch.customer.add(customer)
sweetify.sweetalert(request, icon='success', title='Added Customer Successfully', text='{} successfully added'.format(customer.name), persistent='Dismiss')
except:
sweetify.sweetalert(request, icon='error', title='Something went wrong', persistent='Dismiss')
return redirect('/customer/')
