def register():
"""
注册
"""
logger.info('url = ' + str(request.url))
# 如果用户已经登录，则直接返回主页
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegisterForm()
if form.validate_on_submit():
name = form.name.data
# 将邮箱转成小写，避免验证出问题
email = form.email.data.lower()
username = form.username.data
password = form.password.data
user = User(name=name, email=email, username=username)
user.set_password(password)
db.session.add(user)
db.session.commit()
# 获取token
token = generate_token(user=user, operation='confirm')
# 发送验证邮箱
send_confirm_email(user=user, token=token)
flash('邮件已发送，请登录邮箱验证！', 'info')
return redirect(url_for('.login'))
return render_template('auth/register.html', form=form)
def register():
invite_token = request.args.get('token')
data = None
if invite_token:
data = validate_invite_token(invite_token)
if current_user.is_authenticated:
return redirect(url_for('main.index'))
if data:
form = RegisterForm(email=data['email'], role='Doctor')
else:
form = RegisterForm()
if form.validate_on_submit():
name = form.name.data
email = form.email.data.lower()
username = form.username.data
password = form.password.data
user = User(name=name, email=email, username=username)
user.set_role_with_name(form.role.data)
if form.role.data == 'Doctor':
doctor = Doctor()
user.doctor = doctor
user.set_password(password)
db.session.add(user)
db.session.commit()
token = generate_token(user=user, operation='confirm')
send_confirm_email(user=user, token=token)
flash('Confirm email sent, check your inbox.', 'info')
try:
photo = Photo.query.get(data['photo_id'])
if photo:
invite = Invite(photo_id=data['photo_id'],
user=user,
token_id=data['stripe_token_id'])
db.session.add(invite)
push_invite_notification(photo, user)
else:
flash(
'Unfortunately the photo that you have to comment has been removed by'
+ Markup(' <a href="%s">owner</a>. ' % url_for(
'user.index', username=data['sender_name'])),
'warning')
except Exception as e:
db.session.remove()
return redirect(url_for('.login'))
if data:
flash('Welcome! You will help many people on this site.', 'info')
return render_template('auth/register.html',
form=form,
invite_token=invite_token)
def register():
if current_user.is_authenticated:
return redirect(url_for("main.index"))
form = RegisterForm()
if form.validate_on_submit():
username = form.username.data
name = form.name.data
email = form.email.data.lower()
password = form.password.data
user = User(name=name, username=username, email=email, confirmed=True)
user.set_password(password)
db.session.add(user)
db.session.commit()
flash("Thanks for your register", "success")
return redirect(url_for('.login'))
return render_template('auth/register.html', form=form)
def register():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegisterForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
user = User(username=username)
user.set_password(password)
db.session.add(user)
db.session.commit()
# 邮箱验证部分暂时不做
# token = generate_token(user=user, operation='confirm')
# send_confirm_email(user=user, token=token)
return redirect(url_for('.login'))
return render_template('auth/register.html', form=form)
def register():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegisterForm()
if form.validate_on_submit():
name = form.name.data
email = form.email.data.lower()
username = form.username.data
password = form.password.data
user = User(name=name, email=email, username=username)
user.set_password(password)
db.session.add(user)
db.commit()
token = generate_token(user=user, operation=Operations.CONFIRM)
send_confirm_email(user=user, token=token)
flash('Confirm email sent, check your inbox.', 'info')
return redirect(url_for('.login'))
return render_template('auth/register.html', form=form)
def register():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegisterForm()
if form.validate_on_submit():
user = User(name=form.name.data,
email=form.email.data,
username=form.username.data,
role_id=4,
confirmed=True)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
token = generate_token(user=user, operation='confirm')
send_confirm_email(user=user, token=token)
flash('注册成功，请登录', 'info')
return redirect(url_for('.login'))
return render_template('auth/register.html', form=form)
