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)