updatefitnessmetric as middleware

rowers/tasks.py

@@ -25,13 +25,14 @@ from rowsandall_app.settings import PROGRESS_CACHE_SECRET
 import pandas as pd
 
 from django_rq import job
+from django.utils import timezone
 
 from utils import deserialize_list
 
 from rowers.dataprepnodjango import (
     update_strokedata, new_workout_from_file,
     getsmallrowdata_db, updatecpdata_sql,
-    update_agegroup_db,
+    update_agegroup_db,fitnessmetric_to_sql
 )
 
 from django.core.mail import send_mail, EmailMessage
@@ -612,6 +613,82 @@ def handle_updateergcp(rower_id,workoutfilenames,debug=False,**kwargs):
 
     return 1
 
+@app.task
+def handle_updatefitnessmetric(user_id,mode,workoutids,debug=False,
+                               **kwargs):
+
+    columns = ['power','workoutid','time']
+    df = getsmallrowdata_db(columns,ids=workoutids,debug=debug)
+    df.dropna(inplace=True,axis=0)
+
+    if df.empty:
+        # change this
+        return 2
+
+    # df is not empty. We continue
+    dfgrouped = df.groupby(['workoutid'])
+    maxt = 1.05*df['time'].max()/1000.
+
+    logarr = datautils.getlogarr(maxt)
+        
+    delta,cpvalue,avgpower = datautils.getcp(dfgrouped,logarr)
+
+    powerdf = pd.DataFrame({
+        'Delta':delta,
+        'CP':cpvalue,
+        })
+
+    powerdf = powerdf[powerdf['CP']>0]
+    powerdf.dropna(axis=0,inplace=True)
+    powerdf.sort_values(['Delta','CP'],ascending=[1,0],inplace=True)
+    powerdf.drop_duplicates(subset='Delta',keep='first',inplace=True)
+
+    p1,fitt,fitpower,ratio = datautils.cpfit(powerdf)
+    # This is code duplication from datautils -- correct asap
+    fitfunc = lambda pars,x: abs(pars[0])/(1+(x/abs(pars[2]))) + abs(pars[1])/(1+(x/abs(pars[3])))
+
+    powerfourmin = fitfunc(p1,240.)
+    powerhour = fitfunc(p1,3600.)
+
+    # 2k power
+    velofourmin = (powerfourmin/2.8)**(1./3.)
+    dfourmin = 240.*velofourmin
+    dratio = 2000./dfourmin
+    pacefourmin = 500./velofourmin
+
+    # assume 5 sec per doubling drop
+    pace2k = pacefourmin + 5.*np.log10(dratio)/np.log10(2.)
+    velo2k = 500./pace2k
+    t2k = 2000./velo2k
+    pwr2k = fitfunc(p1,t2k)
+    velo2 = (pwr2k/2.8)**(1./3.)
+    if np.isnan(velo2) or velo2 <= 0:
+        velo2 = 1.0
+
+    t2 = 2000./velo2
+
+    pwr2k = fitfunc(p1,t2)
+
+    velo3 = (pwr2k/2.8)**(1./3.)
+
+    t3 = 2000./velo3
+
+    power2k = fitfunc(p1,t3)
+
+    
+    mdict = {
+        'user_id': user_id,
+        'PowerFourMin': powerfourmin,
+        'PowerTwoK': power2k,
+        'PowerOneHour': powerhour,
+        'workoutmode': mode,
+        'last_workout': max(workoutids),
+        'date': timezone.now().strftime('%Y-%m-%d'),
+             }
+
+    result = fitnessmetric_to_sql(mdict,debug=debug)
+
+    return result
 
 @app.task
 def handle_updatecp(rower_id,workoutids,debug=False,table='cpdata',**kwargs):