diff --git a/rowers/dataprepnodjango.py b/rowers/dataprepnodjango.py index b4a9b6d0..1acccf44 100644 --- a/rowers/dataprepnodjango.py +++ b/rowers/dataprepnodjango.py @@ -35,6 +35,24 @@ database_url = 'mysql://{user}:{password}@{host}:{port}/{database_name}'.format( database_url_debug = 'sqlite:///'+database_name +# mapping the DB column names to the CSV file column names +columndict = { + 'time':'TimeStamp (sec)', + 'hr':' HRCur (bpm)', + 'pace':' Stroke500mPace (sec/500m)', + 'spm':' Cadence (stokes/min)', + 'power':' Power (watts)', + 'averageforce':' AverageDriveForce (lbs)', + 'drivelength':' DriveLength (meters)', + 'peakforce':' PeakDriveForce (lbs)', + 'distance':' Horizontal (meters)', + 'catch':'catch', + 'finish':'finish', + 'peakforceangle':'peakforceangle', + 'wash':'wash', + 'slip':'wash', + 'workoutstate':' WorkoutState', + } from scipy.signal import savgol_filter @@ -563,6 +581,7 @@ def dataprep(rowdatadf,id=0,bands=True,barchart=True,otwpower=True, cumdist = rowdatadf.ix[:,'cum_dist'] power = rowdatadf.ix[:,' Power (watts)'] + averageforce = rowdatadf.ix[:,' AverageDriveForce (lbs)'] drivelength = rowdatadf.ix[:,' DriveLength (meters)'] try: @@ -580,7 +599,7 @@ def dataprep(rowdatadf,id=0,bands=True,barchart=True,otwpower=True, if windowsize <= 3: windowsize = 5 - if windowsize > 3: + if windowsize > 3 and windowsize