diff --git a/urlforward/templates/details.html b/urlforward/templates/details.html index d140b1ada1dd2e209e018c6b734bb553cdcdafbf..d40dcbe93e24fc7b60a5a65e1a35c14a5996742d 100644 --- a/urlforward/templates/details.html +++ b/urlforward/templates/details.html @@ -11,4 +11,9 @@ <li>Followed: {{ url.used }} time(s).</li> </ul> +<p> + <a href="{% url 'delete' url.id %}" role="button" + class="btn btn-danger">Delete</a> +</p> + {% endblock %} diff --git a/urlforward/urls.py b/urlforward/urls.py index 5ec32ea3db8274226deb2d96b4da14e468b556e8..c12d4d31fe60fbd5ed666cedac087b4cf192fcc1 100644 --- a/urlforward/urls.py +++ b/urlforward/urls.py @@ -8,4 +8,5 @@ urlpatterns = [ url(r'^url=(?P<url_id>[0-9]+)$', views.details, name='details'), url(r'^new$', views.new, name='new'), url(r'^generate$', views.generate, name='generate'), + url(r'^delete=(?P<url_id>[0-9]+)$', views.delete, name='delete'), ] diff --git a/urlforward/views.py b/urlforward/views.py index 6935fe60b19c494130f5e1d11eae7bb399b960f2..ef26369f63017924a2202090f7dafff9c7e1a288 100644 --- a/urlforward/views.py +++ b/urlforward/views.py @@ -39,3 +39,9 @@ def generate(request): u = Url(target=request.POST['target'], user=request.user) u.save() return HttpResponseRedirect(reverse('index')) + +@login_required +def delete(request, url_id): + u = get_object_or_404(Url, id=url_id, user__exact=request.user) + u.delete() + return HttpResponseRedirect(reverse('index'))