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
186
187
188
189
190
191
192
193
194
195
196
197
198
|
# Copyright (C) 2008 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, see <http://www.gnu.org/licenses/>.
"""Miscellaneous useful stuff."""
import stat
def escape_commit_message(message):
"""Replace xml-incompatible control characters."""
# This really ought to be provided by bzrlib.
# Code copied from bzrlib.commit.
# Python strings can include characters that can't be
# represented in well-formed XML; escape characters that
# aren't listed in the XML specification
# (http://www.w3.org/TR/REC-xml/#NT-Char).
import re
message, _ = re.subn(
u'[^\x09\x0A\x0D\u0020-\uD7FF\uE000-\uFFFD]+',
lambda match: match.group(0).encode('unicode_escape'),
message)
return message
def best_format_for_objects_in_a_repository(repo):
"""Find the high-level format for branches and trees given a repository.
When creating branches and working trees within a repository, Bazaar
defaults to using the default format which may not be the best choice.
This routine does a reverse lookup of the high-level format registry
to find the high-level format that a shared repository was most likely
created via.
:return: the BzrDirFormat or None if no matches were found.
"""
# Based on code from bzrlib/info.py ...
from bzrlib import bzrdir
repo_format = repo._format
candidates = []
non_aliases = set(bzrdir.format_registry.keys())
non_aliases.difference_update(bzrdir.format_registry.aliases())
for key in non_aliases:
format = bzrdir.format_registry.make_bzrdir(key)
# LocalGitBzrDirFormat has no repository_format
if hasattr(format, "repository_format"):
if format.repository_format == repo_format:
candidates.append((key, format))
if len(candidates):
# Assume the first one. Is there any reason not to do that?
name, format = candidates[0]
return format
else:
return None
def open_destination_directory(location, format=None, verbose=True):
"""Open a destination directory and return the BzrDir.
If destination has a control directory, it will be returned.
Otherwise, the destination should be empty or non-existent and
a shared repository will be created there.
:param location: the destination directory
:param format: the format to use or None for the default
:param verbose: display the format used if a repository is created.
:return: BzrDir for the destination
"""
import os
from bzrlib import bzrdir, errors, trace, transport
try:
control, relpath = bzrdir.BzrDir.open_containing(location)
# XXX: Check the relpath is None here?
return control
except errors.NotBranchError:
pass
# If the directory exists, check it is empty. Otherwise create it.
if os.path.exists(location):
contents = os.listdir(location)
if contents:
errors.BzrCommandError("Destination must have a .bzr directory, "
" not yet exist or be empty - files found in %s" % (location,))
else:
try:
os.mkdir(location)
except IOError, ex:
errors.BzrCommandError("Unable to create %s: %s" %
(location, ex))
# Create a repository for the nominated format.
trace.note("Creating destination repository ...")
if format is None:
format = bzrdir.format_registry.make_bzrdir('default')
to_transport = transport.get_transport(location)
to_transport.ensure_base()
control = format.initialize_on_transport(to_transport)
repo = control.create_repository(shared=True)
if verbose:
from bzrlib.info import show_bzrdir_info
show_bzrdir_info(repo.bzrdir, verbose=0)
return control
def kind_to_mode(kind, executable):
if kind == "file":
if executable == True:
return stat.S_IFREG | 0755
elif executable == False:
return stat.S_IFREG | 0644
else:
raise AssertionError("Executable %r invalid" % executable)
elif kind == "symlink":
return stat.S_IFLNK
elif kind == "directory":
return stat.S_IFDIR
elif kind == "tree-reference":
return 0160000
else:
raise AssertionError("Unknown file kind '%s'" % kind)
def mode_to_kind(mode):
# Note: Output from git-fast-export slightly different to spec
if mode in (0644, 0100644):
return 'file', False
elif mode in (0755, 0100755):
return 'file', True
elif mode == 0040000:
return 'directory', False
elif mode == 0120000:
return 'symlink', False
elif mode == 0160000:
return 'tree-reference', False
else:
raise AssertionError("invalid mode %o" % mode)
def binary_stream(stream):
"""Ensure a stream is binary on Windows.
:return: the stream
"""
try:
import os
if os.name == 'nt':
fileno = getattr(stream, 'fileno', None)
if fileno:
no = fileno()
if no >= 0: # -1 means we're working as subprocess
import msvcrt
msvcrt.setmode(no, os.O_BINARY)
except ImportError:
pass
return stream
def single_plural(n, single, plural):
"""Return a single or plural form of a noun based on number."""
if n == 1:
return single
else:
return plural
def invert_dictset(d):
"""Invert a dictionary with keys matching a set of values, turned into lists."""
# Based on recipe from ASPN
result = {}
for k, c in d.iteritems():
for v in c:
keys = result.setdefault(v, [])
keys.append(k)
return result
def invert_dict(d):
"""Invert a dictionary with keys matching each value turned into a list."""
# Based on recipe from ASPN
result = {}
for k, v in d.iteritems():
keys = result.setdefault(v, [])
keys.append(k)
return result
|
