C2 and ST button to import all new workouts

rowers/views.py

@@ -267,7 +267,7 @@ def splitstdata(lijst):
 
     return [np.array(t),np.array(latlong)]
 	
-from utils import geo_distance,serialize_list,deserialize_list
+from utils import geo_distance,serialize_list,deserialize_list,uniqify
 from rowers.models import checkworkoutuser
 
 # Check if a user is a Coach member
@@ -5764,15 +5764,25 @@ def workout_sporttracksimport_view(request,message=""):
 	    return HttpResponseRedirect(url)
     else:
 	workouts = []
+        r = Rower.objects.get(user=request.user)
+        stids = [int(getidfromsturi(item['uri'])) for item in res.json()['items']]
+        knownstids = uniqify([
+            w.uploadedtosporttracks for w in Workout.objects.filter(user=r)
+                      ])
+        newids = [stid for stid in stids if not stid in knownstids]
 	for item in res.json()['items']:
 	    d = int(float(item['total_distance']))
-	    i = getidfromsturi(item['uri'])
+	    i = int(getidfromsturi(item['uri']))
+            if i in knownstids:
+                nnn = ''
+            else:
+                nnn = 'NEW'
 	    n = item['name']
 	    ttot = str(datetime.timedelta(seconds=int(float(item['duration']))))
 	    s = item['start_time']
 	    r = item['type']
-	    keys = ['id','distance','duration','starttime','type','name']
-	    values = [i,d,ttot,s,r,n]
+	    keys = ['id','distance','duration','starttime','type','name','new']
+	    values = [i,d,ttot,s,r,n,nnn]
 	    res = dict(zip(keys,values))
 	    workouts.append(res)
 	return render(request,'sporttracks_list_import.html',
@@ -5823,7 +5833,58 @@ def c2listdebug_view(request,message=""):
 		      {'workouts':workouts,
                        'teams':get_my_teams(request.user),
                       })
-		      
+
+# Import all unknown workouts available on Concept2 logbook
+@login_required()
+def workout_getc2workout_all(request,message=""):
+    try:
+	thetoken = c2_open(request.user)
+    except C2NoTokenError:
+	return HttpResponseRedirect("/rowers/me/c2authorize/")
+
+    res = c2stuff.get_c2_workout_list(request.user)
+
+    if (res.status_code != 200):
+	message = "Something went wrong in workout_c2import_view (C2 token refresh)"
+        messages.error(request,message)
+    else:
+        r = Rower.objects.get(user=request.user)
+        c2ids = [item['id'] for item in res.json()['data']]
+        knownc2ids = uniqify([
+            w.uploadedtoc2 for w in Workout.objects.filter(user=r)
+        ])
+        newids = [c2id for c2id in c2ids if not c2id in knownc2ids]
+        for c2id in newids:
+            res = c2stuff.get_c2_workout(request.user,c2id)
+            if (res.status_code == 200):
+	        data = res.json()['data']
+                splitdata = None
+                if 'workout' in data:
+                    if 'splits' in data['workout']:
+                        splitdata = data['workout']['splits']
+                    if 'intervals' in data['workout']:
+                        splitdata = data['workout']['intervals']
+
+                # Check if workout has stroke data, and get the stroke data
+                if data['stroke_data']:
+	            res2 = c2stuff.get_c2_workout_strokes(request.user,c2id)
+                    # We have stroke data
+	            if res2.status_code == 200:
+	                strokedata = pd.DataFrame.from_dict(res2.json()['data'])
+                        # create the workout
+	                id,message = add_workout_from_strokedata(
+                            request.user,c2id,data,strokedata,
+			    source='c2')
+	                w = Workout.objects.get(id=id)
+	                w.uploadedtoc2=c2id
+                        w.save()
+                        if message:
+                            messages.error(request,message)
+
+    url = reverse(workouts_view)
+    return HttpResponseRedirect(url)
+            
+    
 # List of workouts available on Concept2 logbook - for import
 @login_required()
 def workout_c2import_view(request,message=""):
@@ -5844,6 +5905,12 @@ def workout_c2import_view(request,message=""):
 	    return HttpResponseRedirect(url)
     else:
 	workouts = []
+        r = Rower.objects.get(user=request.user)
+        c2ids = [item['id'] for item in res.json()['data']]
+        knownc2ids = uniqify([
+            w.uploadedtoc2 for w in Workout.objects.filter(user=r)
+                      ])
+        newids = [c2id for c2id in c2ids if not c2id in knownc2ids]
 	for item in res.json()['data']:
 	    d = item['distance']
 	    i = item['id']
@@ -5852,8 +5919,12 @@ def workout_c2import_view(request,message=""):
 	    r = item['type']
 	    s2 = item['source']
 	    c = item['comments']
-	    keys = ['id','distance','duration','starttime','rowtype','source','comment']
-	    values = [i,d,ttot,s,r,s2,c]
+            if i in knownc2ids:
+                nnn = ''
+            else:
+                nnn = 'NEW'
+	    keys = ['id','distance','duration','starttime','rowtype','source','comment','new']
+	    values = [i,d,ttot,s,r,s2,c,nnn]
 	    res = dict(zip(keys,values))
 	    workouts.append(res)
 
@@ -5959,6 +6030,38 @@ def workout_getsporttracksworkout_view(request,sporttracksid):
                   })
     return HttpResponseRedirect(url)
 
+# Imports all new workouts from SportTracks
+@login_required()
+def workout_getsporttracksworkout_all(request):
+    res = sporttracksstuff.get_sporttracks_workout_list(request.user)
+    if (res.status_code == 200):
+        r = Rower.objects.get(user=request.user)
+        stids = [int(getidfromsturi(item['uri'])) for item in res.json()['items']]
+        knownstids = uniqify([
+            w.uploadedtosporttracks for w in Workout.objects.filter(user=r)
+                      ])
+        newids = [stid for stid in stids if not stid in knownstids]
+        for sporttracksid in newids:
+            res = sporttracksstuff.get_sporttracks_workout(
+                request.user,sporttracksid)
+            data = res.json()
+
+            id,message = add_workout_from_stdata(request.user,sporttracksid,data)
+            print sporttracksid,id
+            if id==0:
+                messages.error(request,message)
+
+            else:
+                w = Workout.objects.get(id=id)
+                w.uploadedtosporttracks=sporttracksid
+                print 'saved'
+                w.save()
+
+    url = reverse(workouts_view)
+    return HttpResponseRedirect(url)
+
+
+
 # Imports a workout from Concept2 
 @login_required()
 def workout_getc2workout_view(request,c2id):