14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
Stores are the main data-storage mechanism for Bazaar-NG.
17
"""Stores are the main data-storage mechanism for Bazaar-NG.
20
19
A store is a simple write-once container indexed by a universally
20
unique ID, which is typically the SHA-1 of the content."""
22
__copyright__ = "Copyright (C) 2005 Canonical Ltd."
23
__author__ = "Martin Pool <mbp@canonical.com>"
24
25
import os, tempfile, types, osutils, gzip, errno
25
26
from stat import ST_SIZE
26
27
from StringIO import StringIO
27
from bzrlib.errors import BzrError
28
from bzrlib.trace import mutter
28
from trace import mutter
31
30
######################################################################
61
60
TODO: Atomic add by writing to a temporary file and renaming.
63
In bzr 0.0.5 and earlier, files within the store were marked
64
readonly on disk. This is no longer done but existing stores need
62
TODO: Perhaps automatically transform to/from XML in a method?
63
Would just need to tell the constructor what class to
66
TODO: Even within a simple disk store like this, we could
67
gzip the files. But since many are less than one disk
68
block, that might not help a lot.
68
72
def __init__(self, basedir):
73
"""ImmutableStore constructor."""
69
74
self._basedir = basedir
71
def _path(self, entry_id):
72
if not isinstance(entry_id, basestring):
73
raise TypeError(type(entry_id))
74
if '\\' in entry_id or '/' in entry_id:
75
raise ValueError("invalid store id %r" % entry_id)
76
return os.path.join(self._basedir, entry_id)
78
return os.path.join(self._basedir, id)
78
80
def __repr__(self):
79
81
return "%s(%r)" % (self.__class__.__name__, self._basedir)
82
84
"""Add contents of a file into the store.
84
86
f -- An open file, or file-like object."""
85
# FIXME: Only works on files that will fit in memory
87
from bzrlib.atomicfile import AtomicFile
87
# FIXME: Only works on smallish files
88
# TODO: Can be optimized by copying at the same time as
89
90
mutter("add store entry %r" % (fileid))
90
91
if isinstance(f, types.StringTypes):
95
96
p = self._path(fileid)
96
97
if os.access(p, os.F_OK) or os.access(p + '.gz', os.F_OK):
97
raise BzrError("store %r already contains id %r" % (self._basedir, fileid))
98
bailout("store %r already contains id %r" % (self._basedir, fileid))
101
f = gzip.GzipFile(p + '.gz', 'wb')
102
os.chmod(p + '.gz', 0444)
103
af = AtomicFile(fn, 'wb')
106
gf = gzip.GzipFile(mode='wb', fileobj=af)
116
def copy_multi(self, other, ids, permit_failure=False):
110
def copy_multi(self, other, ids):
117
111
"""Copy texts for ids from other into self.
119
If an id is present in self, it is skipped.
121
Returns (count_copied, failed), where failed is a collection of ids
122
that could not be copied.
113
If an id is present in self, it is skipped. A count of copied
114
ids is returned, which may be less than len(ids).
124
pb = bzrlib.ui.ui_factory.progress_bar()
126
pb.update('preparing to copy')
127
to_copy = [id for id in ids if id not in self]
128
if isinstance(other, ImmutableStore):
129
return self.copy_multi_immutable(other, to_copy, pb)
134
pb.update('copy', count, len(to_copy))
135
if not permit_failure:
136
self.add(other[id], id)
145
if not permit_failure:
146
assert count == len(to_copy)
150
def copy_multi_immutable(self, other, to_copy, pb, permit_failure=False):
151
from shutil import copyfile
156
other_p = other._path(id)
160
if e.errno == errno.ENOENT:
161
if not permit_failure:
162
copyfile(other_p+".gz", p+".gz")
165
copyfile(other_p+".gz", p+".gz")
167
if e.errno == errno.ENOENT:
175
pb.update('copy', count, len(to_copy))
176
assert count == len(to_copy)
120
self.add(other[id], id)
181
124
def __contains__(self, fileid):
197
140
def __len__(self):
198
141
return len(os.listdir(self._basedir))
201
143
def __getitem__(self, fileid):
202
144
"""Returns a file reading from a particular entry."""
203
145
p = self._path(fileid)
205
147
return gzip.GzipFile(p + '.gz', 'rb')
206
148
except IOError, e:
207
if e.errno != errno.ENOENT:
213
if e.errno != errno.ENOENT:
216
raise IndexError(fileid)
149
if e.errno == errno.ENOENT:
219
154
def total_size(self):
220
155
"""Return (count, bytes)
added
removed
bzrlib/store.py
