67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
import tarfile
|
|
from util.registry.gzipwrap import GzipWrap
|
|
|
|
|
|
class TarImageFormatter(object):
|
|
"""
|
|
Base class for classes which produce a tar containing image and layer data.
|
|
"""
|
|
|
|
def build_stream(self, tag, manifest, synthetic_image_id, layer_iterator,
|
|
tar_stream_getter_iterator, reporter=None):
|
|
"""
|
|
Builds and streams a synthetic .tar.gz that represents the formatted tar created by this class's
|
|
implementation.
|
|
"""
|
|
return GzipWrap(self.stream_generator(tag, manifest, synthetic_image_id, layer_iterator,
|
|
tar_stream_getter_iterator, reporter=reporter))
|
|
|
|
def stream_generator(self, tag, manifest, synthetic_image_id, layer_iterator,
|
|
tar_stream_getter_iterator, reporter=None):
|
|
raise NotImplementedError
|
|
|
|
def tar_file(self, name, contents, mtime=None):
|
|
"""
|
|
Returns the tar binary representation for a file with the given name and file contents.
|
|
"""
|
|
length = len(contents)
|
|
tar_data = self.tar_file_header(name, length, mtime=mtime)
|
|
tar_data += contents
|
|
tar_data += self.tar_file_padding(length)
|
|
return tar_data
|
|
|
|
def tar_file_padding(self, length):
|
|
"""
|
|
Returns tar file padding for file data of the given length.
|
|
"""
|
|
if length % 512 != 0:
|
|
return '\0' * (512 - (length % 512))
|
|
|
|
return ''
|
|
|
|
def tar_file_header(self, name, file_size, mtime=None):
|
|
"""
|
|
Returns tar file header data for a file with the given name and size.
|
|
"""
|
|
info = tarfile.TarInfo(name=name)
|
|
info.type = tarfile.REGTYPE
|
|
info.size = file_size
|
|
|
|
if mtime is not None:
|
|
info.mtime = mtime
|
|
return info.tobuf()
|
|
|
|
def tar_folder(self, name, mtime=None):
|
|
"""
|
|
Returns tar file header data for a folder with the given name.
|
|
"""
|
|
info = tarfile.TarInfo(name=name)
|
|
info.type = tarfile.DIRTYPE
|
|
|
|
if mtime is not None:
|
|
info.mtime = mtime
|
|
|
|
# allow the directory to be readable by non-root users
|
|
info.mode = 0755
|
|
return info.tobuf()
|