Ir para o conteúdo

CSV

Leia e escreva arquivos CSV.

Instalação

pip install botcity-csv-plugin

Importando o Plugin

Depois de instalar este pacote, a próxima etapa é importar o pacote para o seu código e começar a usar as funções.

from botcity.plugins.csv import BotCSVPlugin

Instanciando o Plugin

Para fazer o exemplo, instanciaremos o plugin.

# Instancie o plugin
bot_csv = BotCSVPlugin()

Aviso

Ao ler arquivos do tipo CSV vazios, o plugin lançará uma exceção EmptyDataError. Logo, é necessario verificar a presença de dados no arquivo manipulado para não ocorrer a exceção na execução do código.

Manipulando dados da planilha

Agora, vamos manipular alguns dados do nosso arquivo, adicionando novos dados, classificando e escrevendo o resultado a um novo arquivo.

# Leia um arquivo CSV
bot_csv.read('read.csv')
# Adicione uma linha
bot_csv.add_row([0, 22])
# Classifique por colunas com cabeçalho H1 e H2 em ordem decrescente
bot_csv.sort(['h1', 'h2'], False)

# Imprima o resultado
print(bot_csv.as_list())
# Salve-o em um novo arquivo
bot_csv.write('write.csv')

Código Completo

Vamos dar uma olhada no código completo:

from botcity.plugins.csv import BotCSVPlugin

# Instancie o plugin
bot_csv = BotCSVPlugin()

# Leia um arquivo CSV
bot_csv.read('read.csv')
# Adicione uma linha
bot_csv.add_row([0, 22])
# Classifique por colunas com cabeçalho H1 e H2 em ordem decrescente
bot_csv.sort(['h1', 'h2'], False)

# Imprima o resultado
print(bot_csv.as_list())
# Salve-o em um novo arquivo
bot_csv.write('write.csv')

Dica

Este plugin permite que você use encadeamento do método para que o código acima possa ser escrito assim:

BotCSVPlugin().read('read.csv')
    .add_row([0, 22])
    .sort(['H1', 'H2'], False)
    .write('write.csv')