Coverage for website/shortlinks/views.py: 61.54%

1from django.http import HttpResponseRedirect 

2from django.views.generic.detail import DetailView 

3 

4from .models import ShortLink 

5 

6 

7class ShortLinkView(DetailView): 

8 model = ShortLink 

9 context_object_name = "link" 

10 template_name = "shortlinks/confirm.html" 

11 

12 def render_to_response(self, context, **kwargs): 

13 if self.object.immediate: 

14 return HttpResponseRedirect(self.object.url) 

15 return super().render_to_response(context, **kwargs)