/usr/lib/python2.7/dist-packages/dulwich/archive.py is in python-dulwich 0.18.5-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 | # archive.py -- Creating an archive from a tarball
# Copyright (C) 2015 Jonas Haag <jonas@lophus.org>
# Copyright (C) 2015 Jelmer Vernooij <jelmer@jelmer.uk>
#
# Dulwich is dual-licensed under the Apache License, Version 2.0 and the GNU
# General Public License as public by the Free Software Foundation; version 2.0
# or (at your option) any later version. You can redistribute it and/or
# modify it under the terms of either of these two licenses.
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# You should have received a copy of the licenses; if not, see
# <http://www.gnu.org/licenses/> for a copy of the GNU General Public License
# and <http://www.apache.org/licenses/LICENSE-2.0> for a copy of the Apache
# License, Version 2.0.
#
"""Generates tarballs for Git trees.
"""
import posixpath
import stat
import tarfile
from io import BytesIO
from contextlib import closing
class ChunkedBytesIO(object):
"""Turn a list of bytestrings into a file-like object.
This is similar to creating a `BytesIO` from a concatenation of the
bytestring list, but saves memory by NOT creating one giant bytestring
first::
BytesIO(b''.join(list_of_bytestrings)) =~= ChunkedBytesIO(
list_of_bytestrings)
"""
def __init__(self, contents):
self.contents = contents
self.pos = (0, 0)
def read(self, maxbytes=None):
if maxbytes < 0:
maxbytes = float('inf')
buf = []
chunk, cursor = self.pos
while chunk < len(self.contents):
if maxbytes < len(self.contents[chunk]) - cursor:
buf.append(self.contents[chunk][cursor:cursor+maxbytes])
cursor += maxbytes
self.pos = (chunk, cursor)
break
else:
buf.append(self.contents[chunk][cursor:])
maxbytes -= len(self.contents[chunk]) - cursor
chunk += 1
cursor = 0
self.pos = (chunk, cursor)
return b''.join(buf)
def tar_stream(store, tree, mtime, format=''):
"""Generate a tar stream for the contents of a Git tree.
Returns a generator that lazily assembles a .tar.gz archive, yielding it in
pieces (bytestrings). To obtain the complete .tar.gz binary file, simply
concatenate these chunks.
:param store: Object store to retrieve objects from
:param tree: Tree object for the tree root
:param mtime: UNIX timestamp that is assigned as the modification time for
all files
:param format: Optional compression format for tarball
:return: Bytestrings
"""
buf = BytesIO()
with closing(tarfile.open(None, "w:%s" % format, buf)) as tar:
for entry_abspath, entry in _walk_tree(store, tree):
try:
blob = store[entry.sha]
except KeyError:
# Entry probably refers to a submodule, which we don't yet
# support.
continue
data = ChunkedBytesIO(blob.chunked)
info = tarfile.TarInfo()
# tarfile only works with ascii.
info.name = entry_abspath.decode('ascii')
info.size = blob.raw_length()
info.mode = entry.mode
info.mtime = mtime
tar.addfile(info, data)
yield buf.getvalue()
buf.truncate(0)
buf.seek(0)
yield buf.getvalue()
def _walk_tree(store, tree, root=b''):
"""Recursively walk a dulwich Tree, yielding tuples of
(absolute path, TreeEntry) along the way.
"""
for entry in tree.iteritems():
entry_abspath = posixpath.join(root, entry.path)
if stat.S_ISDIR(entry.mode):
for _ in _walk_tree(store, store[entry.sha], entry_abspath):
yield _
else:
yield (entry_abspath, entry)
|