From 182b6c7b0702b8b4d9dacb3d7837ea776e65d2f1 Mon Sep 17 00:00:00 2001
From: Dion Moult <dion@thinkmoult.com>
Date: Wed, 27 Sep 2023 14:18:48 +1000
Subject: [PATCH] IfcFM now supports writing to Pandas Dataframes

---
 src/ifcfm/ifcfm/__init__.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/ifcfm/ifcfm/__init__.py b/src/ifcfm/ifcfm/__init__.py
index 5b81f02367..c5414a46ac 100644
--- a/src/ifcfm/ifcfm/__init__.py
+++ b/src/ifcfm/ifcfm/__init__.py
@@ -36,6 +36,12 @@ except:
     pass  # No ODF support
 
 
+try:
+    import pandas as pd
+except:
+    pass  # No Pandas support
+
+
 class Parser:
     def __init__(self, preset="basic"):
         self.file = None
@@ -239,3 +245,9 @@ class Writer:
                 r += 1
 
         workbook.save(output)
+
+    def write_pd(self):
+        results = {}
+        for category, data in self.categories.items():
+            results[category] = pd.DataFrame(data["rows"], columns=data["headers"])
+        return results