Python tarfile progress output?

Question

Я использую следующий код, чтобы извлечь файл дегтя:

import tarfile 
tar = tarfile.open("sample.tar.gz") 
tar.extractall() 
tar.close() 

Однако, я хотел бы, чтобы следить за прогрессом в виде файлов которые добывают в данный момент. Как я могу это сделать?

EXTRA BONUS POINTS: возможно ли создать процент от процесса извлечения? Я хотел бы использовать это для tkinter для обновления индикатора выполнения. Благодаря!

Answer 1

Оба файла прогресс и глобальные про гресс:

import tarfile 
import io 
import os 

def get_file_progress_file_object_class(on_progress): 
    class FileProgressFileObject(tarfile.ExFileObject): 
     def read(self, size, *args): 
      on_progress(self.name, self.position, self.size) 
      return tarfile.ExFileObject.read(self, size, *args) 
    return FileProgressFileObject 

class TestFileProgressFileObject(tarfile.ExFileObject): 
    def read(self, size, *args): 
     on_progress(self.name, self.position, self.size) 
     return tarfile.ExFileObject.read(self, size, *args) 

class ProgressFileObject(io.FileIO): 
    def __init__(self, path, *args, **kwargs): 
     self._total_size = os.path.getsize(path) 
     io.FileIO.__init__(self, path, *args, **kwargs) 

    def read(self, size): 
     print("Overall process: %d of %d" %(self.tell(), self._total_size)) 
     return io.FileIO.read(self, size) 

def on_progress(filename, position, total_size): 
    print("%s: %d of %s" %(filename, position, total_size)) 

tarfile.TarFile.fileobject = get_file_progress_file_object_class(on_progress) 
tar = tarfile.open(fileobj=ProgressFileObject("a.tgz")) 
tar.extractall() 
tar.close() 

Answer 2

Вы можете использовать extract вместо extractall - вы сможете печатать имена участников при их извлечении. Чтобы получить список участников, вы можете использовать getmembers.

библиотека текстуальное ProgressBar можно найти здесь:

• http://code.google.com/p/python-progressbar/

Tkinter сниппет:

• http://tkinter.unpythonic.net/wiki/ProgressBar

Answer 3

Вы можете задать параметр members в extractall()

with tarfile.open(<path>, 'r') as tarball: 
    tarball.extractall(path=<some path>, members = track_progress(tarball)) 

def track_progress(members): 
    for member in members: 
     # this will be the current file being extracted 
     yield member 

member являются TarInfo объекты, увидеть все доступные функции и свойства here

Answer 4

Там прохладное решение здесь переопределяет модуль файл обработан в качестве замены для замены и позволяет указать обратный вызов для обновления.

https://github.com/thomaspurchas/tarfile-Progress-Reporter/

обновляется на основе комментариев

Answer 5

Чтобы узнать, какой файл в настоящее время извлечены, следующие работал для меня:

import tarfile 

print "Extracting the contents of sample.tar.gz:" 
tar = tarfile.open("sample.tar.gz") 

for member_info in tar.getmembers(): 
    print "- extracting: " + member_info.name 
    tar.extract(member_info) 

tar.close()