Làm thế nào để nén tệp csv thành lưu trữ zip trực tiếp?

Question

Tôi đang tạo ra một số lượng file csv động, sử dụng đoạn mã sau:

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) 

Để tiết kiệm không gian, tôi muốn nén chúng.
Sử dụng các module gzip là khá dễ dàng:

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

Nhưng tôi muốn các tập tin ở định dạng 'zip'.

Tôi đã thử mô-đun zipfile, nhưng tôi không thể ghi trực tiếp tệp vào lưu trữ zip.

Thay vào đó, tôi phải ghi tệp csv vào đĩa, nén chúng vào tệp zip bằng mã sau và sau đó xóa tệp csv.

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

Làm cách nào để ghi tệp csv trực tiếp vào zip nén tương tự như gzip?

Answer 1

Sử dụng các đối tượng cStringIO.StringIO để bắt chước một file:

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

Cảm ơn kroolik Nó thực hiện với sửa đổi chút.

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())