Adding final support for Parquet/Pickle

headat/lib/constants.py

@@ -36,14 +36,14 @@
         "method": "to_latex"
     },
     "parquet": {
-        "extension": "pqt",
+        "extension": "parquet",
         "method": "to_parquet"
     },
     "pickle": {
-        "extension": "pkl",
+        "extension": "pickle",
         "method": "to_pickle"
     },
-    "sqlite": {
+    "sql": {
         "extension": "db",
         "method": "to_sql"
     },

headat/lib/functions.py

@@ -1,6 +1,7 @@
 from .constants import *
 import datetime
 import os
+import scipy.io
 import warnings
 # Export routine to LaTeX generate a FutureWarning which has to be silenced
 warnings.simplefilter(action='ignore', category=FutureWarning)

headat/main.py

@@ -319,6 +319,46 @@ def t_tex(self, **kwargs) -> bool:
         except:
             return False
 
+    def t_parquet(self, **kwargs) -> bool:
+        """
+        Function converting the record to the Apache Parquet format
+        :rtype: bool
+        :return: Boolean set to True if conversion has been successfully performed
+        """
+        # Gathering the details concerning the specified format
+        df, method, filename = self.get_conversion_details("parquet")
+        cl_m = eval(f"df.{method}")
+        try:
+            cl_m(filename, **kwargs)
+            return True
+        except:
+            return False
+
+    def t_pickle(self, **kwargs) -> bool:
+        """
+        Function converting the record to the Pickle (standard serialization) format
+        :rtype: bool
+        :return: Boolean set to True if conversion has been successfully performed
+        """
+        # Gathering the details concerning the specified format
+        df, method, filename = self.get_conversion_details("pickle")
+        cl_m = eval(f"df.{method}")
+        try:
+            cl_m(filename, **kwargs)
+            return True
+        except:
+            return False
+
+    def t_sql(self, **kwargs) -> bool:
+        """
+        Function converting the record to a database format supported by SQLAlchemy
+        :rtype: bool
+        :return: Boolean set to True if conversion has been successfully performed
+        """
+        # TODO
+        pass
+
+
     # ----------------------------------------------------------------
     #                           GENERIC METHODS
 

requirements.txt

@@ -1,4 +1,5 @@
 openpyxl
 pyarrow
 tabulate
-fastparquet
+fastparquet
+SQLAlchemy

test.py

@@ -15,4 +15,6 @@
 a.t_json()
 a.t_xml()
 a.t_md()
-a.t_tex()
+a.t_tex()
+a.t_parquet()
+a.t_pickle()