python write string trực tiếp tới tarfile

Question

Có cách nào để viết một chuỗi trực tiếp vào một con tarfile không? Từ http://docs.python.org/library/tarfile.html có vẻ như chỉ các tệp đã được ghi vào hệ thống tệp mới có thể được thêm vào.

Answer 1

Tôi có thể nói rằng có thể, bằng cách chơi với TarInfo e TarFile.addfile chuyển qua một StringIO dưới dạng tệp tin.

Rất thô, nhưng hoạt động

import tarfile 
import StringIO 

tar = tarfile.TarFile("test.tar","w") 

string = StringIO.StringIO() 
string.write("hello") 
string.seek(0) 
info = tarfile.TarInfo(name="foo") 
info.size=len(string.buf) 
tar.addfile(tarinfo=info, fileobj=string) 

tar.close() 

Answer 2

Như Stefano chỉ ra, bạn có thể sử dụng TarFile.addfile và StringIO.

import tarfile, StringIO 

data = 'hello, world!' 

tarinfo = tarfile.TarInfo('test.txt') 
tarinfo.size = len(data) 

tar = tarfile.open('test.tar', 'a') 
tar.addfile(tarinfo, StringIO.StringIO(data)) 
tar.close() 

Có thể bạn sẽ muốn để điền vào các lĩnh vực khác của tarinfo (ví dụ mtime, uname vv) cũng.

Answer 3

Bạn phải sử dụng đối tượng TarInfo và phương pháp addfile thay vì phương thức add thông thường:

from StringIO import StringIO 
from tarfile import open, TarInfo 

s = "Hello World!" 
ti = TarInfo("test.txt") 
ti.size = len(s) 

tf = open("testtar.tar", "w") 
tf.addfile(ti, StringIO(s)) 

Answer 4

Trong trường hợp của tôi, tôi muốn đọc từ một file tar hiện có, thêm một số dữ liệu để các nội dung, và viết nó vào một tập tin mới. Một cái gì đó như:

for ti in tar_in: 
    buf_in = tar.extractfile(ti) 
    buf_out = io.BytesIO() 
    size = buf_out.write(buf_in.read()) 
    size += buf_out.write(other data) 
    buf_out.seek(0) 
    ti.size = size 
    tar_out.addfile(ti, fileobj=buf_out) 

Cần thêm mã để xử lý các thư mục và liên kết.

Answer 5

Chỉ để lưu nội dung:
Đối tượng StringIO có thuộc tính .len.
Không cần phải tìm kiếm (0) và làm len (foo.buf)
Không cần phải giữ toàn bộ chuỗi xung quanh để làm len() trên, hoặc Thiên Chúa cấm, làm kế toán cho mình.

(Có lẽ nó đã không tại thời điểm OP đã được viết.)

Answer 6

Tôi thấy điều này tìm cách để phục vụ trong Django một vừa tạo trong bộ nhớ lưu trữ .tgz, có thể ai đó sẽ tìm thấy hữu ích mã của tôi:

import tarfile 
from io import BytesIO 


def serve_file(request): 
    out = BytesIO() 
    tar = tarfile.open(mode = "w:gz", fileobj = out) 
    data = 'lala'.encode('utf-8') 
    file = BytesIO(data) 
    info = tarfile.TarInfo(name="1.txt") 
    info.size = len(data) 
    tar.addfile(tarinfo=info, fileobj=file) 
    tar.close() 

    response = HttpResponse(out.getvalue(), content_type='application/tgz') 
    response['Content-Disposition'] = 'attachment; filename=myfile.tgz' 
    return response