def api_register():
username = request.form.get('username')
emailaddress = request.form.get('emailaddress')
password = request.form.get('password')
confirmpassword = request.form.get('confirmpassword')
if User.query.filter_by(emailaddress=emailaddress).first() is not None:
# print("该邮箱已被使用")
abort(400)
elif password != confirmpassword:
# print("密码不一致")
abort(400)
else:
new_user = User()
new_user.username = username
new_user.emailaddress = emailaddress
new_user.password = password
new_user.avatar = 'default.jpg'
new_user.active = True
db.session.add(new_user)
db.session.commit()
user = User.query.filter_by(emailaddress=emailaddress).first()
login_user(user)
return redirect(url_for('main.index'))
def test_avatar(self):
u = User(username='******', email='*****@*****.**')
self.assertEqual(u.avatar(128), ('https://www.gravatar.com/avatar/'
'd4c74594d841139328695756648b6bd6'
'?d=identicon&s=128'))
