token exchange works

rowers/views.py

@@ -56,6 +56,7 @@ from sporttracksstuff import SportTracksNoTokenError
 from iso8601 import ParseError
 import stravastuff
 import sporttracksstuff
+import underarmourstuff
 import runkeeperstuff
 import ownapistuff
 from ownapistuff import TEST_CLIENT_ID, TEST_CLIENT_SECRET, TEST_REDIRECT_URI
@@ -64,6 +65,8 @@ from rowsandall_app.settings import (
     STRAVA_CLIENT_ID, STRAVA_REDIRECT_URI, STRAVA_CLIENT_SECRET,
     SPORTTRACKS_CLIENT_ID, SPORTTRACKS_REDIRECT_URI,
     SPORTTRACKS_CLIENT_SECRET,
+    UNDERARMOUR_CLIENT_ID, UNDERARMOUR_REDIRECT_URI,
+    UNDERARMOUR_CLIENT_SECRET,UNDERARMOUR_CLIENT_KEY,
     RUNKEEPER_CLIENT_ID,RUNKEEPER_REDIRECT_URI,RUNKEEPER_CLIENT_SECRET,
     )
 
@@ -878,6 +881,20 @@ def sporttracks_open(user):
 
     return thetoken
 
+# Checks if user has UnderArmour token, renews them if they are expired
+def underarmour_open(user):
+    r = Rower.objects.get(user=user)
+    if (r.underarmourtoken == '') or (r.underarmourtoken is None):
+	s = "Token doesn't exist. Need to authorize"
+	raise UnderarmourNoTokenError("User has no token")
+    else:
+	if (timezone.now()>r.underarmourtokenexpirydate):
+	    thetoken = underarmourstuff.rower_underarmour_token_refresh(user)
+	else:
+	    thetoken = r.underarmourtoken
+
+    return thetoken
+
 # Checks if user has SportTracks token, renews them if they are expired
 def runkeeper_open(user):
     r = Rower.objects.get(user=user)
@@ -1373,6 +1390,24 @@ def rower_sporttracks_authorize(request):
 
     return HttpResponseRedirect(url)
 
+# Underarmour Authorization
+@login_required()
+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
+    redirect_uri = 'http://localhost:8000/underarmour_callback'
+    
+    url = 'https://api.mapmyfitness.com/v7.1/oauth2/authorize/?' \
+                'client_id={0}&response_type=code&redirect_uri={1}'.format(
+                    UNDERARMOUR_CLIENT_KEY, redirect_uri
+                )
+
+    return HttpResponseRedirect(url)
+
 # Concept2 token refresh. URL for manual refresh. Not visible to users
 @login_required()
 def rower_c2_token_refresh(request):
@@ -1535,6 +1570,28 @@ def rower_process_sporttrackscallback(request):
     successmessage = "Tokens stored. Good to go"
     return imports_view(request,successmessage=successmessage)
 
+# Process Underarmour callback
+@login_required()
+def rower_process_underarmourcallback(request):
+    code = request.GET['code']
+    res = underarmourstuff.get_token(code)
+
+
+    access_token = res[0]
+    expires_in = res[1]
+    refresh_token = res[2]
+    expirydatetime = timezone.now()+datetime.timedelta(seconds=expires_in)
+    
+    r = Rower.objects.get(user=request.user)
+    r.underarmourtoken = access_token
+    r.underarmourtokenexpirydate = expirydatetime
+    r.underarmourrefreshtoken = refresh_token
+
+    r.save()
+
+    successmessage = "Tokens stored. Good to go"
+    return imports_view(request,successmessage=successmessage)
+
 # Process Own API callback - for API testing purposes
 @login_required()
 def rower_process_testcallback(request):