Allow to save csv without headers

Author: RichardHitier

speasy/core/codecs/bundled_codecs/hapi_csv/codec.py

@@ -197,7 +197,7 @@ def save_variables(self,
                        **kwargs
                        ) -> Union[bool, Buffer]:
         hapi_csv_file = _speasy_variables_to_hapi_csv(variables)
-        return save_hapi_csv(hapi_csv_file, file)
+        return save_hapi_csv(hapi_csv_file, file, **kwargs)
 
     @property
     def supported_extensions(self) -> List[str]:

speasy/core/codecs/bundled_codecs/hapi_csv/writer.py

@@ -6,7 +6,7 @@
 import pandas as pds
 
 
-def _to_csv(hapi_csv_file: HapiCsvFile, dest:io.IOBase, with_headers=True) -> bool:
+def _to_csv(hapi_csv_file: HapiCsvFile, dest:io.IOBase, with_headers: bool = True) -> bool:
     np_startDate = hapi_csv_file.time_axis[0]
     np_stopDate = hapi_csv_file.time_axis[-1]
     startDate  = np_startDate.astype("datetime64[us]").astype("O")
@@ -39,14 +39,14 @@ def _to_csv(hapi_csv_file: HapiCsvFile, dest:io.IOBase, with_headers=True) -> bo
     return True
 
 
-def save_hapi_csv(hapi_csv_file: HapiCsvFile, file: Optional[Union[str, io.IOBase]] = None) -> Union[bool, Buffer]:
+def save_hapi_csv(hapi_csv_file: HapiCsvFile, file: Optional[Union[str, io.IOBase]] = None, with_headers: bool = True) -> Union[bool, Buffer]:
     if type(file) is str:
         with open(file, 'wb') as f:
-            return _to_csv(hapi_csv_file, f)
+            return _to_csv(hapi_csv_file, f, with_headers=with_headers)
     elif hasattr(file, 'write'):
-        return _to_csv(hapi_csv_file, file)
+        return _to_csv(hapi_csv_file, file, with_headers=with_headers)
     elif file is None:
         buff = io.BytesIO()
-        _to_csv(hapi_csv_file, buff)
+        _to_csv(hapi_csv_file, buff, with_headers=with_headers)
         return buff.getvalue()
     raise ValueError("Invalid file type")