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))