Handling empty dataframe

0e60af8c · Akshay G · 451dceb8 · 0e60af8c
Commit 0e60af8c authored Feb 15, 2021 by Akshay G
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 2 deletions

main.py main.py +9 -2

No files found.
--- a/main.py
+++ b/main.py
@@ -84,7 +84,14 @@ def get_data(query):
            [KariosConstants.VALUES_KEY]
        df = DataFrame.from_dict(data)
        logger.debug("Dataframe --> {}".format(df.head()))
-        df.columns = [KariosConstants.TIMESTAMP_COLUMN_NAME, list(config[KariosConstants.TAG_HIERARCHY_KEY].keys())[0]]
+
+        if len(df) > 0:
+            df.columns = [KariosConstants.TIMESTAMP_COLUMN_NAME, list(config[KariosConstants.TAG_HIERARCHY_KEY].keys())
+            [0]]
+        else:
+            logger.warn("The dataframe is empty!")
+            df = DataFrame(columns=[KariosConstants.TIMESTAMP_COLUMN_NAME, list(config[KariosConstants.
+                                                                                TAG_HIERARCHY_KEY].keys())[0]])
        return df
    else:
        raise Exception(response.json()[KariosConstants.ERRORS_KEY])