diff --git a/brutaldon/views.py b/brutaldon/views.py index accc028..303e443 100644 --- a/brutaldon/views.py +++ b/brutaldon/views.py @@ -199,10 +199,11 @@ def fav(request, id): mastodon = get_mastodon(request) toot = mastodon.status(id) if request.method == 'POST': - if toot.favourited: - mastodon.status_unfavourite(id) - else: - mastodon.status_favourite(id) + if not request.POST['cancel']: + if toot.favourited: + mastodon.status_unfavourite(id) + else: + mastodon.status_favourite(id) return redirect(thread, id) else: return render(request, 'main/fav.html', {"toot": toot}) @@ -211,10 +212,11 @@ def boost(request, id): mastodon = get_mastodon(request) toot = mastodon.status(id) if request.method == 'POST': - if toot.reblogged: - mastodon.status_unreblog(id) - else: - mastodon.status_reblog(id) + if not request.POST['cancel']: + if toot.reblogged: + mastodon.status_unreblog(id) + else: + mastodon.status_reblog(id) return redirect(thread, id) else: return render(request, 'main/boost.html', {"toot": toot})