adding team to workout upon creation

rowers/dataprep.py

@@ -25,6 +25,8 @@ from rowingdata import (
     summarydata,get_file_type,
     )
 
+from rowers.models import Team
+
 import os
 import pandas as pd
 import numpy as np
@@ -320,7 +322,14 @@ def save_workout_database(f2,r,dosmooth=True,workouttype='rower',
 		maxhr=maxhr,averagehr=averagehr,
 		startdatetime=workoutstartdatetime)
 
+
     w.save()
+
+    ts = Team.objects.filter(rower=r)
+
+    for t in ts:
+        w.team.add(t)
+
     # put stroke data in database
     res = dataprep(row.df,id=w.id,bands=True,
                    barchart=True,otwpower=True,empower=True)