add exeption handling when no milestone

maxleow · May 18, 2022 · a73851a · a73851a
1 parent e2c8655
commit a73851a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="testrail-data",
-    version="0.0.9",
+    version="0.0.10",
     install_requires=[
         "pandas",
         "testrail-api>=1.10",

diff --git a/testrail_data/_category.py b/testrail_data/_category.py
@@ -172,12 +172,12 @@ def get_runs(*mile_ids):
                 if df_run2.shape[0] > 0:
                     dataframes.append(df_run2)
             return dataframes
-
         dfs = get_runs(milestone_ids)
         milestone = Milestones(self._session)
         df_milestone = milestone.sub_milestones_to_dataframe(*milestone_ids).filter(['id'])
-        dfs.extend(get_runs(*df_milestone['id'].to_list()))
-        return pd.concat(dfs).reset_index(drop=False)
+        if df_milestone.shape[0] > 0:
+            dfs.extend(get_runs(*df_milestone['id'].to_list()))
+        return pd.concat(dfs).reset_index(drop=False) if dfs else pd.DataFrame()
 
     def get_runs_by_plan(self, *plan_ids: int) -> list:
         """