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
|
from bzrlib.lazy_import import lazy_import
lazy_import(globals(), """
import (
errno,
os,
sys,
time,
)
from bzrlib import (
commands,
urlutils
)
from bzrlib.workingtree import WorkingTree
from bzrlib.tests import TestUtil
from bzrlib.plugins.multiparent.multiparent import (
MultiVersionedFile,
)
""")
class cmd_mp_regen(commands.Command):
"""Generate a multiparent versionedfile"""
takes_args = ['file?']
takes_options = [commands.Option('sync-snapshots',
help='Snapshots follow source'),
commands.Option('snapshot-interval', type=int,
help='take snapshots every x revisions'),
commands.Option('lsprof-timed', help='Use lsprof'),
commands.Option('outfile', type=unicode,
help='Write pseudo-knit to this file'),
commands.Option('extract', help='test extract time'),
commands.Option('single', help='use a single parent'),
commands.Option('verify', help='verify added texts'),
]
hidden = True
def run(self, file=None, sync_snapshots=False, snapshot_interval=26,
lsprof_timed=False, dump=False, extract=False, single=False,
verify=False, outfile=None):
if outfile is None:
filename = 'pknit'
else:
filename = outfile
if file is None:
wt, path = WorkingTree.open_containing('.')
file_weave = wt.branch.repository.get_inventory_weave()
else:
wt, path = WorkingTree.open_containing(file)
file_id = wt.path2id(path)
bt = wt.branch.repository.revision_tree(wt.last_revision())
file_weave = bt.get_weave(file_id)
url = file_weave.transport.abspath(file_weave.filename)
print >> sys.stderr, 'Importing: %s' % \
urlutils.local_path_from_url(url)
if sync_snapshots:
print >> sys.stderr, 'Snapshots follow input'
else:
print >> sys.stderr, 'Snapshot interval: %d' % snapshot_interval
try:
os.unlink(filename)
except OSError, e:
if e.errno != errno.ENOENT:
raise
vf = MultiVersionedFile(filename, snapshot_interval)
snapshots = set(r for r in file_weave.versions() if
file_weave._index.get_method(r) == 'fulltext')
if sync_snapshots:
to_sync = snapshots
else:
to_sync = vf.select_snapshots(file_weave)
print >> sys.stderr, "%d fulltexts" % len(snapshots)
print >> sys.stderr, "%d planned snapshots" % len(to_sync)
try:
vf.import_versionedfile(file_weave, to_sync, single_parent=single,
verify=verify)
except:
try:
os.unlink(filename)
except OSError, e:
if e.errno != errno.ENOENT:
raise
raise
try:
print >> sys.stderr, "%d actual snapshots" % len(to_sync)
vf.clear_cache()
if False:
for revision_id in file_weave.get_ancestry(
[bt.inventory[file_id].revision]):
if vf.get_line_list([revision_id])[0] != \
file_weave.get_lines(revision_id):
open(revision_id + '.old', 'wb').writelines(
file_weave.get_lines(revision_id))
open(revision_id + '.new', 'wb').writelines(
vf.get_line_list(revision_id)[0])
if extract:
revisions = file_weave.versions()[-1:]
if lsprof_timed:
from bzrlib.lsprof import profile
ret, stats = profile(vf.get_line_list, revisions)
stats.sort()
stats.pprint()
start = time.clock()
print >> sys.stderr, revisions[0]
for x in range(1000):
vf.clear_cache()
vf.get_line_list(revisions)
print >> sys.stderr, time.clock() - start
start = time.clock()
for x in range(1000):
file_weave.get_line_list(revisions)
print >> sys.stderr, time.clock() - start
finally:
if outfile is None:
try:
os.unlink()
except OSError, e:
if e.errno != ENOENT:
raise
commands.register_command(cmd_mp_regen)
def test_suite():
from bzrlib.plugins.multiparent import test_multiparent
return TestUtil.TestLoader().loadTestsFromModule(test_multiparent)
|