From d0ecaf1ced7d7fe88a0cd571e3f9a7d7ca33170f Mon Sep 17 00:00:00 2001
From: Nicolas Schmid <nicolas@breiten.ch>
Date: Sat, 27 Apr 2024 15:35:49 +0200
Subject: [PATCH] fix: check if dataframe empty before dropping column
---
hydws/routers/boreholes.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/hydws/routers/boreholes.py b/hydws/routers/boreholes.py
index e08a048..ac12623 100644
--- a/hydws/routers/boreholes.py
+++ b/hydws/routers/boreholes.py
@@ -167,7 +167,10 @@ async def get_section_hydraulics(borehole_id: uuid.UUID,
db_result_df = await crud.read_hydraulics_df(
section_oid, starttime, endtime, defer_cols)
- db_result_df = db_result_df.dropna(axis=1, how='all').drop(columns=['_oid'])
+ db_result_df = db_result_df.dropna(axis=1, how='all')
+
+ if '_oid' in db_result_df.columns:
+ db_result_df = db_result_df.drop(columns=['_oid'])
if format == 'csv':
return csv_response(db_result_df)
--
GitLab