/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/__init__.py

  • Committer: Jelmer Vernooij
  • Date: 2018-05-06 11:48:54 UTC
  • mto: This revision was merged to the branch mainline in revision 6960.
  • Revision ID: jelmer@jelmer.uk-20180506114854-h4qd9ojaqy8wxjsd
Move .mailmap to root.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/archive/__init__.py
1
 
# Copyright (C) 2018 Breezy Developers
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
"""Export trees to tarballs, zipfiles, etc.
18
 
"""
19
 
 
20
 
from __future__ import absolute_import
21
 
 
22
 
import os
23
 
import time
24
 
import warnings
25
 
 
26
 
from .. import (
27
 
    errors,
28
 
    pyutils,
29
 
    registry,
30
 
    trace,
31
 
    )
32
 
 
33
 
 
34
 
class ArchiveFormatInfo(object):
35
 
 
36
 
    def __init__(self, extensions):
37
 
        self.extensions = extensions
38
 
 
39
 
 
40
 
class ArchiveFormatRegistry(registry.Registry):
41
 
    """Registry of archive formats."""
42
 
 
43
 
    def __init__(self):
44
 
        self._extension_map = {}
45
 
        super(ArchiveFormatRegistry, self).__init__()
46
 
 
47
 
    @property
48
 
    def extensions(self):
49
 
        return self._extension_map.keys()
50
 
 
51
 
    def register(self, key, factory, extensions, help=None):
52
 
        """Register an archive format.
53
 
        """
54
 
        registry.Registry.register(self, key, factory, help,
55
 
                                   ArchiveFormatInfo(extensions))
56
 
        self._register_extensions(key, extensions)
57
 
 
58
 
    def register_lazy(self, key, module_name, member_name, extensions,
59
 
                      help=None):
60
 
        registry.Registry.register_lazy(self, key, module_name, member_name,
61
 
                                        help, ArchiveFormatInfo(extensions))
62
 
        self._register_extensions(key, extensions)
63
 
 
64
 
    def _register_extensions(self, name, extensions):
65
 
        for ext in extensions:
66
 
            self._extension_map[ext] = name
67
 
 
68
 
    def get_format_from_filename(self, filename):
69
 
        """Determine the archive format from an extension.
70
 
 
71
 
        :param filename: Filename to guess from
72
 
        :return: A format name, or None
73
 
        """
74
 
        for ext, format in self._extension_map.items():
75
 
            if filename.endswith(ext):
76
 
                return format
77
 
        else:
78
 
            return None
79
 
 
80
 
 
81
 
def create_archive(format, tree, name, root=None, subdir=None,
82
 
                   force_mtime=None):
83
 
    try:
84
 
        archive_fn = format_registry.get(format)
85
 
    except KeyError:
86
 
        raise errors.NoSuchExportFormat(format)
87
 
    return archive_fn(tree, name, root=root, subdir=subdir,
88
 
                      force_mtime=force_mtime)
89
 
 
90
 
 
91
 
format_registry = ArchiveFormatRegistry()
92
 
format_registry.register_lazy('tar', 'breezy.archive.tar',
93
 
                              'plain_tar_generator', ['.tar'], )
94
 
format_registry.register_lazy('tgz', 'breezy.archive.tar',
95
 
                              'tgz_generator', ['.tar.gz', '.tgz'])
96
 
format_registry.register_lazy('tbz2', 'breezy.archive.tar',
97
 
                              'tbz_generator', ['.tar.bz2', '.tbz2'])
98
 
format_registry.register_lazy('tlzma', 'breezy.archive.tar',
99
 
                              'tar_lzma_generator', ['.tar.lzma'])
100
 
format_registry.register_lazy('txz', 'breezy.archive.tar',
101
 
                              'tar_xz_generator', ['.tar.xz'])
102
 
format_registry.register_lazy('zip', 'breezy.archive.zip',
103
 
                              'zip_archive_generator', ['.zip'])