Add the ability to redirect to the resulting link, some styling

[bigly] / src / bigly / views.py

diff --git a/src/bigly/views.py b/src/bigly/views.py
index f88753a..4a3317c 100644 (file)
--- a/src/bigly/views.py
+++ b/src/bigly/views.py
@@ -1,6 +1,6 @@
 from urllib.parse import urlparse, urlunparse, parse_qs
 
-from django.shortcuts import render
+from django.shortcuts import redirect, render
 from django.views.generic.base import TemplateView
 
 from rest_framework import status, viewsets
@@ -78,11 +78,15 @@ def embiggen(request):
         remove_utm = serializer.data['remove_utm'],
     )
 
-    return render(
-        request,
-        'bigly/link_info.html',
-        result,
-    )
+    if serializer.data['handler'] == 'redirect':
+        return redirect(result['link'])
+
+    else:
+        return render(
+            request,
+            'bigly/link_info.html',
+            result,
+        )
 
 class FollowRedirectsViewSet(viewsets.ViewSet):
     serializer_class = serializers.FollowRedirectsSerializer