def add_post():
"""Add new post"""
user = User.get_single_user(request.form["user"])
title = request.form["title"]
content = request.form["content"]
tags = request.form.getlist("tags")
if (title == "" or content == ""):
flash("Missing Input: Please fill out both Title and Content")
return redirect(f"/create_post/{user.id}")
post = Post(title=title, content=content, user_id=user.id)
Post.add_post(post)
for tag in tags:
PostTag.add_post_tag(PostTag(post_id=post.id, tag_id=tag))
return redirect(f"/post_detail/{post.id}")
def edit_post(id):
"""Edit existing customer information in database"""
title = request.form["title"]
content = request.form["content"]
tags = request.form.getlist("tags")
if (title == "" or content == ""):
flash("Missing Input: Please fill out both Title and Content")
return redirect(url_for("edit_post_form", id=id))
post = Post.get_single_post(id)
Post.update_post(post, [title, content])
PostTag.delete_post_tags(post.id)
for tag in tags:
PostTag.add_post_tag(PostTag(post_id=post.id, tag_id=tag))
return redirect(f"/post_detail/{id}")
def setUp(self):
"""Clear Database"""
User.query.delete()
Post.query.delete()
Tag.query.delete()
self.user = User(**USER_DATA)
db.session.add(self.user)
db.session.commit()
self.post = Post(**POST_DATA, user_id=self.user.id)
db.session.add(self.post)
db.session.commit()
self.tag = Tag(**TAG_DATA)
db.session.add(self.tag)
db.session.commit()
PostTag.add_post_tag(PostTag(post_id=self.post.id, tag_id=self.tag.id))
