← 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-06-22 06:37:43 UTC
  • Revision ID: mbp@sourcefrog.net-20050622063743-e395f04c4db8977f
- move old blackbox code from testbzr into bzrlib.selftest.blackbox

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.trace import mutter
28
 
import bzrlib.ui
 
27
from trace import mutter
29
28
 
30
29
######################################################################
31
30
# stores
91
90
            
92
91
        p = self._path(fileid)
93
92
        if os.access(p, os.F_OK) or os.access(p + '.gz', os.F_OK):
94
 
            from bzrlib.errors import bailout
95
93
            raise BzrError("store %r already contains id %r" % (self._basedir, fileid))
96
94
 
97
95
        fn = p
111
109
            af.close()
112
110
 
113
111
 
114
 
    def copy_multi(self, other, ids, permit_failure=False):
 
112
    def copy_multi(self, other, ids):
115
113
        """Copy texts for ids from other into self.
116
114
 
117
115
        If an id is present in self, it is skipped.  A count of copied
118
116
        ids is returned, which may be less than len(ids).
119
117
        """
120
 
        pb = bzrlib.ui.ui_factory.progress_bar()
121
 
        
 
118
        from bzrlib.progress import ProgressBar
 
119
        pb = ProgressBar()
122
120
        pb.update('preparing to copy')
123
121
        to_copy = [id for id in ids if id not in self]
124
 
        if isinstance(other, ImmutableStore):
125
 
            return self.copy_multi_immutable(other, to_copy, pb)
126
122
        count = 0
127
123
        for id in to_copy:
128
124
            count += 1
129
125
            pb.update('copy', count, len(to_copy))
130
 
            if not permit_failure:
131
 
                self.add(other[id], id)
132
 
            else:
133
 
                try:
134
 
                    entry = other[id]
135
 
                except IndexError:
136
 
                    failures.add(id)
137
 
                    continue
138
 
                self.add(entry, id)
139
 
                
 
126
            self.add(other[id], id)
140
127
        assert count == len(to_copy)
141
128
        pb.clear()
142
129
        return count
143
 
 
144
 
 
145
 
    def copy_multi_immutable(self, other, to_copy, pb, permit_failure=False):
146
 
        from shutil import copyfile
147
 
        count = 0
148
 
        failed = set()
149
 
        for id in to_copy:
150
 
            p = self._path(id)
151
 
            other_p = other._path(id)
152
 
            try:
153
 
                copyfile(other_p, p)
154
 
            except IOError, e:
155
 
                if e.errno == errno.ENOENT:
156
 
                    if not permit_failure:
157
 
                        copyfile(other_p+".gz", p+".gz")
158
 
                    else:
159
 
                        try:
160
 
                            copyfile(other_p+".gz", p+".gz")
161
 
                        except IOError, e:
162
 
                            if e.errno == errno.ENOENT:
163
 
                                failed.add(id)
164
 
                            else:
165
 
                                raise
166
 
                else:
167
 
                    raise
168
 
            
169
 
            count += 1
170
 
            pb.update('copy', count, len(to_copy))
171
 
        assert count == len(to_copy)
172
 
        pb.clear()
173
 
        return count, failed
174
130
    
175
131
 
176
132
    def __contains__(self, fileid):
192
148
    def __len__(self):
193
149
        return len(os.listdir(self._basedir))
194
150
 
195
 
 
196
151
    def __getitem__(self, fileid):
197
152
        """Returns a file reading from a particular entry."""
198
153
        p = self._path(fileid)
199
154
        try:
200
155
            return gzip.GzipFile(p + '.gz', 'rb')
201
156
        except IOError, e:
202
 
            if e.errno != errno.ENOENT:
203
 
                raise
204
 
 
205
 
        try:
206
 
            return file(p, 'rb')
207
 
        except IOError, e:
208
 
            if e.errno != errno.ENOENT:
209
 
                raise
210
 
 
211
 
        raise IndexError(fileid)
212
 
 
 
157
            if e.errno == errno.ENOENT:
 
158
                return file(p, 'rb')
 
159
            else:
 
160
                raise e
213
161
 
214
162
    def total_size(self):
215
163
        """Return (count, bytes)