Strava, ST, RK done

rowers/c2stuff.py

@@ -27,6 +27,7 @@ from rowingdata import rowingdata
 import pandas as pd
 import numpy as np
 from rowers.models import Rower,Workout
+from rowers.models import checkworkoutuser
 import sys
 import urllib
 from requests import Request, Session
@@ -60,13 +61,26 @@ def custom_exception_handler(exc,message):
 
     return res
 
-# Check if workout is owned by this user
-def checkworkoutuser(user,workout):
-    try:
-	r = Rower.objects.get(user=user)
-	return (workout.user == r)
-    except Rower.DoesNotExist:
-	return(False)
+# Checks if user has Concept2 tokens, resets tokens if they are
+# expired.
+def c2_open(user):
+    r = Rower.objects.get(user=user)
+    if (r.c2token == '') or (r.c2token is None):
+	s = "Token doesn't exist. Need to authorize"
+	raise C2NoTokenError("User has no token")
+    else:
+	if (timezone.now()>r.tokenexpirydate):
+	    res = rower_c2_token_refresh(user)
+            if res[0] != None:
+                thetoken = res[0]
+            else:
+                raise C2NoTokenError("User has no token")
+	else:
+	    thetoken = r.c2token
+
+    return thetoken
+
+
 
 # convert datetime object to seconds
 def makeseconds(t):
@@ -249,9 +263,13 @@ def createc2workoutdata(w):
         row = rowingdata(filename)
     except IOError:
         return 0
-    
-    averagehr = int(row.df[' HRCur (bpm)'].mean())
-    maxhr = int(row.df[' HRCur (bpm)'].max())
+
+    try:
+        averagehr = int(row.df[' HRCur (bpm)'].mean())
+        maxhr = int(row.df[' HRCur (bpm)'].max())
+    except ValueError:
+        averagehr = 0
+        maxhr = 0
 
     # adding diff, trying to see if this is valid
     t = 10*row.df.ix[:,'TimeStamp (sec)'].values-10*row.df.ix[0,'TimeStamp (sec)']
@@ -265,7 +283,10 @@ def createc2workoutdata(w):
     p = p.astype(int)
     spm = row.df[' Cadence (stokes/min)'].astype(int)
     spm[0] = spm[1]
-    hr = row.df[' HRCur (bpm)'].astype(int)
+    try:
+        hr = row.df[' HRCur (bpm)'].astype(int)
+    except ValueError:
+        hr = 0*d
     stroke_data = []
     for i in range(len(t)):
 	thisrecord = {"t":t[i],"d":d[i],"p":p[i],"spm":spm[i],"hr":hr[i]}
@@ -521,34 +542,49 @@ def process_callback(request):
 # Uploading workout
 def workout_c2_upload(user,w):
     response = 'trying C2 upload'
-    r = Rower.objects.get(user=user)
-    if (r.c2token == '') or (r.c2token is None):
-	s = "Token doesn't exist. Need to authorize"
-	return custom_exception_handler(401,s)
-    elif (timezone.now()>r.tokenexpirydate):
-	s = "Token expired. Needs to refresh."
-	return custom_exception_handler(401,s)
-    else:
-	# ready to upload. Hurray
-	if (checkworkoutuser(user,w)):
-	    c2userid = get_userid(r.c2token)
-	    data = createc2workoutdata(w)
-	    authorizationstring = str('Bearer ' + r.c2token)
-	    headers = {'Authorization': authorizationstring,
-	       'user-agent': 'sanderroosendaal',
-	       'Content-Type': 'application/json'}
-	    import urllib
-	    url = "https://log.concept2.com/api/users/%s/results" % (c2userid)
-	    response = requests.post(url,headers=headers,data=json.dumps(data))
-	    if (response.status_code == 201):
-		s= json.loads(response.text)
-		c2id = s['data']['id']
-		w.uploadedtoc2 = c2id
-		w.save()
-	else:
-	    response = "You are not authorized to upload this workout"
+    thetoken = c2_open(user)
 
-    return response
+    r = Rower.objects.get(user=user)
+
+    # ready to upload. Hurray
+    if (checkworkoutuser(user,w)):
+	c2userid = get_userid(r.c2token)
+        if not c2userid:
+            raise C2NoTokenError
+
+	data = createc2workoutdata(w)
+        if data == 0:
+            return "Error: No data file. Contact info@rowsandall.com if the problem persists",0
+            
+	authorizationstring = str('Bearer ' + r.c2token)
+	headers = {'Authorization': authorizationstring,
+	           'user-agent': 'sanderroosendaal',
+	           'Content-Type': 'application/json'}
+	import urllib
+	url = "https://log.concept2.com/api/users/%s/results" % (c2userid)
+	response = requests.post(url,headers=headers,data=json.dumps(data))
+
+        if (response.status_code == 409 ):
+	    message = "Duplicate error"
+	    w.uploadedtoc2 = -1
+            c2id = -1
+	    w.save()
+        elif (response.status_code == 201 or response.status_code == 200):
+            try:
+	        s= json.loads(response.text)
+	        c2id = s['data']['id']
+	        w.uploadedtoc2 = c2id
+	        w.save()
+                message = ""
+            except:
+                message = "Something went wrong in workout_c2_upload_view. Response code 200/201 but C2 sync failed: "+response.text
+                c2id = 0
+
+	else:
+	    message = "You are not authorized to upload this workout"
+            c2id = 0
+
+    return message,c2id
 
 # This is token refresh. Looks for tokens in our database, then refreshes
 def rower_c2_token_refresh(user):