def registerdone(request):
user = User(
username=request.POST.get( 'ID', False ),
last_name=request.POST.get( 'NAME', False ),
email=request.POST.get( 'EMAIL', False )
)
user.set_password(request.POST.get('PWD',False))
user.save()
return render(request,'demo/registerdone.html')
def register():
# 表单处理
if request.method == "POST":
username = request.form['username']
password = request.form['password']
check_password = request.form['check_password']
if(password != check_password):
flash("两次密码输入不同")
redirect(url_for('register'))
user_num = User.query.filter_by(username=username).count()
if user_num == 0:
user = User(username=username)
user.set_password(password)
db.session.add(user)
db.session.commit()
redirect(url_for('login'))
else:
flash("用户已存在")
redirect(url_for('register'))
return render_template('register.html')
def create(self, validated_data):
user = User(email=validated_data['email'])
user.set_password(validated_data['password'])
user.save()
return user
