Untitled

def save_tables(self, STATE, data_contratos, data_convenios):

        header_contratos = ['No. Contrato', 'Data do Contrato', 'Razão Social/Nome', 'CNPJ', 'Valor do Contrato']
        header_convenios = ['No. Convênio', 'Data do Convênio', 'Descrição do Objeto', 'Razão Social do Convenente', 'CNPJ', 'Valor do Convênio']

        dataFrameContratos = pd.DataFrame(data=data_contratos, columns=header_contratos)
        dataFrameConvenios = pd.DataFrame(data=data_convenios, columns=header_convenios)

        # %% Setup Local File System
        try:
            dataFrameContratos = pd.DataFrame(data=data_contratos, columns=header_contratos)
            dataFrameConvenios = pd.DataFrame(data=data_convenios, columns=header_convenios)
        except Exception as e:
            print(f"ERRO: {e}")
            logger.error(f"ERRO: {e}")
            ERROR = True

        dataFrameContratos.to_csv('{}/contratos/{}.csv'.format(self.DOWNLOAD_FOLDER, STATE), index=False)
        dataFrameConvenios.to_csv('{}/convenios/{}.csv'.format(self.DOWNLOAD_FOLDER, STATE), index=False)
        logger.info(f'INFO: As planilhas do DR {STATE} foram salvos com sucesso')

    def save_tables_with_contrato_categoria(self, STATE, data_contratos, data_convenios):

        header_contratos = ['No. Contrato', 'Data do Contrato', 'Categoria da Descrição do Objeto', 'Razão Social/Nome', 'CNPJ', 'Valor do Contrato']
        header_convenios = ['No. Convênio', 'Data do Convênio', 'Descrição do Objeto', 'Razão Social do Convenente', 'CNPJ', 'Valor do Convênio']

        try:
            dataFrameContratos = pd.DataFrame(data=data_contratos, columns=header_contratos)
            os.makedirs("{}/contratos/".format(self.DOWNLOAD_FOLDER, STATE), exist_ok=True)
            dataFrameContratos.to_csv('{}/contratos/{}.csv'.format(self.DOWNLOAD_FOLDER, STATE), index=False)
        except Exception as e:
            print(f"ERRO: {e}")
            logger.error(f"ERRO: {e}")
            ERROR = True

        # %% Setup Local File System
        try:
            dataFrameConvenios = pd.DataFrame(data=data_convenios, columns=header_convenios)
            os.makedirs("{}/convenios/".format(self.DOWNLOAD_FOLDER, STATE), exist_ok=True)
            dataFrameConvenios.to_csv('{}/convenios/{}.csv'.format(self.DOWNLOAD_FOLDER, STATE), index=False)
        except Exception as e:
            print(f"ERRO: {e}")
            logger.error(f"ERRO: {e}")
            ERROR = True
        
        logger.info(f'INFO: As planilhas do DR {STATE} foram salvos com sucesso')

    def save_tables_with_convenio_tipo(self, STATE, data_contratos, data_convenios):

        header_contratos = ['No. Contrato', 'Data do Contrato', 'Razão Social/Nome', 'CNPJ', 'Valor do Contrato']
        header_convenios = ['No. Convênio', 'Data do Convênio', 'Descrição do Objeto', 'Tipo', 'Razão Social do Convenente', 'CNPJ', 'Valor do Convênio']

        try:
            dataFrameContratos = pd.DataFrame(data=data_contratos, columns=header_contratos)
            os.makedirs("{}/contratos/".format(self.DOWNLOAD_FOLDER, STATE), exist_ok=True)
            dataFrameContratos.to_csv('{}/contratos/{}.csv'.format(self.DOWNLOAD_FOLDER, STATE), index=False)
        except Exception as e:
            print(f"ERRO: {e}")
            logger.error(f"ERRO: {e}")
            ERROR = True

        # %% Setup Local File System
        try:
            dataFrameConvenios = pd.DataFrame(data=data_convenios, columns=header_convenios)
            os.makedirs("{}/convenios/".format(self.DOWNLOAD_FOLDER, STATE), exist_ok=True)
            dataFrameConvenios.to_csv('{}/convenios/{}.csv'.format(self.DOWNLOAD_FOLDER, STATE), index=False)
        except Exception as e:
            print(f"ERRO: {e}")
            logger.error(f"ERRO: {e}")
            ERROR = True
        
        logger.info(f'INFO: As planilhas do DR {STATE} foram salvos com sucesso')