45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
import hashlib
|
|
import io
|
|
import tarfile
|
|
import time
|
|
|
|
from promenade import logging
|
|
|
|
__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
|
|
tar_info.mtime = int(time.time())
|
|
|
|
if tar_info.size > 0:
|
|
# Ignore bandit false positive: B303:blacklist
|
|
# This is a basic checksum for debugging not a secure hash.
|
|
LOG.debug( # nosec
|
|
'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()
|