/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/store.py

  • Committer: Martin Pool
  • Date: 2005-09-23 05:57:06 UTC
  • Revision ID: mbp@sourcefrog.net-20050923055706-afd25ee2a988286d
- update NEWS files

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/store.py
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
16
16
 
 
17
# TODO: Could remember a bias towards whether a particular store is typically
 
18
# compressed or not.
 
19
 
17
20
"""
18
21
Stores are the main data-storage mechanism for Bazaar-NG.
19
22
 
24
27
import os, tempfile, types, osutils, gzip, errno
25
28
from stat import ST_SIZE
26
29
from StringIO import StringIO
27
 
from trace import mutter
 
30
from bzrlib.errors import BzrError
 
31
from bzrlib.trace import mutter
 
32
import bzrlib.ui
28
33
 
29
34
######################################################################
30
35
# stores
66
71
    def __init__(self, basedir):
67
72
        self._basedir = basedir
68
73
 
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)
 
74
    def _path(self, entry_id):
 
75
        if not isinstance(entry_id, basestring):
 
76
            raise TypeError(type(entry_id))
 
77
        if '\\' in entry_id or '/' in entry_id:
 
78
            raise ValueError("invalid store id %r" % entry_id)
 
79
        return os.path.join(self._basedir, entry_id)
73
80
 
74
81
    def __repr__(self):
75
82
        return "%s(%r)" % (self.__class__.__name__, self._basedir)
90
97
            
91
98
        p = self._path(fileid)
92
99
        if os.access(p, os.F_OK) or os.access(p + '.gz', os.F_OK):
93
 
            from bzrlib.errors import bailout
94
100
            raise BzrError("store %r already contains id %r" % (self._basedir, fileid))
95
101
 
96
102
        fn = p
110
116
            af.close()
111
117
 
112
118
 
113
 
    def copy_multi(self, other, ids):
 
119
    def copy_multi(self, other, ids, permit_failure=False):
114
120
        """Copy texts for ids from other into self.
115
121
 
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).
 
122
        If an id is present in self, it is skipped.
 
123
 
 
124
        Returns (count_copied, failed), where failed is a collection of ids
 
125
        that could not be copied.
118
126
        """
119
 
        from bzrlib.progress import ProgressBar
120
 
        pb = ProgressBar()
 
127
        pb = bzrlib.ui.ui_factory.progress_bar()
 
128
        
121
129
        pb.update('preparing to copy')
122
130
        to_copy = [id for id in ids if id not in self]
123
131
        if isinstance(other, ImmutableStore):
124
132
            return self.copy_multi_immutable(other, to_copy, pb)
125
133
        count = 0
 
134
        failed = set()
126
135
        for id in to_copy:
127
136
            count += 1
128
137
            pb.update('copy', count, len(to_copy))
129
 
            self.add(other[id], id)
130
 
        assert count == len(to_copy)
 
138
            if not permit_failure:
 
139
                self.add(other[id], id)
 
140
            else:
 
141
                try:
 
142
                    entry = other[id]
 
143
                except IndexError:
 
144
                    failed.add(id)
 
145
                    continue
 
146
                self.add(entry, id)
 
147
                
 
148
        if not permit_failure:
 
149
            assert count == len(to_copy)
131
150
        pb.clear()
132
 
        return count
133
 
 
134
 
 
135
 
    def copy_multi_immutable(self, other, to_copy, pb):
 
151
        return count, failed
 
152
 
 
153
    def copy_multi_immutable(self, other, to_copy, pb, permit_failure=False):
136
154
        from shutil import copyfile
137
155
        count = 0
 
156
        failed = set()
138
157
        for id in to_copy:
139
158
            p = self._path(id)
140
159
            other_p = other._path(id)
142
161
                copyfile(other_p, p)
143
162
            except IOError, e:
144
163
                if e.errno == errno.ENOENT:
145
 
                    copyfile(other_p+".gz", p+".gz")
 
164
                    if not permit_failure:
 
165
                        copyfile(other_p+".gz", p+".gz")
 
166
                    else:
 
167
                        try:
 
168
                            copyfile(other_p+".gz", p+".gz")
 
169
                        except IOError, e:
 
170
                            if e.errno == errno.ENOENT:
 
171
                                failed.add(id)
 
172
                            else:
 
173
                                raise
146
174
                else:
147
175
                    raise
148
176
            
150
178
            pb.update('copy', count, len(to_copy))
151
179
        assert count == len(to_copy)
152
180
        pb.clear()
153
 
        return count
 
181
        return count, failed
154
182
    
155
183
 
156
184
    def __contains__(self, fileid):
159
187
        return (os.access(p, os.R_OK)
160
188
                or os.access(p + '.gz', os.R_OK))
161
189
 
162
 
    # TODO: Guard against the same thing being stored twice, compressed and uncompresse
 
190
    # TODO: Guard against the same thing being stored twice,
 
191
    # compressed and uncompressed
163
192
 
164
193
    def __iter__(self):
165
194
        for f in os.listdir(self._basedir):