Update tools.py

* Improve `ControllerFeatures` class
jzsmoreno · Sep 28, 2023 · a9e4823 · a9e4823
1 parent bf6b8e9
commit a9e4823
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 19 deletions.
diff --git a/pydbsmgr/utils/tools.py b/pydbsmgr/utils/tools.py
@@ -90,13 +90,14 @@ def write_pyarrow(
         """Write pyarrow table as `parquet` format"""
         format_type = "parquet"
         for table, blob_name in zip(pytables, names):
-            buf = pa.BufferOutputStream()
-            pq.write_table(table, buf)
-            parquet_data = buf.getvalue().to_pybytes()
-            blob_path_name = directory_name + "/" + blob_name
-            self._container_client.upload_blob(
-                name=blob_path_name + "." + format_type, data=parquet_data, overwrite=overwrite
-            )
+            if table != None:
+                buf = pa.BufferOutputStream()
+                pq.write_table(table, buf)
+                parquet_data = buf.getvalue().to_pybytes()
+                blob_path_name = directory_name + "/" + blob_name
+                self._container_client.upload_blob(
+                    name=blob_path_name + "." + format_type, data=parquet_data, overwrite=overwrite
+                )
 
     def write_parquet(
         self,
@@ -110,17 +111,18 @@ def write_parquet(
         files = []
         format_type = "parquet"
         for data, blob_name in zip(dfs, names):
-            table = pa.Table.from_pandas(data)
-            buf = pa.BufferOutputStream()
-            pq.write_table(table, buf)
-            parquet_data = buf.getvalue().to_pybytes()
-            blob_path_name = directory_name + "/" + blob_name
-            if upload:
-                self._container_client.upload_blob(
-                    name=blob_path_name + "." + format_type, data=parquet_data, overwrite=overwrite
-                )
-            else:
-                files.append(parquet_data)
+            if data != None:
+                table = pa.Table.from_pandas(data)
+                buf = pa.BufferOutputStream()
+                pq.write_table(table, buf)
+                parquet_data = buf.getvalue().to_pybytes()
+                blob_path_name = directory_name + "/" + blob_name
+                if upload:
+                    self._container_client.upload_blob(
+                        name=blob_path_name + "." + format_type, data=parquet_data, overwrite=overwrite
+                    )
+                else:
+                    files.append(parquet_data)
 
         if not upload:
             return files

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="pydbsmgr",
-    version="0.4.7",
+    version="0.4.8",
     author="J. A. Moreno-Guerra",
     author_email="[email protected]",
     description="Testing installation of Package",