← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/store.py

  • Committer: Martin Pool
  • Date: 2005-07-04 12:26:02 UTC
  • Revision ID: mbp@sourcefrog.net-20050704122602-69901910521e62c3
- check command checks that all inventory-ids are the same as in the revision.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/store.py
24
24
import os, tempfile, types, osutils, gzip, errno
25
25
from stat import ST_SIZE
26
26
from StringIO import StringIO
27
 
from bzrlib.errors import BzrError
28
 
from bzrlib.trace import mutter
29
 
import bzrlib.ui
 
27
from trace import mutter
30
28
 
31
29
######################################################################
32
30
# stores
68
66
    def __init__(self, basedir):
69
67
        self._basedir = basedir
70
68
 
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)
 
69
    def _path(self, id):
 
70
        if '\\' in id or '/' in id:
 
71
            raise ValueError("invalid store id %r" % id)
 
72
        return os.path.join(self._basedir, id)
77
73
 
78
74
    def __repr__(self):
79
75
        return "%s(%r)" % (self.__class__.__name__, self._basedir)
113
109
            af.close()
114
110
 
115
111
 
116
 
    def copy_multi(self, other, ids, permit_failure=False):
 
112
    def copy_multi(self, other, ids):
117
113
        """Copy texts for ids from other into self.
118
114
 
119
 
        If an id is present in self, it is skipped.
120
 
 
121
 
        Returns (count_copied, failed), where failed is a collection of ids
122
 
        that could not be copied.
 
115
        If an id is present in self, it is skipped.  A count of copied
 
116
        ids is returned, which may be less than len(ids).
123
117
        """
124
 
        pb = bzrlib.ui.ui_factory.progress_bar()
125
 
        
 
118
        from bzrlib.progress import ProgressBar
 
119
        pb = ProgressBar()
126
120
        pb.update('preparing to copy')
127
121
        to_copy = [id for id in ids if id not in self]
128
122
        if isinstance(other, ImmutableStore):
129
123
            return self.copy_multi_immutable(other, to_copy, pb)
130
124
        count = 0
131
 
        failed = set()
132
125
        for id in to_copy:
133
126
            count += 1
134
127
            pb.update('copy', count, len(to_copy))
135
 
            if not permit_failure:
136
 
                self.add(other[id], id)
137
 
            else:
138
 
                try:
139
 
                    entry = other[id]
140
 
                except IndexError:
141
 
                    failed.add(id)
142
 
                    continue
143
 
                self.add(entry, id)
144
 
                
145
 
        if not permit_failure:
146
 
            assert count == len(to_copy)
 
128
            self.add(other[id], id)
 
129
        assert count == len(to_copy)
147
130
        pb.clear()
148
 
        return count, failed
149
 
 
150
 
    def copy_multi_immutable(self, other, to_copy, pb, permit_failure=False):
 
131
        return count
 
132
 
 
133
 
 
134
    def copy_multi_immutable(self, other, to_copy, pb):
151
135
        from shutil import copyfile
152
136
        count = 0
153
 
        failed = set()
154
137
        for id in to_copy:
155
138
            p = self._path(id)
156
139
            other_p = other._path(id)
158
141
                copyfile(other_p, p)
159
142
            except IOError, e:
160
143
                if e.errno == errno.ENOENT:
161
 
                    if not permit_failure:
162
 
                        copyfile(other_p+".gz", p+".gz")
163
 
                    else:
164
 
                        try:
165
 
                            copyfile(other_p+".gz", p+".gz")
166
 
                        except IOError, e:
167
 
                            if e.errno == errno.ENOENT:
168
 
                                failed.add(id)
169
 
                            else:
170
 
                                raise
 
144
                    copyfile(other_p+".gz", p+".gz")
171
145
                else:
172
146
                    raise
173
147
            
175
149
            pb.update('copy', count, len(to_copy))
176
150
        assert count == len(to_copy)
177
151
        pb.clear()
178
 
        return count, failed
 
152
        return count
179
153
    
180
154
 
181
155
    def __contains__(self, fileid):
197
171
    def __len__(self):
198
172
        return len(os.listdir(self._basedir))
199
173
 
200
 
 
201
174
    def __getitem__(self, fileid):
202
175
        """Returns a file reading from a particular entry."""
203
176
        p = self._path(fileid)
204
177
        try:
205
178
            return gzip.GzipFile(p + '.gz', 'rb')
206
179
        except IOError, e:
207
 
            if e.errno != errno.ENOENT:
208
 
                raise
209
 
 
210
 
        try:
211
 
            return file(p, 'rb')
212
 
        except IOError, e:
213
 
            if e.errno != errno.ENOENT:
214
 
                raise
215
 
 
216
 
        raise IndexError(fileid)
217
 
 
 
180
            if e.errno == errno.ENOENT:
 
181
                return file(p, 'rb')
 
182
            else:
 
183
                raise e
218
184
 
219
185
    def total_size(self):
220
186
        """Return (count, bytes)