def admin_send_message():
if request.method == 'POST':
mode = request.form["radio"]
if mode == 'one':
email = request.form["user_email"]
user = User.find_by_email(email)
if user:
subject = request.form["title"]
content = request.form["content"]
Post("admin", subject, content).admin_insert_post_by_type(
current_user, email, 'private')
posts = Post.admin_sent_posts()
flash("Message sent to {0}".format(email))
return render_template("admin/home.html", posts=posts)
else:
flash(
"The user you choose by {} does not exist.".format(email))
return render_template("admin/send_message.html")
elif mode == 'all':
subject = request.form["title"]
content = request.form["content"]
users = User.admin_find_all_users()
new_post = Post("admin", subject, content)
new_post.admin_insert_post(current_user, 'private')
for user in users:
Post.connect(user.email, new_post._id, 'private')
posts = Post.admin_sent_posts()
flash("Post sent to all.")
return render_template("admin/home.html", posts=posts)
return render_template("admin/send_message.html")
def admin_home():
posts = Post.admin_sent_posts()
if posts:
return render_template('admin/home.html',
posts=posts,
admin_email=current_user.email)
return render_template('admin/home.html', admin_email=current_user.email)