Jak skompresować plik CSV bezpośrednio do archiwum zip?

Question

jestem generowania liczbę plików csv dynamicznie, przy użyciu następującego kodu:

import csv 
fieldnames = ['foo1', 'foo2', 'foo3', 'foo4'] 
with open(csvfilepath, 'wb') as csvfile: 
    csvwrite = csv.DictWriter(csvfile, delimiter=',', fieldnames=fieldnames) 
    csvwrite.writeheader() 
    for row in data: 
     csvwrite.writerow(row) 

Aby zaoszczędzić miejsce, chcę je skompresować.
Korzystanie z modułu gzip jest dość proste:

with gzip.open("foo.gz", "w") as csvfile : 
    csvwrite = csv.DictWriter(csvfile, delimiter=',', fieldnames=fieldnames) 
    csvwrite.writeheader() 
    for row in data: 
     csvwrite.writerow(row) 

Ale chcę plik w formacie „zip”.

Próbowałem modułu zipfile, ale nie jestem w stanie bezpośrednio zapisywać plików w archiwum zip.

Zamiast tego, muszę napisać plik csv na dysk, skompresować go w pliku zip za pomocą następującego kodu, a następnie usunąć plik csv.

with ZipFile(zipfilepath, 'w') as zipfile: 
    zipfile.write(csvfilepath, csvfilename, ZIP_DEFLATED) 

Jak mogę napisać plik csv bezpośrednio na skompresowanym pliku ZIP podobnym do gzip?

Answer 1

Użyj obiektu cStringIO.StringIO naśladować plik:

with ZipFile(your_zip_file, 'w', ZIP_DEFLATED) as zip_file: 
    string_buffer = StringIO() 
    writer = csv.writer(string_buffer) 

    # Write data using the writer object. 

    zip_file.writestr(filename + '.csv', string_buffer.getvalue()) 

Answer 2

Dzięki kroolik Zrobione z niewielkimi modyfikacjami.

with ZipFile(your_zip_file, 'w', ZIP_DEFLATED) as zip_file: 
    string_buffer = StringIO() 
    csvwriter = csv.DictWriter(string_buffer, delimiter=',', fieldnames=fieldnames) 
    csvwrite.writeheader() 
    for row in cdrdata: 
     csvwrite.writerow(row) 
    zip_file.writestr(filename + '.csv', string_buffer.getvalue())