Adding .dta (STATA) file conversion support

headat/lib/constants.py

@@ -54,18 +54,23 @@
         "extension": "mat",
         "method": "custom"
     },
-    "wav":{
+    "wav": {
         "extension": "wav",
         "method": "custom"
     },
-    "edf":{
+    "edf": {
         "extension": "edf",
         "method": "custom"
     },
-    "feather":{
+    "feather": {
         "extension": "fea",
-        "method": "to_feather"
+        "method": "to_feather",
+    },
+    "stata": {
+        "extension": "dta",
+        "method": "to_stata",
     }
 }
+
 EXPORT_FOLDERS = "out"
 EXCEL_ROW_LIMIT = 1048576 - 2

headat/main.py

@@ -470,6 +470,20 @@ def t_feather(self, **kwargs) -> bool:
         except:
             return False
 
+    def t_stata(self, **kwargs) -> bool:
+        """
+        Function converting the record to a .dta file (STATA)
+        :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("stata")
+        cl_m = eval(f"df.{method}")
+        try:
+            cl_m(filename, **kwargs)
+            return True
+        except:
+            return False
     # ----------------------------------------------------------------
     #                           GENERIC METHODS
 

test.py

@@ -23,5 +23,4 @@
 #a.t_edf()
 a.t_csv()
 a.t_feather()
-"""
-pd.DataFrame().to_records()
+"""