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
199
200
201
202
|
#! /usr/bin/python
# 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
# XXX: Some consideration of the problems that might occur if there are
# files whose id differs only in case. That should probably be forbidden.
import errno
import os
from cStringIO import StringIO
import urllib
from bzrlib.weavefile import read_weave, write_weave_v5
from bzrlib.weave import WeaveFile
from bzrlib.store import TransportStore, hash_prefix
from bzrlib.atomicfile import AtomicFile
from bzrlib.errors import NoSuchFile, FileExists
from bzrlib.symbol_versioning import *
from bzrlib.trace import mutter
class VersionedFileStore(TransportStore):
"""Collection of many versioned files in a transport."""
def __init__(self, transport, prefixed=False, precious=False,
dir_mode=None, file_mode=None,
versionedfile_class=WeaveFile):
super(WeaveStore, self).__init__(transport,
dir_mode=dir_mode, file_mode=file_mode,
prefixed=prefixed, compressed=False)
self._precious = precious
self._versionedfile_class = versionedfile_class
def filename(self, file_id):
"""Return the path relative to the transport root."""
if self._prefixed:
return hash_prefix(file_id) + urllib.quote(file_id)
else:
return urllib.quote(file_id)
def __iter__(self):
suffixes = self._versionedfile_class.get_suffixes()
ids = set()
for relpath in self._iter_files_recursive():
for suffix in suffixes:
if relpath.endswith(suffix):
id = os.path.basename(relpath[:-len(suffix)])
if not id in ids:
yield id
ids.add(id)
def has_id(self, fileid):
suffixes = self._versionedfile_class.get_suffixes()
filename = self.filename(fileid)
for suffix in suffixes:
if not self._transport.has(filename + suffix):
return False
return True
def get_weave(self, file_id, transaction):
weave = transaction.map.find_weave(file_id)
if weave:
mutter("cache hit in %s for %s", self, file_id)
return weave
w = self._versionedfile_class(self.filename(file_id), self._transport, self._file_mode)
transaction.map.add_weave(file_id, w)
transaction.register_clean(w, precious=self._precious)
return w
@deprecated_method(zero_eight)
def get_lines(self, file_id, rev_id, transaction):
"""Return text from a particular version of a weave.
Returned as a list of lines."""
w = self.get_weave(file_id, transaction)
return w.get_lines(rev_id)
def _new_weave(self, file_id, transaction):
"""Make a new weave for file_id and return it."""
weave = self._make_new_versionedfile(file_id)
transaction.map.add_weave(file_id, weave)
transaction.register_clean(weave, precious=self._precious)
return weave
def _make_new_versionedfile(self, file_id):
try:
weave = self._versionedfile_class(self.filename(file_id), self._transport, self._file_mode)
except NoSuchFile:
if not self._prefixed:
# unexpected error - NoSuchFile is raised on a missing dir only and that
# only occurs when we are prefixed.
raise
self._transport.mkdir(hash_prefix(file_id), mode=self._dir_mode)
weave = self._versionedfile_class(self.filename(file_id), self._transport, self._file_mode)
return weave
def get_weave_or_empty(self, file_id, transaction):
"""Return a weave, or an empty one if it doesn't exist."""
try:
return self.get_weave(file_id, transaction)
except NoSuchFile:
return self._new_weave(file_id, transaction)
@deprecated_method(zero_eight)
def put_weave(self, file_id, weave, transaction):
"""This is a deprecated API: It writes an entire collection of ids out.
This became inappropriate when we made a versioned file api which
tracks the state of the collection of versions for a single id.
Its maintained for backwards compatability but will only work on
weave stores - pre 0.8 repositories.
"""
self._put_weave(self, file_id, weave, transaction)
def _put_weave(self, file_id, weave, transaction):
"""Preserved here for upgrades-to-weaves to use."""
myweave = self._make_new_versionedfile(file_id)
myweave.join(weave)
@deprecated_method(zero_eight)
def add_text(self, file_id, rev_id, new_lines, parents, transaction):
"""This method was a shorthand for
vfile = self.get_weave_or_empty(file_id, transaction)
vfile.add_lines(rev_id, parents, new_lines)
"""
vfile = self.get_weave_or_empty(file_id, transaction)
vfile.add_lines(rev_id, parents, new_lines)
@deprecated_method(zero_eight)
def add_identical_text(self, file_id, old_rev_id, new_rev_id, parents,
transaction):
"""This method was a shorthand for
vfile = self.get_weave_or_empty(file_id, transaction)
vfile.clone_text(new_rev_id, old_rev_id, parents)
"""
vfile = self.get_weave_or_empty(file_id, transaction)
vfile.clone_text(new_rev_id, old_rev_id, parents)
def copy_all_ids(self, store_from, pb=None, from_transaction=None):
"""Copy all the file ids from store_from into self."""
if from_transaction is None:
warn("Please pase from_transaction into "
"versioned_store.copy_all_ids.", stacklevel=2)
if not store_from.listable():
raise UnlistableStore(store_from)
ids = []
for count, file_id in enumerate(store_from):
if pb:
pb.update('listing files', count, count)
ids.append(file_id)
if pb:
pb.clear()
mutter('copy_all ids: %r', ids)
self.copy_multi(store_from, ids, pb=pb,
from_transaction=from_transaction)
def copy_multi(self, from_store, file_ids, pb=None, from_transaction=None):
"""Copy all the versions for multiple file_ids from from_store.
:param from_transaction: required current transaction in from_store.
"""
assert isinstance(from_store, WeaveStore)
if from_transaction is None:
warn("WeaveStore.copy_multi without a from_transaction parameter "
"is deprecated. Please provide a from_transaction.",
DeprecationWarning,
stacklevel=2)
for count, f in enumerate(file_ids):
mutter("copy weave {%s} into %s", f, self)
if pb:
pb.update('copy', count, len(file_ids))
# if we have it in cache, its faster.
if not from_transaction:
from bzrlib.transactions import PassThroughTransaction
from_transaction = PassThroughTransaction()
# joining is fast with knits, and bearable for weaves -
# indeed the new case can be optimised
target = self._make_new_versionedfile(f)
target.join(from_store.get_weave(f, from_transaction))
if pb:
pb.clear()
WeaveStore = VersionedFileStore
|