From 0af56b6e2f6a45ee6a95504cc456dd98804ad4d5 Mon Sep 17 00:00:00 2001
From: Sander Roosendaal <roosendaalsander@gmail.com>
Date: Wed, 23 Aug 2017 10:57:27 +0200
Subject: [PATCH] added OTE split

---
 rowers/dataprep.py                 | 24 ++++++++++++++++--------
 rowers/templates/advancededit.html | 16 ++++++++++++++--
 rowers/templates/advancedotw.html  |  2 +-
 3 files changed, 31 insertions(+), 11 deletions(-)

diff --git a/rowers/dataprep.py b/rowers/dataprep.py
index d495185a..e2e040d2 100644
--- a/rowers/dataprep.py
+++ b/rowers/dataprep.py
@@ -91,7 +91,7 @@ columndict = {
     'finish':'finish',
     'peakforceangle':'peakforceangle',
     'wash':'wash',
-    'slip':'wash',
+    'slip':'slip',
     'workoutstate':' WorkoutState',
     }
     
@@ -107,13 +107,17 @@ def get_latlon(id):
         
     rowdata = rdata(w.csvfilename)
     try:
-	latitude = rowdata.df.ix[:,' latitude']
-        longitude = rowdata.df.ix[:,' longitude']
-    except KeyError:
-        latitude = 0*rowdata.df.ix[:,'TimeStamp (sec)']
-        latitude = 0*rowdata.df.ix[:,'TimeStamp (sec)']
-
-    return [latitude,longitude]
+        try:
+	    latitude = rowdata.df.ix[:,' latitude']
+            longitude = rowdata.df.ix[:,' longitude']
+        except KeyError:
+            latitude = 0*rowdata.df.ix[:,'TimeStamp (sec)']
+            longitude = 0*rowdata.df.ix[:,'TimeStamp (sec)']
+        return [latitude,longitude]
+    except AttributeError:
+        return [pd.Series([]),pd.Series([])]
+        
+    return [pd.Series([]),pd.Series([])]
 
 def get_workouts(ids,userid):
     goodids = []
@@ -822,6 +826,10 @@ def new_workout_from_file(r,f2,
 
 def split_workout(r,parent,splitsecond,splitmode):
     data,row = getrowdata_db(id=parent.id)
+    latitude,longitude = get_latlon(parent.id)
+    if not latitude.empty and not longitude.empty:
+        data[' latitude'] = latitude
+        data[' longitude'] = longitude
 
     data['time'] = data['time']/1000.
     
diff --git a/rowers/templates/advancededit.html b/rowers/templates/advancededit.html
index 6db5b03c..55de324b 100644
--- a/rowers/templates/advancededit.html
+++ b/rowers/templates/advancededit.html
@@ -133,18 +133,30 @@
       Plot the Power Histogram of this workout
     </p>
   </div>
-  <div class="grid_2 suffix_2 omega">
+  <div class="grid_2">
     <p>
     {% if user.rower.rowerplan == 'pro' or user.rower.rowerplan == 'coach' %}
     <a class="button blue small" href="/rowers/workout/fusion/{{ workout.id }}/">Sensor Fusion</a>
     {% else %}
-      <a class="button blue small" href="/rowers/promembership">Dist Metrics Plot</a>
+      <a class="button blue small" href="/rowers/promembership">Sensor Fusion</a>
       {% endif %}
     </p>
     <p>
       Merge data from another source into this workout
     </p>
   </div>
+  <div class="grid_2 omega">
+    <p>
+    {% if user.rower.rowerplan == 'pro' or user.rower.rowerplan == 'coach' %}
+    <a class="button blue small" href="/rowers/workout/{{ workout.id }}/split">Split Workout</a>
+    {% else %}
+      <a class="button blue small" href="/rowers/promembership">Split Workout</a>
+      {% endif %}
+    </p>
+    <p>
+      Split workout into two seperate workouts
+    </p>
+  </div>
 </div>
 </div>
 <div id="advancedplots" class="grid_6 omega">
diff --git a/rowers/templates/advancedotw.html b/rowers/templates/advancedotw.html
index abbccd67..3e3a2e39 100644
--- a/rowers/templates/advancedotw.html
+++ b/rowers/templates/advancedotw.html
@@ -195,7 +195,7 @@
     {% if user.rower.rowerplan == 'pro' or user.rower.rowerplan == 'coach' %}
     <a class="button blue small" href="/rowers/workout/fusion/{{ workout.id }}/">Sensor Fusion</a>
     {% else %}
-      <a class="button blue small" href="/rowers/promembership">Split Workout</a>
+      <a class="button blue small" href="/rowers/promembership">Sensor Fusion</a>
       {% endif %}
     </p>
     <p>