def delete_post(post_id):
posts = models.Post.query.filter(models.Post.id == post_id).first()
if (posts):
db_session.delete(posts)
db_session.commit()
flash("You've successfully deleted the post!", "success")
return redirect(url_for('index'))
else:
abort(404)
def post():
form = forms.PostForm()
if form.validate_on_submit():
user = current_user.id
post = models.Post(user_id=user, content=form.content.data.strip())
flash("Message posted! Thanks!", "success")
db_session.add(post)
db_session.commit()
return redirect(url_for('index'))
return render_template('post.html', form=form)
def register():
form = forms.RegisterForm()
if form.validate_on_submit():
flash("Yay, you registered!", "success")
username = form.username.data
email = form.email.data
hashed_pswd = generate_password_hash(form.password.data).decode('utf8')
user = models.User(username=username,
email=email,
password=hashed_pswd)
db_session.add(user)
db_session.commit()
return redirect(url_for('index'))
return render_template('register.html', form=form)
def unfollow(username):
try:
to_user = models.User.query.filter(
models.User.username.like(username)).first()
except models.DoesNotExist:
abort(404)
else:
try:
relationship = db_session.query(models.Relationship).filter(
and_((models.Relationship.from_user_id == current_user.id),
(models.Relationship.to_user_id == to_user.id))).first()
db_session.delete(relationship)
db_session.commit()
except models.IntegrityError:
pass
else:
flash("You've unfollowed {}!".format(to_user.username), "success")
return redirect(url_for('stream', username=to_user.username))
def follow(username):
try:
to_user = models.User.query.filter(
models.User.username.like(username)).first()
except models.DoesNotExist:
abort(404)
else:
try:
relationship = models.Relationship(from_user_id=current_user.id,
to_user_id=to_user.id)
db_session.add(relationship)
db_session.commit()
except models.IntegrityError:
pass
else:
flash("You're now following {}!".format(to_user.username),
"success")
return redirect(url_for('stream', username=to_user.username))
def index():
reg_form = RegistrationForm()
# Update database if validation success
if reg_form.validate_on_submit():
username = reg_form.username.data
password = reg_form.password.data
# Hash password
hashed_pswd = pbkdf2_sha256.hash(password)
# Add username & hashed password to DB
user = User(username=username, hashed_pswd=hashed_pswd)
db_session.add(user)
db_session.commit()
flash('Registered successfully. Please login.', 'success')
return redirect(url_for('login'))
return render_template("index.html", form=reg_form)