39 lines
1018 B
Python
39 lines
1018 B
Python
from . import logging
|
|
import hashlib
|
|
import io
|
|
import tarfile
|
|
|
|
__all__ = ['TarBundler']
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class TarBundler:
|
|
def __init__(self):
|
|
self._tar_blob = io.BytesIO()
|
|
self._tf = tarfile.open(fileobj=self._tar_blob, mode='w|gz')
|
|
|
|
def add(self, *, path, data, mode):
|
|
if path.startswith('/'):
|
|
path = path[1:]
|
|
|
|
tar_info = tarfile.TarInfo(name=path)
|
|
if isinstance(data, str):
|
|
data_bytes = data.encode('utf-8')
|
|
else:
|
|
data_bytes = data
|
|
tar_info.size = len(data_bytes)
|
|
tar_info.mode = mode
|
|
|
|
if tar_info.size > 0:
|
|
LOG.debug('Adding file path=%s size=%s md5=%s', path,
|
|
tar_info.size, hashlib.md5(data_bytes).hexdigest())
|
|
else:
|
|
LOG.warning('Zero length file added to path=%s', path)
|
|
|
|
self._tf.addfile(tar_info, io.BytesIO(data_bytes))
|
|
|
|
def as_blob(self):
|
|
self._tf.close()
|
|
return self._tar_blob.getvalue()
|