« back to all changes in this revision
Viewing changes to baz2bzr
- browse files at revision 0.1.27
- compare with another revision
- download diff
- download tarball
- view history from revision 0.1.27
- Committer: Michael Ellerman
- Date: 2005-10-19 11:34:39 UTC
- mto: (0.3.1 shelf-dev) (325.1.2 bzrtools)
- mto: This revision was merged to the branch mainline in revision 246.
- Revision ID: michael@ellerman.id.au-20051019113439-193bca379eec5798
Move all shelf functions into a class. Only logic change is we save the
bzr root dir rather than recomputing it again and again.
bzr root dir rather than recomputing it again and again.
- files added:
- .bzrignore
- files removed:
- .be
- .be/bugs
- .be/bugs/07f6e226-092a-4e69-8891-307903604c93
- .be/bugs/07f6e226-092a-4e69-8891-307903604c93/comments
- .be/bugs/07f6e226-092a-4e69-8891-307903604c93/comments/a78766ca-05bb-4875-b716-7f22ec430c52
added
removed
baz2bzr
1
#!/usr/bin/env python
2
3
# Copyright (C) 2005 Aaron Bentley
4
# <aaron.bentley@utoronto.ca>
5
#
6
# This program is free software; you can redistribute it and/or modify
7
# it under the terms of the GNU General Public License as published by
8
# the Free Software Foundation; either version 2 of the License, or
9
# (at your option) any later version.
10
#
11
# This program is distributed in the hope that it will be useful,
12
# but WITHOUT ANY WARRANTY; without even the implied warranty of
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
# GNU General Public License for more details.
15
#
16
# You should have received a copy of the GNU General Public License
17
# along with this program; if not, write to the Free Software
18
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19
20
try:
21
import pybaz
22
except ImportError:
23
print "This command requires PyBaz. Please ensure that it is installed."
24
import sys
25
sys.exit(1)
26
from pybaz.backends.baz import null_cmd
27
import tempfile
28
import os
29
import os.path
30
import shutil
31
import bzrlib
32
import bzrlib.trace
33
import sys
34
import email.Utils
35
from progress import *
36
37
def add_id(files, id=None):
38
"""Adds an explicit id to a list of files.
39
40
:param files: the name of the file to add an id to
41
:type files: list of str
42
:param id: tag one file using the specified id, instead of generating id
43
:type id: str
44
"""
45
args = ["add-id"]
46
if id is not None:
47
args.extend(["--id", id])
48
args.extend(files)
49
return null_cmd(args)
50
51
def test_environ():
52
"""
53
>>> q = test_environ()
54
>>> os.path.exists(q)
55
True
56
>>> os.path.exists(os.path.join(q, "home", ".arch-params"))
57
True
58
>>> teardown_environ(q)
59
>>> os.path.exists(q)
60
False
61
"""
62
tdir = tempfile.mkdtemp(prefix="baz2bzr-")
63
os.environ["HOME"] = os.path.join(tdir, "home")
64
os.mkdir(os.environ["HOME"])
65
arch_dir = os.path.join(tdir, "archive_dir")
66
pybaz.make_archive("test@example.com", arch_dir)
67
work_dir = os.path.join(tdir, "work_dir")
68
os.mkdir(work_dir)
69
os.chdir(work_dir)
70
pybaz.init_tree(work_dir, "test@example.com/test--test--0")
71
lib_dir = os.path.join(tdir, "lib_dir")
72
os.mkdir(lib_dir)
73
pybaz.register_revision_library(lib_dir)
74
pybaz.set_my_id("Test User<test@example.org>")
75
return tdir
76
77
def add_file(path, text, id):
78
"""
79
>>> q = test_environ()
80
>>> add_file("path with space", "text", "lalala")
81
>>> tree = pybaz.tree_root(".")
82
>>> inv = list(tree.iter_inventory_ids(source=True, both=True))
83
>>> ("x_lalala", "path with space") in inv
84
True
85
>>> teardown_environ(q)
86
"""
87
file(path, "wb").write(text)
88
add_id([path], id)
89
90
91
def add_dir(path, id):
92
"""
93
>>> q = test_environ()
94
>>> add_dir("path with\(sp) space", "lalala")
95
>>> tree = pybaz.tree_root(".")
96
>>> inv = list(tree.iter_inventory_ids(source=True, both=True))
97
>>> ("x_lalala", "path with\(sp) space") in inv
98
True
99
>>> teardown_environ(q)
100
"""
101
os.mkdir(path)
102
add_id([path], id)
103
104
def teardown_environ(tdir):
105
os.chdir("/")
106
shutil.rmtree(tdir)
107
108
def timport(tree, summary):
109
msg = tree.log_message()
110
msg["summary"] = summary
111
tree.import_(msg)
112
113
def commit(tree, summary):
114
"""
115
>>> q = test_environ()
116
>>> tree = pybaz.tree_root(".")
117
>>> timport(tree, "import")
118
>>> commit(tree, "commit")
119
>>> logs = [str(l.revision) for l in tree.iter_logs()]
120
>>> len(logs)
121
2
122
>>> logs[0]
123
'test@example.com/test--test--0--base-0'
124
>>> logs[1]
125
'test@example.com/test--test--0--patch-1'
126
>>> teardown_environ(q)
127
"""
128
msg = tree.log_message()
129
msg["summary"] = summary
130
tree.commit(msg)
131
132
def commit_test_revisions():
133
"""
134
>>> q = test_environ()
135
>>> commit_test_revisions()
136
>>> a = pybaz.Archive("test@example.com")
137
>>> revisions = list(a.iter_revisions("test--test--0"))
138
>>> len(revisions)
139
3
140
>>> str(revisions[2])
141
'test@example.com/test--test--0--base-0'
142
>>> str(revisions[1])
143
'test@example.com/test--test--0--patch-1'
144
>>> str(revisions[0])
145
'test@example.com/test--test--0--patch-2'
146
>>> teardown_environ(q)
147
"""
148
tree = pybaz.tree_root(".")
149
add_file("mainfile", "void main(void){}", "mainfile by aaron")
150
timport(tree, "Created mainfile")
151
file("mainfile", "wb").write("or something like that")
152
commit(tree, "altered mainfile")
153
add_file("ofile", "this is another file", "ofile by aaron")
154
commit(tree, "altered mainfile")
155
156
def version_ancestry(version):
157
"""
158
>>> q = test_environ()
159
>>> commit_test_revisions()
160
>>> version = pybaz.Version("test@example.com/test--test--0")
161
>>> ancestors = version_ancestry(version)
162
>>> str(ancestors[0])
163
'test@example.com/test--test--0--base-0'
164
>>> str(ancestors[1])
165
'test@example.com/test--test--0--patch-1'
166
>>> teardown_environ(q)
167
"""
168
revision = version.iter_revisions(reverse=True).next()
169
ancestors = list(revision.iter_ancestors(metoo=True))
170
ancestors.reverse()
171
return ancestors
172
173
174
def import_version(output_dir, version, fancy=True):
175
"""
176
>>> q = test_environ()
177
>>> result_path = os.path.join(q, "result")
178
>>> commit_test_revisions()
179
>>> version = pybaz.Version("test@example.com/test--test--0")
180
>>> import_version(result_path, version, fancy=False)
181
not fancy
182
....
183
Import complete.
184
>>> teardown_environ(q)
185
"""
186
progress_bar = ProgressBar()
187
tempdir = tempfile.mkdtemp(prefix="baz2bzr")
188
try:
189
if not fancy:
190
print "not fancy"
191
for result in iter_import_version(output_dir, version, tempdir):
192
if fancy:
193
progress_bar(result)
194
else:
195
sys.stdout.write('.')
196
finally:
197
if fancy:
198
clear_progress_bar()
199
else:
200
sys.stdout.write('\n')
201
shutil.rmtree(tempdir)
202
print "Import complete."
203
204
class UserError(Exception):
205
def __init__(self, message):
206
"""Exception to throw when a user makes an impossible request
207
:param message: The message to emit when printing this exception
208
:type message: string
209
"""
210
Exception.__init__(self, message)
211
212
def revision_id(arch_revision):
213
"""
214
Generate a Bzr revision id from an Arch revision id. 'x' in the id
215
designates a revision imported with an experimental algorithm. A number
216
would indicate a particular standardized version.
217
218
:param arch_revision: The Arch revision to generate an ID for.
219
220
>>> revision_id(pybaz.Revision("you@example.com/cat--br--0--base-0"))
221
'Arch-x:you@example.com%cat--br--0--base-0'
222
"""
223
return "Arch-x:%s" % str(arch_revision).replace('/', '%')
224
225
def iter_import_version(output_dir, version, tempdir):
226
revdir = None
227
ancestors = version_ancestry(version)
228
for i in range(len(ancestors)):
229
revision = ancestors[i]
230
yield Progress("revisions", i, len(ancestors))
231
if revdir is None:
232
revdir = os.path.join(tempdir, "rd")
233
baz_inv, log = get_revision(revdir, revision)
234
branch = bzrlib.Branch(revdir, init=True)
235
else:
236
old = os.path.join(revdir, ".bzr")
237
new = os.path.join(tempdir, ".bzr")
238
os.rename(old, new)
239
baz_inv, log = apply_revision(revdir, revision)
240
os.rename(new, old)
241
branch.unlock()
242
branch = bzrlib.Branch(revdir)
243
branch.set_inventory(baz_inv)
244
timestamp = email.Utils.mktime_tz(log.date + (0,))
245
rev_id = revision_id(revision)
246
bzrlib.trace.silent = True
247
try:
248
branch.commit(log.summary, verbose=False, committer=log.creator,
249
timestamp=timestamp, timezone=0, rev_id=rev_id)
250
finally:
251
bzrlib.trace.silent = False
252
yield Progress("revisions", len(ancestors), len(ancestors))
253
unlink_unversioned(branch, revdir)
254
os.rename(revdir, output_dir)
255
256
def unlink_unversioned(branch, revdir):
257
for unversioned in branch.working_tree().extras():
258
path = os.path.join(revdir, unversioned)
259
if os.path.isdir(path):
260
shutil.rmtree(path)
261
else:
262
os.unlink(path)
263
264
def get_log(tree, revision):
265
log = tree.iter_logs(version=revision.version, reverse=True).next()
266
assert log.revision == revision
267
return log
268
269
def get_revision(revdir, revision):
270
revision.get(revdir)
271
tree = pybaz.tree_root(revdir)
272
log = get_log(tree, revision)
273
try:
274
return bzr_inventory_data(tree), log
275
except BadFileKind, e:
276
raise UserError("Cannot convert %s because %s is a %s" % (revision,e.path, e.kind) )
277
278
279
def apply_revision(revdir, revision):
280
tree = pybaz.tree_root(revdir)
281
revision.apply(tree)
282
log = get_log(tree, revision)
283
try:
284
return bzr_inventory_data(tree), log
285
except BadFileKind, e:
286
raise UserError("Cannot convert %s because %s is a %s" % (revision,e.path, e.kind) )
287
288
289
290
291
class BadFileKind(Exception):
292
"""The file kind is not permitted in bzr inventories"""
293
def __init__(self, tree_root, path, kind):
294
self.tree_root = tree_root
295
self.path = path
296
self.kind = kind
297
Exception.__init__(self, "File %s is of forbidden type %s" %
298
(os.path.join(tree_root, path), kind))
299
300
def bzr_inventory_data(tree):
301
inv_iter = tree.iter_inventory_ids(source=True, both=True)
302
inv_map = {}
303
for file_id, path in inv_iter:
304
inv_map[path] = file_id
305
306
bzr_inv = []
307
for path, file_id in inv_map.iteritems():
308
full_path = os.path.join(tree, path)
309
kind = bzrlib.osutils.file_kind(full_path)
310
if kind not in ("file", "directory"):
311
raise BadFileKind(tree, path, kind)
312
parent_dir = os.path.dirname(path)
313
if parent_dir != "":
314
parent_id = inv_map[parent_dir]
315
else:
316
parent_id = bzrlib.inventory.ROOT_ID
317
bzr_inv.append((path, file_id, parent_id, kind))
318
bzr_inv.sort()
319
return bzr_inv
320
321
if len(sys.argv) == 2 and sys.argv[1] == "test":
322
print "Running tests"
323
import doctest
324
doctest.testmod()
325
elif len(sys.argv) == 3:
326
try:
327
output_dir = sys.argv[2]
328
if os.path.exists(output_dir):
329
raise UserError("Directory \"%s\" already exists" % output_dir)
330
import_version(output_dir, pybaz.Version(sys.argv[1]))
331
except UserError, e:
332
print e
333
except KeyboardInterrupt:
334
print "Aborted."
335
else:
336
print "usage: %s VERSION OUTDIR" % os.path.basename(sys.argv[0])
Loggerhead is a web-based interface for Breezy
Version: 2.0.1