auth working, getting user data a problem

rowers/views.py

@@ -21,6 +21,7 @@ from django.views.decorators.csrf import csrf_exempt
 from matplotlib.backends.backend_agg import FigureCanvas
 import gc
 from pyparsing import ParseException
+from uuid import uuid4
 
 from django.shortcuts import render
 from django.http import (
@@ -93,6 +94,8 @@ from sporttracksstuff import SportTracksNoTokenError,sporttracks_open
 from tpstuff import TPNoTokenError,tp_open
 from iso8601 import ParseError
 import stravastuff
+import polarstuff
+from polarstuff import PolarNoTokenError
 from stravastuff import StravaNoTokenError
 import sporttracksstuff
 import underarmourstuff
@@ -104,6 +107,7 @@ from ownapistuff import TEST_CLIENT_ID, TEST_CLIENT_SECRET, TEST_REDIRECT_URI
 from rowsandall_app.settings import (
     C2_CLIENT_ID, C2_REDIRECT_URI, C2_CLIENT_SECRET,
     STRAVA_CLIENT_ID, STRAVA_REDIRECT_URI, STRAVA_CLIENT_SECRET,
+    POLAR_CLIENT_ID, POLAR_REDIRECT_URI, POLAR_CLIENT_SECRET,
     SPORTTRACKS_CLIENT_ID, SPORTTRACKS_REDIRECT_URI,
     SPORTTRACKS_CLIENT_SECRET,
     UNDERARMOUR_CLIENT_ID, UNDERARMOUR_REDIRECT_URI,
@@ -2358,7 +2362,7 @@ def workout_sporttracks_upload_view(request,id=0):
 def rower_c2_authorize(request):
     # Generate a random string for the state parameter
     # Save it for use later to prevent xsrf attacks
-    from uuid import uuid4
+
     state = str(uuid4())
     scope = "user:read,results:write"
     params = {"client_id": C2_CLIENT_ID,
@@ -2373,7 +2377,7 @@ def rower_c2_authorize(request):
 def rower_strava_authorize(request):
     # Generate a random string for the state parameter
     # Save it for use later to prevent xsrf attacks
-    from uuid import uuid4
+
     state = str(uuid4())
 
     params = {"client_id": STRAVA_CLIENT_ID,
@@ -2385,12 +2389,30 @@ def rower_strava_authorize(request):
 
     return HttpResponseRedirect(url)
 
+# Polar Authorization
+@login_required()
+def rower_polar_authorize(request):
+
+    state = str(uuid4())
+
+    params = {"client_id": POLAR_CLIENT_ID,
+              "response_type": "code",
+              "redirect_uri": POLAR_REDIRECT_URI,
+              "state": state,
+#	      "scope":"accesslink.read_all"
+    }
+    url = "https://flow.polar.com/oauth2/authorization?" +urllib.urlencode(params)
+
+    return HttpResponseRedirect(url)
+
+    
+
 # Runkeeper authorization
 @login_required()
 def rower_runkeeper_authorize(request):
     # Generate a random string for the state parameter
     # Save it for use later to prevent xsrf attacks
-    from uuid import uuid4
+
     state = str(uuid4())
 
     params = {"client_id": RUNKEEPER_CLIENT_ID,
@@ -2408,7 +2430,7 @@ def rower_runkeeper_authorize(request):
 def rower_sporttracks_authorize(request):
     # Generate a random string for the state parameter
     # Save it for use later to prevent xsrf attacks
-    from uuid import uuid4
+
     state = str(uuid4())
 
     params = {"client_id": SPORTTRACKS_CLIENT_ID,
@@ -2426,7 +2448,7 @@ def rower_sporttracks_authorize(request):
 def rower_underarmour_authorize(request):
     # Generate a random string for the state parameter
     # Save it for use later to prevent xsrf attacks
-    from uuid import uuid4
+
     state = str(uuid4())
 
     redirect_uri = UNDERARMOUR_REDIRECT_URI
@@ -2443,7 +2465,7 @@ def rower_underarmour_authorize(request):
 def rower_tp_authorize(request):
     # Generate a random string for the state parameter
     # Save it for use later to prevent xsrf attacks
-    from uuid import uuid4
+
     state = str(uuid4())
     params = {"client_id": TP_CLIENT_KEY,
               "response_type": "code",
@@ -2608,6 +2630,40 @@ def test_reverse_view(request):
 def rower_process_twittercallback(request):
     return "dummy"
 
+# Process Polar Callback
+@login_required()
+def rower_process_polarcallback(request):
+    try:
+        code = request.GET['code']
+    except MultiValueDictKeyError:
+        try:
+            message = request.GET['error']
+        except MultiValueDictKeyError:
+            message = "access error"
+
+        messages.error(request,message)
+        return imports_view(request)
+
+    res = polarstuff.get_token(code)
+
+    access_token = res[0]
+    expires_in = res[1]
+    user_id = res[2]
+
+    expirydatetime = timezone.now()+datetime.timedelta(seconds=expires_in)
+    
+    r = getrower(request.user)
+    r.polartoken = access_token
+    r.polartokenexpirydate = expirydatetime
+    r.polaruserid = user_id
+
+    r.save()
+
+    successmessage = "Tokens stored. Good to go"
+    messages.info(request,successmessage)
+    return imports_view(request)
+
+
 # Process Strava Callback
 @login_required()
 def rower_process_stravacallback(request):