stringa di scrittura Python direttamente su tarfile

Question

C'è un modo per scrivere una stringa direttamente su un file? Da http://docs.python.org/library/tarfile.html sembra che solo i file già scritti nel file system possano essere aggiunti.

Answer 1

Direi che è possibile, giocando con TarInfo e TarFile.addfile passando un StringIO come un oggetto file.

Molto grezzo, ma funziona

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

Come Stefano ha sottolineato, è possibile utilizzare TarFile.addfile e 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() 

probabilmente si vorrà per riempire altri campi di tarinfo (ad esempio mtime, uname ecc) così.

Answer 3

è necessario utilizzare gli oggetti TarInfo e il metodo addfile invece del solito metodo Add:

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

Nel mio caso ho voluto leggere da un file tar esistente, aggiungere alcuni dati ai contenuti, e scrivere in un nuovo file. Qualcosa di simile:

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) 

Codice aggiuntivo è necessario per la gestione di directory e collegamenti.

Answer 5

Solo per la cronologia:
Gli oggetti StringIO hanno una proprietà .len.
Non c'è bisogno di cercare (0) e fare len (foo.buf)
Non c'è bisogno di tenere l'intera stringa in giro per fare len(), o Dio non voglia, fare la contabilità da soli.

(Forse non ha al momento il PO è stato scritto.)

Answer 6

ho trovato questo guardando come servire in Django un appena creato in memoria .tgz archivio, può essere qualcun altro troverà il mio codice utile:

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