Thank you page + example workout

rowers/dataprep.py

@@ -1,10 +1,18 @@
 from rowers.models import Workout, User, Rower
 from rowingdata import rowingdata as rrdata
 
+from rowers.tasks import handle_sendemail_unrecognized
+
 from rowingdata import rower as rrower
 from rowingdata import main as rmain
 
+from rowingdata import get_file_type
+
 from pandas import DataFrame,Series
+from pytz import timezone as tz,utc
+
+from django.utils.timezone import get_current_timezone
+thetimezone = get_current_timezone()
 
 import pandas as pd
 import numpy as np
@@ -80,6 +88,164 @@ def timedeltaconv(x):
         
     return dt
 
+def new_workout_from_file(r,f2,
+                          workouttype='rower',
+                          title='Workout',
+                          notes=''):
+
+    fileformat = get_file_type(f2)
+    summary = ''
+    # handle non-Painsled
+    if (fileformat != 'csv'):
+	# handle RowPro:
+	if (fileformat == 'rp'):
+	    row = RowProParser(f2)
+	    # handle TCX
+	if (fileformat ==  'tcx'):
+	    row = TCXParser(f2)
+                    
+	# handle Mystery
+	if (fileformat == 'mystery'):
+	    row = MysteryParser(f2)
+
+	# handle TCX no HR
+	if (fileformat == 'tcxnohr'):
+	    row = TCXParserNoHR(f2)
+		    
+	# handle ErgData
+	if (fileformat == 'ergdata'):
+	    row = ErgDataParser(f2)
+
+	# handle BoatCoach
+	if (fileformat == 'boatcoach'):
+	    row = BoatCoachParser(f2)
+
+	# handle painsled desktop
+	if (fileformat == 'painsleddesktop'):
+	    row = painsledDesktopParser(f2)
+
+	# handle speed coach GPS
+	if (fileformat == 'speedcoach'):
+	    row = speedcoachParser(f2)
+
+	# handle speed coach GPS 2 
+	if (fileformat == 'speedcoach2'):
+	    row = SpeedCoach2Parser(f2)
+            summary = row.allstats()
+
+
+	# handle ErgStick
+        if (fileformat == 'ergstick'):
+	    row = ErgStickParser(f2)
+		    
+	# handle FIT
+	if (fileformat == 'fit'):
+	    row = FITParser(f2)
+	    s = fitsummarydata(f2)
+	    s.setsummary()
+	    summary = s.summarytext
+
+
+	    f_to_be_deleted = f2
+	    # should delete file
+	    f2 = f2[:-4]+'o.csv'
+	    row.write_csv(f2,gzip=True)
+
+            #os.remove(f2)
+            try:
+	        os.remove(f_to_be_deleted)
+            except:
+                os.remove(f_to_be_deleted+'.gz')
+
+    # make workout and put in database
+    rr = rrower(hrmax=r.max,hrut2=r.ut2,
+		hrut1=r.ut1,hrat=r.at,
+		hrtr=r.tr,hran=r.an,ftp=r.ftp)
+    row = rdata(f2,rower=rr)
+    if row == 0:
+	return HttpResponse("Error: CSV Data File Not Found")
+
+    # auto smoothing
+    pace = row.df[' Stroke500mPace (sec/500m)'].values
+    velo = 500./pace
+                
+    f = row.df['TimeStamp (sec)'].diff().mean()
+    windowsize = 2*(int(10./(f)))+1
+    if not 'originalvelo' in row.df:
+	row.df['originalvelo'] = velo
+
+    if windowsize > 3 and windowsize<len(velo):
+	velo2 = savgol_filter(velo,windowsize,3)
+    else:
+	velo2 = velo
+
+    velo3 = pd.Series(velo2)
+    velo3 = velo3.replace([-np.inf,np.inf],np.nan)
+    velo3 = velo3.fillna(method='ffill')
+                
+    pace2 = 500./abs(velo3)
+
+    row.df[' Stroke500mPace (sec/500m)'] = pace2
+
+    row.df = row.df.fillna(0)
+
+    row.write_csv(f2,gzip=True)
+    try:
+        os.remove(f2)
+    except:
+        pass
+                
+    # recalculate power data
+    if workouttype == 'rower' or workouttype == 'dynamic' or workouttype == 'slides':
+	try:
+	    row.erg_recalculatepower()
+            row.write_csv(f2,gzip=True)
+	except:
+	    pass
+            
+    if fileformat != 'fit' and summary == '':
+	summary = row.summary()
+	summary += '\n'
+	summary += row.intervalstats_painsled()
+
+    averagehr = row.df[' HRCur (bpm)'].mean()
+    maxhr = row.df[' HRCur (bpm)'].max()
+    
+    totaldist = row.df['cum_dist'].max()
+    totaltime = row.df['TimeStamp (sec)'].max()-row.df['TimeStamp (sec)'].min()
+    totaltime = totaltime+row.df.ix[0,' ElapsedTime (sec)']
+
+    hours = int(totaltime/3600.)
+    minutes = int((totaltime - 3600.*hours)/60.)
+    seconds = int(totaltime - 3600.*hours - 60.*minutes)
+    tenths = int(10*(totaltime - 3600.*hours - 60.*minutes - seconds))
+    duration = "%s:%s:%s.%s" % (hours,minutes,seconds,tenths)
+
+    workoutdate = row.rowdatetime.strftime('%Y-%m-%d')
+    workoutstarttime = row.rowdatetime.strftime('%H:%M:%S')
+    workoutstartdatetime = thetimezone.localize(row.rowdatetime).astimezone(utc)
+
+    # check for duplicate start times
+    ws = Workout.objects.filter(starttime=workoutstarttime,
+				    user=r)
+    if (len(ws) != 0):
+	message =  "Warning: This workout probably already exists in the database"
+
+    w = Workout(user=r,name=title,date=workoutdate,
+		workouttype=workouttype,
+		duration=duration,distance=totaldist,
+		weightcategory=r.weightcategory,
+		starttime=workoutstarttime,
+		csvfilename=f2,notes=notes,summary=summary,
+		maxhr=maxhr,averagehr=averagehr,
+		startdatetime=workoutstartdatetime)
+
+    w.save()
+    # put stroke data in database
+    res = dataprep(row.df,id=w.id,bands=True,barchart=True,otwpower=True,empower=True)
+
+    return True
+
 def compare_data(id):
     row = Workout.objects.get(id=id)
     f1 = row.csvfilename