demo version new strava import

rowers/management/commands/processemail.py

@@ -307,9 +307,10 @@ class Command(BaseCommand):
                 message.delete()
 
         # Strava
-        #rowers = Rower.objects.filter(strava_auto_import=True)
+        rowers = Rower.objects.filter(strava_auto_import=True)
-        #for r in rowers:
+        for r in rowers:
-        #    stravastuff.get_strava_workouts(r)
+            if user_is_not_basic(r.user):
+                stravastuff.get_strava_workouts(r)
 
         self.stdout.write(self.style.SUCCESS(
             'Successfully processed email attachments'))

rowers/templates/strava_list_import.html

@@ -18,6 +18,10 @@
   </li>
 
   <li class="grid_4">
+    <form enctype="multipart/form-data" method="post">
+      {% csrf_token %}
+        <input name='workouts' type="submit" value="Import selected workouts">
+        <a href="/rowers/workouts/stravaimport/?selectallnew=true">Select All New</a>
     <table width="70%" class="listtable">
       <thead>
         <tr>
@@ -34,7 +38,8 @@
         {% for workout in workouts %}
         <tr>
           <td>
-            <a href="/rowers/workout/stravaimport/{{ workout|lookup:'id' }}/async/">Import</a></td>
+            <input type="checkbox" value={{ workout|lookup:'id' }} name="workoutid">
+          </td>
           <td>{{ workout|lookup:'name' }}</td>
           <td>{{ workout|lookup:'starttime' }}</td>
           <td>{{ workout|lookup:'duration' }} </td>
@@ -46,6 +51,7 @@
         {% endfor %}
       </tbody>
     </table>
+    </form>
   </li>
 </ul>
 {% else %}

rowers/views/importviews.py

@@ -5,6 +5,7 @@ from __future__ import unicode_literals
 
 from rowers.views.statements import *
 from rowers.plannedsessions import get_dates_timeperiod
+from rowers.tasks import fetch_strava_workout
 
 import numpy
 
@@ -923,6 +924,7 @@ def workout_rp3import_view(request,userid=0):
 @permission_required('rower.is_coach',fn=get_user_by_userid,raise_exception=True)
 @permission_required('rower.is_not_freecoach',fn=get_user_by_userid, raise_exception=True)
 def workout_stravaimport_view(request,message="",userid=0):
+
     r = getrequestrower(request,userid=userid)
     if r.user != request.user:
         messages.error(request,'You can only access your own workouts on the NK Logbook, not those of your athletes')
@@ -939,6 +941,7 @@ def workout_stravaimport_view(request,message="",userid=0):
     res = stravastuff.get_strava_workout_list(request.user)
 
 
+
     if (res.status_code != 200): # pragma: no cover
         if (res.status_code == 401):
             r = getrower(request.user)
@@ -952,6 +955,7 @@ def workout_stravaimport_view(request,message="",userid=0):
     else:
         workouts = []
         r = getrower(request.user)
+        rower = r
         stravaids = [int(item['id']) for item in res.json()]
         stravadata = [{
             'id':int(item['id']),
@@ -995,8 +999,31 @@ def workout_stravaimport_view(request,message="",userid=0):
             r = item['type']
             keys = ['id','distance','duration','starttime','type','name','new']
             values = [i,d,ttot,s,r,n,nnn]
-            res = dict(zip(keys,values))
+            res2 = dict(zip(keys,values))
-            workouts.append(res)
+            workouts.append(res2)
+
+        if request.method == "POST":
+            try:
+                tdict = dict(request.POST.lists())
+                ids = tdict['workoutid']
+                stravaids = [int(id) for id in ids]
+                alldata = {}
+                for item in res.json():
+                    alldata[item['id']] = item
+                for stravaid in stravaids:
+                    csvfilename = 'media/{code}_{stravaid}.csv'.format(code=uuid4().hex[:16],stravaid=stravaid)
+                    result = myqueue(
+                        queue,
+                        fetch_strava_workout,
+                        rower.stravatoken,
+                        stravastuff.oauth_data,
+                        stravaid,
+                        csvfilename,
+                        rower.user.id
+                    )
+                    messages.info(request,'Strava workout nr {stravaid} will be imported in the background'.format(stravaid=stravaid))
+            except KeyError:
+                pass
 
         breadcrumbs = [
             {
@@ -1014,7 +1041,7 @@ def workout_stravaimport_view(request,message="",userid=0):
 
         return render(request,'strava_list_import.html',
                       {'workouts':workouts,
-                       'rower':r,
+                       'rower':rower,
                        'active':'nav-workouts',
                        'breadcrumbs':breadcrumbs,
                        'teams':get_my_teams(request.user),

rowers/views/workoutviews.py

@@ -3502,7 +3502,10 @@ def workout_stats_view(request,id=0,message="",successmessage=""):
         hrdrift = ((pwr1/hr1)-(pwr2/hr2))/(pwr1/hr1)
         hrdrift *= 100.
         if not np.isnan(hrdrift):
-            hrdrift = int(100*hrdrift)/100.
+            try:
+                hrdrift = int(100*hrdrift)/100.
+            except:
+                hrdrift = 0
             otherstats['hrdrift'] = {
                 'verbose_name': 'Heart Rate Drift',
                 'value': hrdrift,