← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/store.py

  • Committer: Aaron Bentley
  • Date: 2005-09-12 13:48:32 UTC
  • mfrom: (1185.3.4)
  • mto: (1185.1.16)
  • mto: This revision was merged to the branch mainline in revision 1390.
  • Revision ID: abentley@panoramicfeedback.com-20050912134832-c23db11dc63170b6
Merged from mpool

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 trace import mutter
 
27
from bzrlib.errors import BzrError
 
28
from bzrlib.trace import mutter
 
29
import bzrlib.ui
28
30
 
29
31
######################################################################
30
32
# stores
66
68
    def __init__(self, basedir):
67
69
        self._basedir = basedir
68
70
 
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)
 
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)
73
77
 
74
78
    def __repr__(self):
75
79
        return "%s(%r)" % (self.__class__.__name__, self._basedir)
90
94
            
91
95
        p = self._path(fileid)
92
96
        if os.access(p, os.F_OK) or os.access(p + '.gz', os.F_OK):
93
 
            from bzrlib.errors import bailout
94
97
            raise BzrError("store %r already contains id %r" % (self._basedir, fileid))
95
98
 
96
99
        fn = p
110
113
            af.close()
111
114
 
112
115
 
113
 
    def copy_multi(self, other, ids):
 
116
    def copy_multi(self, other, ids, permit_failure=False):
114
117
        """Copy texts for ids from other into self.
115
118
 
116
 
        If an id is present in self, it is skipped.  A count of copied
117
 
        ids is returned, which may be less than len(ids).
 
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.
118
123
        """
119
 
        from bzrlib.progress import ProgressBar
120
 
        pb = ProgressBar()
 
124
        pb = bzrlib.ui.ui_factory.progress_bar()
 
125
        
121
126
        pb.update('preparing to copy')
122
127
        to_copy = [id for id in ids if id not in self]
123
128
        if isinstance(other, ImmutableStore):
124
129
            return self.copy_multi_immutable(other, to_copy, pb)
125
130
        count = 0
 
131
        failed = set()
126
132
        for id in to_copy:
127
133
            count += 1
128
134
            pb.update('copy', count, len(to_copy))
129
 
            self.add(other[id], id)
130
 
        assert 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)
131
147
        pb.clear()
132
 
        return count
133
 
 
134
 
 
135
 
    def copy_multi_immutable(self, other, to_copy, pb):
 
148
        return count, failed
 
149
 
 
150
    def copy_multi_immutable(self, other, to_copy, pb, permit_failure=False):
136
151
        from shutil import copyfile
137
152
        count = 0
 
153
        failed = set()
138
154
        for id in to_copy:
139
155
            p = self._path(id)
140
156
            other_p = other._path(id)
142
158
                copyfile(other_p, p)
143
159
            except IOError, e:
144
160
                if e.errno == errno.ENOENT:
145
 
                    copyfile(other_p+".gz", p+".gz")
 
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
146
171
                else:
147
172
                    raise
148
173
            
150
175
            pb.update('copy', count, len(to_copy))
151
176
        assert count == len(to_copy)
152
177
        pb.clear()
153
 
        return count
 
178
        return count, failed
154
179
    
155
180
 
156
181
    def __contains__(self, fileid):
172
197
    def __len__(self):
173
198
        return len(os.listdir(self._basedir))
174
199
 
 
200
 
175
201
    def __getitem__(self, fileid):
176
202
        """Returns a file reading from a particular entry."""
177
203
        p = self._path(fileid)
178
204
        try:
179
205
            return gzip.GzipFile(p + '.gz', 'rb')
180
206
        except IOError, e:
181
 
            if e.errno == errno.ENOENT:
182
 
                return file(p, 'rb')
183
 
            else:
184
 
                raise 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
 
185
218
 
186
219
    def total_size(self):
187
220
        """Return (count, bytes)