adding update wps to upload routine

rowers/tasks.py

@@ -2797,6 +2797,33 @@ def add2(x, y,debug=False,**kwargs): # pragma: no cover
 
 graphql_url = "https://rp3rowing-app.com/graphql"
 
+@app.task
+def handle_update_wps(rid,types,ids,mode,debug=False,**kwargs):
+    df = getsmallrowdata_db(['time','driveenergy'],ids=ids)
+    try:
+        mask = df['driveenergy'] > 100
+    except KeyError:
+        return 0
+    wps_median = int(df.loc[mask,'driveenergy'].median())
+
+    if mode == 'water':
+        query = "UPDATE `rowers_rower` SET `median_wps` = '%s' WHERE `id` = '%s'" % (wps_median,rid)
+    else:
+        query = "UPDATE `rowers_rower` SET `median_wps_erg` = '%s' WHERE `id` = '%s'" % (wps_median,rid)
+
+    if debug: # pragma: no cover
+        engine = create_engine(database_url_debug, echo=False)
+    else:
+        engine = create_engine(database_url, echo=False)
+
+    with engine.connect() as conn, conn.begin():
+        result = conn.execute(query)
+
+    conn.close()
+    engine.dispose()
+
+    return wps_median
+
 @app.task
 def handle_rp3_async_workout(userid,rp3token,rp3id,startdatetime,max_attempts,debug=False,**kwargs):
     headers = {'Authorization': 'Bearer ' + rp3token }