UA done

rowers/underarmourstuff.py

@@ -28,7 +28,7 @@ from django.contrib.auth.decorators import login_required
 # from .models import Profile
 from rowingdata import rowingdata
 import pandas as pd
-from rowers.models import Rower,Workout
+from rowers.models import Rower,Workout,checkworkoutuser
 
 from rowsandall_app.settings import (
     C2_CLIENT_ID, C2_REDIRECT_URI, C2_CLIENT_SECRET,
@@ -88,6 +88,20 @@ def custom_exception_handler(exc,message):
 
     return res
 
+# 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
+
 # Refresh ST token using refresh token
 def do_refresh_token(refreshtoken,access_token):
     client_auth = requests.auth.HTTPBasicAuth(UNDERARMOUR_CLIENT_KEY, UNDERARMOUR_CLIENT_SECRET)
@@ -385,3 +399,55 @@ def get_userid(access_token):
         res = 0
         
     return res
+
+def workout_ua_upload(user,w):
+    message = ""
+    uaid = 0
+
+    r = w.user
+
+    thetoken = underarmour_open(r.user)
+
+    # ready to upload. Hurray
+
+    if (checkworkoutuser(user,w)):
+	data = createunderarmourworkoutdata(w)
+#        return HttpResponse(json.dumps(data))
+        if not data:
+            message = "Data error"
+            uaid = 0
+            return message, uaid
+        
+	authorizationstring = str('Bearer ' + thetoken)
+	headers = {'Authorization': authorizationstring,
+                   'Api-Key': UNDERARMOUR_CLIENT_KEY,
+		   'user-agent': 'sanderroosendaal',
+		   'Content-Type': 'application/json',
+                   }
+
+	url = "https://api.ua.com/v7.1/workout/"
+	response = requests.post(url,headers=headers,data=json.dumps(data))
+
+	# check for duplicate error first
+	if (response.status_code == 409 ):
+	    message = "Duplicate error"
+	    w.uploadedtounderarmour = -1
+            uaid = -1
+	    w.save()
+	elif (response.status_code == 201 or response.status_code==200):
+	    uaid = getidfromresponse(response)
+	    w.uploadedtounderarmour = uaid            
+	    w.save()
+            return '',uaid
+	else:
+	    s = response
+	    message = "Something went wrong in workout_underarmour_upload_view: %s - %s" % (s.reason,s.text)
+            uaid = 0
+            return message, uaid
+		
+    else:
+	message = "You are not authorized to upload this workout"
+        uaid = 0
+        return message, uaid
+
+    return message, uaid