Incluído Writer para MS Excel no formato XLSX

convert.py

@@ -55,9 +55,11 @@ def main():
     wcpickle = writer.PickleWriter(logger, path.join(current_base_dir, 'pickle'))  # Pickle writer
     wparquet = writer.ParquetWriter(logger, path.join(output_dir, 'parquet'))  # Parquet writer
     wcparquet = writer.ParquetWriter(logger, path.join(current_base_dir, 'parquet'))  # Parquet writer
+    wxlsx = writer.XlsxWriter(logger, path.join(output_dir, 'excel'))  # Xlsx writer
+    wcxlsx = writer.XlsxWriter(logger, path.join(current_base_dir, 'excel'))  # Xlsx writer
 
     # Executa o módulo principal do programa
-    running = app.App(logger, [pm_input_dir, cm_input_dir], [wcsv, wccsv, wpickle, wcpickle, wparquet, wcparquet], mes, ano)
+    running = app.App(logger, [pm_input_dir, cm_input_dir], [wparquet, wcparquet, wxlsx, wcxlsx], mes, ano)
     running.run()
 
 

pad/converter/writer/__init__.py

@@ -129,4 +129,33 @@ def write(self, df, filename):
         """
         destination = path.join(self._dir, f'{filename}.parquet')
         self._logger.debug(f'Escrevendo {len(df)} linhas para {destination} ...')
-        df.to_parquet(destination)
+        df.to_parquet(destination)
+
+class XlsxWriter:
+    """Writer para arquivos Excel xlsx.
+    """
+    _logger = None # Objeto logger
+    _dir = '' # Diretório onde os arquivos xlsx serão salvos.
+
+    def __init__(self, logger, dirname: str) -> None:
+        """Construtor do writer
+
+        :param logger: Objeto logger
+        :param dirname: Diretório onde os arquivos xlsx serão salvos.
+        """
+        self._logger = logger
+        self._dir = dirname
+        # Verifica se o diretório existe, se não, cria ele.
+        if not path.exists(dirname):
+            logger.warn(f'{dirname} não existe e será criado...')
+            makedirs(dirname)
+
+    def write(self, df, filename):
+        """Salva um pandas.DataFrame para um arquivo xlsx.
+
+        :param df pandas.DataFrame
+        :param filename Um nome de arquivo sem a extensão.
+        """
+        destination = path.join(self._dir, f'{filename}.xlsx')
+        self._logger.debug(f'Escrevendo {len(df)} linhas para {destination} ...')
+        df.to_excel(destination, sheet_name=filename, index=False)