From 10fafd1e84e9c34da763e2fe1e66818c56bef5d3 Mon Sep 17 00:00:00 2001
From: Di Wu <dw323@cornell.edu>
Date: Tue, 12 Jan 2016 23:56:22 -0800
Subject: [PATCH] fix docs page

---
 hc/front/views.py | 23 ++++++++++++-----------
 1 file changed, 12 insertions(+), 11 deletions(-)

diff --git a/hc/front/views.py b/hc/front/views.py
index 312820f8..be714ff9 100644
--- a/hc/front/views.py
+++ b/hc/front/views.py
@@ -55,10 +55,7 @@ def my_checks(request):
     return render(request, "front/my_checks.html", ctx)
 
 
-def index(request):
-    if request.user.is_authenticated():
-        return redirect("hc-checks")
-
+def _welcome_check(request):
     check = None
     if "welcome_code" in request.session:
         code = request.session["welcome_code"]
@@ -67,8 +64,16 @@ def index(request):
     if check is None:
         check = Check()
         check.save()
-        code = str(check.code)
-        request.session["welcome_code"] = code
+        request.session["welcome_code"] = str(check.code)
+
+    return check
+
+
+def index(request):
+    if request.user.is_authenticated():
+        return redirect("hc-checks")
+
+    check = _welcome_check(request)
 
     ctx = {
         "page": "welcome",
@@ -80,11 +85,7 @@ def index(request):
 
 
 def docs(request):
-    if "welcome_code" in request.session:
-        code = request.session["welcome_code"]
-        check = Check.objects.get(code=code)
-    else:
-        check = Check(code="uuid-goes-here")
+    check = _welcome_check(request)
 
     ctx = {
         "page": "docs",