From f64ce1a627585da97bef8c50c02943e0657f0127 Mon Sep 17 00:00:00 2001
From: Klaas <kliffen@astron.nl>
Date: Fri, 24 Dec 2021 11:19:19 +0100
Subject: [PATCH] Check items before concatenating
---
shopping_client/shopping_client.py | 26 +++++++++++++++-----------
1 file changed, 15 insertions(+), 11 deletions(-)
diff --git a/shopping_client/shopping_client.py b/shopping_client/shopping_client.py
index ba643e0..b5c17f6 100644
--- a/shopping_client/shopping_client.py
+++ b/shopping_client/shopping_client.py
@@ -136,21 +136,25 @@ class shopping_client:
def _basket_to_pandas(self):
if len(self.connectors):
- converted_basket = {
- connector.name: pd.concat(
- [
- connector.basket_item_to_pandas(item)
- for item in self.basket
- if connector.validate_basket_item(item)
- ],
- axis=1,
- )
- for connector in self.connectors
- }
+
+ converted_basket = {}
+
+ for connector in self.connectors:
+
+ items = [
+ connector.basket_item_to_pandas(item)
+ for item in self.basket
+ if connector.validate_basket_item(item)
+ ]
+
+ if len(items):
+ converted_basket[connector.name] = pd.concat(items, axis=1)
+
return {
name: data.to_frame().T if data.ndim < 2 else data.T
for name, data in converted_basket.items()
}
+
warn(
"No archive connectors specified - could not convert any basket items to Pandas DataFrame"
)
--
GitLab