def home():
posts = Post.query.order_by('posted_on desc').all()
form = AddPost()
if request.method == 'GET':
if 'email' in session:
user_in_session = User.query.filter_by(email = session['email']).first()
user_in_session_username = user_in_session.username
return render_template('home.html',
posts = posts,
user_in_session_username = user_in_session_username,
form=form)
else:
return render_template('home.html',
posts = posts,
form = form)
if request.method == 'POST':
if form.validate() == False:
return render_template('home',
form=form)
else:
user_in_session = User.query.filter_by(email = session['email']).first()
user_in_session_username = user_in_session.username
post = Post(form.title.data,
form.text.data,
posted_on = func.now(),
user_id = user_in_session.id)
db_session.add(post)
db_session.commit()
flash('New entry was successfully posted')
return redirect(url_for('home'))
def add_entry():
if not 'email' in session:
abort(401)
user_in_session = User.query.filter_by(email = session['email']).first()
form = AddPost()
if request.method == 'POST':
if form.validate() == False:
return render_template('add_entry',
form=form)
else:
post = Post(form.title.data,
form.text.data,
posted_on = func.now(),
user_id = user_in_session.id)
db_session.add(post)
db_session.commit()
flash('New entry was successfully posted')
return redirect(url_for('home'))
if request.method == 'GET':
return render_template('add_entry.html',
form=form)
def signup():
form = SignupForm()
if 'email' in session:
return redirect(url_for('home'))
if request.method == 'POST':
if form.validate() == False:
return render_template('signup.html',
form=form)
else:
newuser = User(form.username.data, form.email.data, form.password.data)
db_session.add(newuser)
db_session.commit()
session['email'] = newuser.email
return redirect(url_for('home'))
elif request.method == 'GET':
return render_template('signup.html',
form=form)
