/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/archive/zip.py

  • Committer: Gustav Hartvigsson
  • Date: 2021-01-09 21:36:27 UTC
  • Revision ID: gustav.hartvigsson@gmail.com-20210109213627-h1xwcutzy9m7a99b
Added 'Case Preserving Working Tree Use Cases' from Canonical Wiki

* Addod a page from the Canonical Bazaar wiki
  with information on the scmeatics of case
  perserving filesystems an a case insensitive
  filesystem works.
  
  * Needs re-work, but this will do as it is the
    same inforamoton as what was on the linked
    page in the currint documentation.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/archive/zip.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
"""Export a Tree to a non-versioned directory.
 
17
"""Export a Tree to a zip file.
18
18
"""
19
19
 
 
20
from contextlib import closing
20
21
import os
21
22
import stat
 
23
import sys
 
24
import tempfile
22
25
import time
23
26
import zipfile
24
27
 
25
 
from bzrlib import (
 
28
from .. import (
26
29
    osutils,
27
30
    )
28
 
from bzrlib.export import _export_iter_entries
29
 
from bzrlib.filters import (
30
 
    ContentFilterContext,
31
 
    filtered_output_bytes,
32
 
    )
33
 
from bzrlib.trace import mutter
34
 
 
35
 
 
36
 
# Windows expects this bit to be set in the 'external_attr' section
37
 
# Or it won't consider the entry a directory
 
31
from ..export import _export_iter_entries
 
32
from ..trace import mutter
 
33
 
 
34
 
 
35
# Windows expects this bit to be set in the 'external_attr' section,
 
36
# or it won't consider the entry a directory.
38
37
ZIP_DIRECTORY_BIT = (1 << 4)
39
 
FILE_PERMISSIONS = (0644 << 16)
 
38
FILE_PERMISSIONS = (0o644 << 16)
 
39
DIR_PERMISSIONS = (0o755 << 16)
40
40
 
41
41
_FILE_ATTR = stat.S_IFREG | FILE_PERMISSIONS
42
 
_DIR_ATTR = stat.S_IFDIR | ZIP_DIRECTORY_BIT
43
 
 
44
 
 
45
 
def zip_exporter(tree, dest, root, subdir, filtered=False,
46
 
                 per_file_timestamps=False):
 
42
_DIR_ATTR = stat.S_IFDIR | ZIP_DIRECTORY_BIT | DIR_PERMISSIONS
 
43
 
 
44
 
 
45
def zip_archive_generator(tree, dest, root, subdir=None,
 
46
                          force_mtime=None):
47
47
    """ Export this tree to a new zip file.
48
48
 
49
49
    `dest` will be created holding the contents of this tree; if it
50
50
    already exists, it will be overwritten".
51
51
    """
52
 
    mutter('export version %r', tree)
53
 
 
54
 
    now = time.localtime()[:6]
55
 
 
56
52
    compression = zipfile.ZIP_DEFLATED
57
 
    zipf = zipfile.ZipFile(dest, "w", compression)
58
 
 
59
 
    try:
60
 
        for dp, ie in _export_iter_entries(tree, subdir):
61
 
            file_id = ie.file_id
62
 
            mutter("  export {%s} kind %s to %s", file_id, ie.kind, dest)
63
 
 
64
 
            # zipfile.ZipFile switches all paths to forward
65
 
            # slashes anyway, so just stick with that.
66
 
            if per_file_timestamps:
67
 
                mtime = tree.get_file_mtime(ie.file_id, dp)
68
 
            else:
69
 
                mtime = now
70
 
            filename = osutils.pathjoin(root, dp).encode('utf8')
71
 
            if ie.kind == "file":
72
 
                zinfo = zipfile.ZipInfo(
73
 
                            filename=filename,
74
 
                            date_time=mtime)
75
 
                zinfo.compress_type = compression
76
 
                zinfo.external_attr = _FILE_ATTR
77
 
                if filtered:
78
 
                    chunks = tree.get_file_lines(file_id)
79
 
                    filters = tree._content_filter_stack(dp)
80
 
                    context = ContentFilterContext(dp, tree, ie)
81
 
                    contents = filtered_output_bytes(chunks, filters, context)
82
 
                    content = ''.join(contents)
 
53
    with tempfile.SpooledTemporaryFile() as buf:
 
54
        with closing(zipfile.ZipFile(buf, "w", compression)) as zipf, \
 
55
                tree.lock_read():
 
56
            for dp, tp, ie in _export_iter_entries(tree, subdir):
 
57
                mutter("  export {%s} kind %s to %s", tp, ie.kind, dest)
 
58
 
 
59
                # zipfile.ZipFile switches all paths to forward
 
60
                # slashes anyway, so just stick with that.
 
61
                if force_mtime is not None:
 
62
                    mtime = force_mtime
83
63
                else:
84
 
                    content = tree.get_file_text(file_id)
85
 
                zipf.writestr(zinfo, content)
86
 
            elif ie.kind == "directory":
87
 
                # Directories must contain a trailing slash, to indicate
88
 
                # to the zip routine that they are really directories and
89
 
                # not just empty files.
90
 
                zinfo = zipfile.ZipInfo(
91
 
                            filename=filename + '/',
92
 
                            date_time=mtime)
93
 
                zinfo.compress_type = compression
94
 
                zinfo.external_attr = _DIR_ATTR
95
 
                zipf.writestr(zinfo,'')
96
 
            elif ie.kind == "symlink":
97
 
                zinfo = zipfile.ZipInfo(
98
 
                            filename=(filename + '.lnk'),
99
 
                            date_time=mtime)
100
 
                zinfo.compress_type = compression
101
 
                zinfo.external_attr = _FILE_ATTR
102
 
                zipf.writestr(zinfo, ie.symlink_target)
103
 
 
104
 
        zipf.close()
105
 
 
106
 
    except UnicodeEncodeError:
107
 
        zipf.close()
108
 
        os.remove(dest)
109
 
        from bzrlib.errors import BzrError
110
 
        raise BzrError("Can't export non-ascii filenames to zip")
 
64
                    mtime = tree.get_file_mtime(tp)
 
65
                date_time = time.localtime(mtime)[:6]
 
66
                filename = osutils.pathjoin(root, dp)
 
67
                if ie.kind == "file":
 
68
                    zinfo = zipfile.ZipInfo(
 
69
                        filename=filename,
 
70
                        date_time=date_time)
 
71
                    zinfo.compress_type = compression
 
72
                    zinfo.external_attr = _FILE_ATTR
 
73
                    content = tree.get_file_text(tp)
 
74
                    zipf.writestr(zinfo, content)
 
75
                elif ie.kind in ("directory", "tree-reference"):
 
76
                    # Directories must contain a trailing slash, to indicate
 
77
                    # to the zip routine that they are really directories and
 
78
                    # not just empty files.
 
79
                    zinfo = zipfile.ZipInfo(
 
80
                        filename=filename + '/',
 
81
                        date_time=date_time)
 
82
                    zinfo.compress_type = compression
 
83
                    zinfo.external_attr = _DIR_ATTR
 
84
                    zipf.writestr(zinfo, '')
 
85
                elif ie.kind == "symlink":
 
86
                    zinfo = zipfile.ZipInfo(
 
87
                        filename=(filename + '.lnk'),
 
88
                        date_time=date_time)
 
89
                    zinfo.compress_type = compression
 
90
                    zinfo.external_attr = _FILE_ATTR
 
91
                    zipf.writestr(zinfo, tree.get_symlink_target(tp))
 
92
        # Urgh, headers are written last since they include e.g. file size.
 
93
        # So we have to buffer it all :(
 
94
        buf.seek(0)
 
95
        for chunk in osutils.file_iterator(buf):
 
96
            yield chunk