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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
# Copyright (C) 2005 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
"""Export functionality, which can take a Tree and create a different representation.
Such as non-controlled directories, tarfiles, zipfiles, etc.
"""
import os
import bzrlib.errors as errors
# Maps format name => export function
_exporters = {}
# Maps filename extensions => export format name
_exporter_extensions = {}
def register_exporter(format, extensions, func, override=False):
"""Register an exporter.
:param format: This is the name of the format, such as 'tgz' or 'zip'
:param extensions: Extensions which should be used in the case that a
format was not explicitly specified.
:type extensions: List
:param func: The function. It will be called with (tree, dest, root)
:param override: Whether to override an object which already exists.
Frequently plugins will want to provide functionality
until it shows up in mainline, so the default is False.
"""
global _exporters, _exporter_extensions
if (format not in _exporters) or override:
_exporters[format] = func
for ext in extensions:
if (ext not in _exporter_extensions) or override:
_exporter_extensions[ext] = format
def register_lazy_exporter(scheme, extensions, module, funcname):
"""Register lazy-loaded exporter function.
When requesting a specific type of export, load the respective path.
"""
def _loader(tree, dest, root, subdir, filtered, per_file_timestamps):
mod = __import__(module, globals(), locals(), [funcname])
func = getattr(mod, funcname)
return func(tree, dest, root, subdir, filtered=filtered,
per_file_timestamps=per_file_timestamps)
register_exporter(scheme, extensions, _loader)
def export(tree, dest, format=None, root=None, subdir=None, filtered=False,
per_file_timestamps=False):
"""Export the given Tree to the specific destination.
:param tree: A Tree (such as RevisionTree) to export
:param dest: The destination where the files,etc should be put
:param format: The format (dir, zip, etc), if None, it will check the
extension on dest, looking for a match
:param root: The root location inside the format.
It is common practise to have zipfiles and tarballs
extract into a subdirectory, rather than into the
current working directory.
If root is None, the default root will be
selected as the destination without its
extension.
:param subdir: A starting directory within the tree. None means to export
the entire tree, and anything else should specify the relative path to
a directory to start exporting from.
:param filtered: If True, content filtering is applied to the
files exported.
:param per_file_timestamps: Whether to use the timestamp stored in the
tree rather than now(). This will do a revision lookup
for every file so will be significantly slower.
"""
global _exporters, _exporter_extensions
if format is None:
for ext in _exporter_extensions:
if dest.endswith(ext):
format = _exporter_extensions[ext]
break
# Most of the exporters will just have to call
# this function anyway, so why not do it for them
if root is None:
root = get_root_name(dest)
if format not in _exporters:
raise errors.NoSuchExportFormat(format)
tree.lock_read()
try:
return _exporters[format](tree, dest, root, subdir, filtered=filtered,
per_file_timestamps=per_file_timestamps)
finally:
tree.unlock()
def get_root_name(dest):
"""Get just the root name for an export.
>>> get_root_name('../mytest.tar')
'mytest'
>>> get_root_name('mytar.tar')
'mytar'
>>> get_root_name('mytar.tar.bz2')
'mytar'
>>> get_root_name('tar.tar.tar.tgz')
'tar.tar.tar'
>>> get_root_name('bzr-0.0.5.tar.gz')
'bzr-0.0.5'
>>> get_root_name('bzr-0.0.5.zip')
'bzr-0.0.5'
>>> get_root_name('bzr-0.0.5')
'bzr-0.0.5'
>>> get_root_name('a/long/path/mytar.tgz')
'mytar'
>>> get_root_name('../parent/../dir/other.tbz2')
'other'
"""
global _exporter_extensions
dest = os.path.basename(dest)
for ext in _exporter_extensions:
if dest.endswith(ext):
return dest[:-len(ext)]
return dest
def _export_iter_entries(tree, subdir):
"""Iter the entries for tree suitable for exporting.
:param tree: A tree object.
:param subdir: None or the path of an entry to start exporting from.
"""
inv = tree.inventory
if subdir is None:
subdir_object = None
else:
subdir_id = inv.path2id(subdir)
if subdir_id is not None:
subdir_object = inv[subdir_id]
# XXX: subdir is path not an id, so NoSuchId isn't proper error
else:
raise errors.NoSuchId(tree, subdir)
if subdir_object is not None and subdir_object.kind != 'directory':
yield subdir_object.name, subdir_object
return
else:
entries = inv.iter_entries(subdir_object)
if subdir is None:
entries.next() # skip root
for entry in entries:
# The .bzr* namespace is reserved for "magic" files like
# .bzrignore and .bzrrules - do not export these
if entry[0].startswith(".bzr"):
continue
if subdir is None:
if not tree.has_filename(entry[0]):
continue
else:
if not tree.has_filename(os.path.join(subdir, entry[0])):
continue
yield entry
register_lazy_exporter(None, [], 'bzrlib.export.dir_exporter', 'dir_exporter')
register_lazy_exporter('dir', [], 'bzrlib.export.dir_exporter', 'dir_exporter')
register_lazy_exporter('tar', ['.tar'], 'bzrlib.export.tar_exporter', 'tar_exporter')
register_lazy_exporter('tgz', ['.tar.gz', '.tgz'], 'bzrlib.export.tar_exporter', 'tgz_exporter')
register_lazy_exporter('tbz2', ['.tar.bz2', '.tbz2'], 'bzrlib.export.tar_exporter', 'tbz_exporter')
register_lazy_exporter('zip', ['.zip'], 'bzrlib.export.zip_exporter', 'zip_exporter')
|