diff --git a/rowers/dataroutines.py b/rowers/dataroutines.py index ac60f398..976342c0 100644 --- a/rowers/dataroutines.py +++ b/rowers/dataroutines.py @@ -642,6 +642,8 @@ def clean_df_stats(datadf, workstrokesonly=True, ignorehr=True, pass after = {} + + return datadf for workoutid in data_orig['workoutid'].unique(): after[workoutid] = len( datadf[datadf['workoutid'] == workoutid].dropna()) @@ -1273,7 +1275,6 @@ def getsmallrowdata_db(columns, ids=[], doclean=True, workstrokesonly=True, comp data.append(df) except (OSError, ArrowInvalid, IndexError): pass - df['workoutid'] = id try: df = pd.concat(data, axis=0) except ValueError: # pragma: no cover @@ -1313,7 +1314,7 @@ def getsmallrowdata_db(columns, ids=[], doclean=True, workstrokesonly=True, comp data = clean_df_stats(data, ignorehr=True, workstrokesonly=workstrokesonly) data.dropna(axis=1, how='all', inplace=True) - data.dropna(axis=0, how='any', inplace=True) + data.dropna(axis=0, how='all', inplace=True) return data except TypeError: pass diff --git a/rowers/interactiveplots.py b/rowers/interactiveplots.py index 57a709b4..e1745a17 100644 --- a/rowers/interactiveplots.py +++ b/rowers/interactiveplots.py @@ -95,7 +95,7 @@ def get_chart(end_point, chart_data, debug=False): headers = {'authorization':"Bearer {token}".format(token=settings.ROWSANDALL_CHARTS_TOKEN)} try: response = requests.post(url, json=chart_data, headers=headers) - except Error as err: + except Exception as err: if debug: print("Chart Server Error") print(err) @@ -5276,8 +5276,14 @@ def interactive_multiple_compare_chart(ids, xparam, yparam, plottype='line', columns = [name for name, d in metrics.rowingmetrics] columns_basic = [name for name, d in metrics.rowingmetrics if d['group'] == 'basic'] - columns = columns + ['spm', 'driveenergy', 'distance'] - columns_basic = columns_basic + ['spm', 'driveenergy', 'distance'] + add_columns = [ + 'ftime', 'distance', 'fpace', + 'power', 'hr', 'spm', + 'time', 'pace', 'workoutstate', + 'workoutid' + ] + columns = columns + add_columns + columns_basic = columns_basic + add_columns compute = False doclean = False @@ -5296,8 +5302,9 @@ def interactive_multiple_compare_chart(ids, xparam, yparam, plottype='line', compute=compute, workstrokesonly=workstrokesonly) + datadf['workoutid'] = datadf['workoutid'].astype(int) datadf.dropna(axis=1, how='all', inplace=True) - datadf.dropna(axis=0, how='any', inplace=True) + datadf.dropna(axis=0, how='all', inplace=True) nrworkouts = len(ids) @@ -5317,8 +5324,9 @@ def interactive_multiple_compare_chart(ids, xparam, yparam, plottype='line', if (xparam == 'time'): datadf[xparam] = datadf[xparam] - datadf[xparam].iloc[0] - data_dict = datadf.to_dict("records") + datadf = datadf.fillna(0) + data_dict = datadf.to_dict("records") metrics_list = [{'name': name, 'rowingmetrics':d } for name, d in metrics.rowingmetrics]