tasks

rowers/tasks.py

@@ -2553,9 +2553,10 @@ def handle_otwsetpower(self, f1, boattype, boatclass, coastalbrand, weightvalue,
         totaltime = totaltime + rowdata.df.loc[0, ' ElapsedTime (sec)']
     except KeyError:  # pragma: no cover
         pass
-    df = getsmallrowdata_db(
+    df = getsmallrowdata_pd(
         ['power', 'workoutid', 'time'], ids=[workoutid],
         debug=debug)
+
     thesecs = totaltime
     maxt = 1.05 * thesecs
     logarr = datautils.getlogarr(maxt)
@@ -3171,23 +3172,22 @@ 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)
+    df = read_data(['time', 'driveenergy'], ids=ids)
     try:
-        mask = df['driveenergy'] > 100
-    except (KeyError, TypeError):  # pragma: no cover
-        return 0
-    try:
-        wps_median = int(df.loc[mask, 'driveenergy'].median())
+        wps_median = int(df.filter(pl.col("driveenergy")>100)["driveenergy"].median())
+        rower = Rower.objects.get(id=rid)
+        if mode == 'water':
+            rower.median_wps = wps_median
+        else:  # pragma: no cover
+            rower.median_wps_erg = wps_median
+
+        rower.save()
     except ValueError:  # pragma: no cover
-        return 0
-
-    rower = Rower.objects.get(id=rid)
-    if mode == 'water':
-        rower.median_wps = wps_median
-    else:
-        rower.median_wps_erg = wps_median
-
-    rower.save()
+        wps_median = 0
+    except OverflowError:
+        wps_median = 0
+    except ColumnNotFoundError:
+        wps_median = 0
 
     return wps_median