Pisilinux3/pisi
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

archive: Allow passing a file-like object to ArchiveTar

Browse Source 
Also add an "open" function to ArchiveZip. This function
returns a file-like object for files in Zip archives.
This commit is contained in:
Fatih Aşıcı
2010-06-13 15:27:09 +00:00
parent aa7c031bf4
commit 86598d32a5
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pisi/archive.py
+14 -4
View File
@@ -213,11 +213,15 @@ class ArchiveTar(ArchiveBase):
type. Provides access to tar, tar.gz and tar.bz2 files. type. Provides access to tar, tar.gz and tar.bz2 files.
This class provides the unpack magic for tar archives.""" This class provides the unpack magic for tar archives."""
def __init__(self, file_path, arch_type = "tar", no_same_permissions = True, no_same_owner = True): def __init__(self, file_path=None, arch_type="tar",
no_same_permissions=True,
no_same_owner=True,
fileobj=None):
super(ArchiveTar, self).__init__(file_path, arch_type) super(ArchiveTar, self).__init__(file_path, arch_type)
self.tar = None self.tar = None
self.no_same_permissions = no_same_permissions self.no_same_permissions = no_same_permissions
self.no_same_owner = no_same_owner self.no_same_owner = no_same_owner
self.fileobj = fileobj
def unpack(self, target_dir, clean_dir = False): def unpack(self, target_dir, clean_dir = False):
"""Unpack tar archive to a given target directory(target_dir).""" """Unpack tar archive to a given target directory(target_dir)."""
@@ -234,12 +238,13 @@ class ArchiveTar(ArchiveBase):
elif self.type == 'tarbz2': elif self.type == 'tarbz2':
rmode = 'r:bz2' rmode = 'r:bz2'
elif self.type == 'tarlzma': elif self.type == 'tarlzma':
self.tar = TarFile.lzmaopen(self.file_path) self.tar = TarFile.lzmaopen(self.file_path, fileobj=self.fileobj)
else: else:
raise UnknownArchiveType raise UnknownArchiveType
if self.tar is None: if self.tar is None:
self.tar = tarfile.open(self.file_path, rmode) self.tar = tarfile.open(self.file_path, rmode,
fileobj=self.fileobj)
oldwd = None oldwd = None
try: try:
@@ -296,12 +301,14 @@ class ArchiveTar(ArchiveBase):
elif self.type == 'tarlzma': elif self.type == 'tarlzma':
level = int(ctx.config.values.build.compressionlevel) level = int(ctx.config.values.build.compressionlevel)
self.tar = TarFile.lzmaopen(self.file_path, "w", self.tar = TarFile.lzmaopen(self.file_path, "w",
fileobj=self.fileobj,
compresslevel=level) compresslevel=level)
else: else:
raise UnknownArchiveType raise UnknownArchiveType
if self.tar is None: if self.tar is None:
self.tar = tarfile.open(self.file_path, wmode) self.tar = tarfile.open(self.file_path, wmode,
fileobj=self.fileobj)
self.tar.add(file_name, arc_name) self.tar.add(file_name, arc_name)
@@ -392,6 +399,9 @@ class ArchiveZip(ArchiveBase):
self.zip_obj = zipfile.ZipFile(self.file_path, mode) self.zip_obj = zipfile.ZipFile(self.file_path, mode)
def open(self, file_path, mode="r"):
return self.zip_obj.open(file_path, mode)
def close(self): def close(self):
"""Close the zip archive.""" """Close the zip archive."""
self.zip_obj.close() self.zip_obj.close()