improve time get all data by chuck category

chuongmep · Mar 6, 2024 · 8dbd627 · 8dbd627
1 parent 24599c1
commit 8dbd627
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/APSToolkitPython/setup.py b/APSToolkitPython/setup.py
@@ -6,7 +6,7 @@
 
 setuptools.setup(
     name="aps-toolkit",
-    version="0.2.8",
+    version="0.2.9",
     author="chuong mep",
     author_email="[email protected]",
     description="A Toolkit Autodesk Platform Services for Python",

diff --git a/APSToolkitPython/src/aps_toolkit/ProDbReaderRevit.py b/APSToolkitPython/src/aps_toolkit/ProDbReaderRevit.py
@@ -33,8 +33,12 @@ def get_all_categories(self) -> dict:
         return categories
 
     def get_all_data(self, is_get_sub_family=False) -> pd.DataFrame:
-        childs = self.get_children(1)
-        dataframe = self._get_recursive_ids(childs,is_get_sub_family)
+        categories_dict = self.get_all_categories()
+        dbids = list(categories_dict.keys())
+        dataframe = pd.DataFrame()
+        for dbid in dbids:
+            df = self._get_recursive_ids([dbid], is_get_sub_family)
+            dataframe = pd.concat([dataframe, df], ignore_index=True)
         return dataframe
     def get_all_families(self) -> dict:
         families = {}