diff --git a/app/main/routes.py b/app/main/routes.py index 82b10fb..af640fc 100644 --- a/app/main/routes.py +++ b/app/main/routes.py @@ -153,6 +153,13 @@ def archive_remove_user(post_user, post_id): flash(_('You have removed %(username)s post from your archive!', username=post_user)) return redirect(url_for('main.view_archive', username=current_user.username)) +@bp.route('/delete/') +@login_required +def delete(post_id): + current_user.delete_post(post_id) + db.session.commit() + flash(_('You have delete your post!')) + return redirect(url_for('main.index')) @bp.route('/follow/', methods=['POST']) @login_required diff --git a/app/models.py b/app/models.py index b7c8c8d..8c51e96 100644 --- a/app/models.py +++ b/app/models.py @@ -141,6 +141,9 @@ class User(UserMixin, PaginatedAPIMixin, db.Model): db.session.add(a) print("Archived!") + def delete_post(self, post_id): + return self.posts.filter_by(id=post_id).delete() + def archive_remove(self, post_id): self.archived.filter_by(id=post_id).delete() print("Removed Archive!") diff --git a/app/templates/_post.html b/app/templates/_post.html index cded509..02ace7f 100644 --- a/app/templates/_post.html +++ b/app/templates/_post.html @@ -36,6 +36,9 @@ Remove from Archive {% endif %} {% endif %} + {% if post.author.username == current_user.username %} + Delete + {% endif %}