def join_with_invitation():
form = InvitationForm()
next = request.args.get('next') or url_for('.register')
if form.validate_on_submit():
username = form.username.data
recommender_id = current_app.redis.get('uname:%s:uid'%username)
current_app.logger.info('recommender_id:%s' % recommender_id)
if not recommender_id:
flash("查无此人")
return render_template('join_with_invitation.html', form=form, next=next)
fields = ['id', 'username', 'photo']
recommender = current_app.redis.hmget('user:%s'%recommender_id, fields)
recommender = dict(zip(fields, recommender))
current_app.logger.info('recommender: %s' % recommender)
if int(recommender.get('invite_quota_left', 0)) < 1:
flash("邀请名额已用完")
return render_template('join_with_invitation.html', form=form, next=next)
session['uid'] = recommender_id
return redirect(next)
return render_template('join_with_invitation.html', form=form, next=next)
def index():
form = InvitationForm()
if form.validate_on_submit():
person = Person(booking_number=form.booking_number.data,
name=form.name.data,
last_name=form.last_name.data,
sex=form.sex.data,
passport_number=form.passport_number.data,
delivery_needed=form.delivery_needed.data,
address=form.address.data,
citizenship=form.citizenship.data,
birth_date=form.birth_date.data,
entry_date=form.entry_date.data,
exit_date=form.exit_date.data,
kids=form.kids.data,
transport=form.transport.data,
email=form.email.data,
cities=form.cities.data,
if_double=form.if_double.data)
db.session.add(person)
db.session.commit()
flash('Thank you!')
return redirect('/')
return render_template('index.html',
title="Invitation form",
form=form)
def invite():
form = InvitationForm()
if request.method == 'POST' and form.validate_on_submit():
emails = request.form.get('emails')
message = request.form.get('message')
invite_info = Invitations(
user_id=current_user.id,
emails=emails,
message=message
)
db.session.add(invite_info)
db.session.commit()
email_list = emails.split()
for email in email_list:
msg_body = '%s <br><br> %s <br><br> esyakutuphanesi.com'\
% (current_user.name, message)
html_msg = '%s <br><br> %s <br><br> <a href="http://esyakutuphanesi.com/">esyakutuphanesi.com</a>' \
% (current_user.name, message)
msg_subject = u"%s seni Eşya Kütüphanesi'ne davet ediyor!" % current_user.name
msg = MailMessage(
body=msg_body,
html=html_msg,
subject=msg_subject,
sender=(u"Eşya Kütüphanesi", "*****@*****.**"),
recipients=[email]
)
mail.send(msg)
flash(u"Davetini ilettik!")
return redirect(url_for('invite'))
return render_template("invite.html", form=form, user=current_user)
