/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/remotebranch.py

  • Committer: Martin Pool
  • Date: 2005-05-18 12:08:22 UTC
  • Revision ID: mbp@sourcefrog.net-20050518120821-6f80eab9f273edcf
- fix import

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
25
25
 
26
26
 
27
27
import gzip
 
28
from sets import Set
28
29
from cStringIO import StringIO
29
30
import urllib2
30
31
 
38
39
 
39
40
ENABLE_URLGRABBER = True
40
41
 
41
 
from bzrlib.errors import BzrError
42
 
 
43
 
class GetFailed(BzrError):
44
 
    def __init__(self, url, status):
45
 
        BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
46
 
        self.url = url
47
 
        self.status = status
48
42
 
49
43
if ENABLE_URLGRABBER:
50
44
    import urlgrabber
57
51
                url += '.gz'
58
52
            mutter("grab url %s" % url)
59
53
            url_f = urlgrabber.urlopen(url, keepalive=1, close_connection=0)
60
 
            if url_f.status != 200:
61
 
                raise GetFailed(url, url_f.status)
62
54
            if not compressed:
63
55
                return url_f
64
56
            else:
108
100
 
109
101
 
110
102
class RemoteBranch(Branch):
111
 
    def __init__(self, baseurl, find_root=True):
 
103
    def __init__(self, baseurl, find_root=True, lock_mode='r'):
112
104
        """Create new proxy for a remote branch."""
 
105
        if lock_mode not in ('', 'r'):
 
106
            raise BzrError('lock mode %r is not supported for remote branches'
 
107
                           % lock_mode)
 
108
 
113
109
        if find_root:
114
110
            self.baseurl = _find_remote_root(baseurl)
115
111
        else:
118
114
 
119
115
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
120
116
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
121
 
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
122
117
 
123
118
    def __str__(self):
124
 
        b = getattr(self, 'baseurl', 'undefined')
125
 
        return '%s(%r)' % (self.__class__.__name__, b)
 
119
        return '%s(%r)' % (self.__class__.__name__, self.baseurl)
126
120
 
127
121
    __repr__ = __str__
128
122
 
131
125
            raise BzrError("file mode %r not supported for remote branches" % mode)
132
126
        return get_url(self.baseurl + '/.bzr/' + filename, False)
133
127
 
134
 
 
135
 
    def lock_read(self):
136
 
        # no locking for remote branches yet
137
 
        pass
138
 
 
139
 
    def lock_write(self):
140
 
        from errors import LockError
141
 
        raise LockError("write lock not supported for remote branch %s"
142
 
                        % self.baseurl)
143
 
 
144
 
    def unlock(self):
145
 
        pass
146
 
    
 
128
    def _need_readlock(self):
 
129
        # remote branch always safe for read
 
130
        pass
 
131
 
 
132
    def _need_writelock(self):
 
133
        raise BzrError("cannot get write lock on HTTP remote branch")
147
134
 
148
135
    def relpath(self, path):
149
136
        if not path.startswith(self.baseurl):
152
139
        pl = len(self.baseurl)
153
140
        return path[pl:].lstrip('/')
154
141
 
155
 
 
156
142
    def get_revision(self, revision_id):
157
 
        from bzrlib.revision import Revision
158
 
        from bzrlib.xml import unpack_xml
159
 
        revf = self.revision_store[revision_id]
160
 
        r = unpack_xml(Revision, revf)
 
143
        from revision import Revision
 
144
        revf = get_url(self.baseurl + '/.bzr/revision-store/' + revision_id,
 
145
                       True)
 
146
        r = Revision.read_xml(revf)
161
147
        if r.revision_id != revision_id:
162
148
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
163
149
                                % (revision_id, r.revision_id))
164
150
        return r
165
151
 
166
152
 
167
 
class RemoteStore(object):
 
153
class RemoteStore:
168
154
    def __init__(self, baseurl):
169
155
        self._baseurl = baseurl
170
156
        
176
162
        
177
163
    def __getitem__(self, fileid):
178
164
        p = self._path(fileid)
179
 
        try:
180
 
            return get_url(p, compressed=True)
181
 
        except:
182
 
            raise KeyError(fileid)
 
165
        return get_url(p, compressed=True)
183
166
    
184
167
 
 
168
def simple_walk():
 
169
    """For experimental purposes, traverse many parts of a remote branch"""
 
170
    from revision import Revision
 
171
    from branch import Branch
 
172
    from inventory import Inventory
 
173
 
 
174
    got_invs = Set()
 
175
    got_texts = Set()
 
176
 
 
177
    print 'read history'
 
178
    history = get_url('/.bzr/revision-history').readlines()
 
179
    num_revs = len(history)
 
180
    for i, rev_id in enumerate(history):
 
181
        rev_id = rev_id.rstrip()
 
182
        print 'read revision %d/%d' % (i, num_revs)
 
183
 
 
184
        # python gzip needs a seekable file (!!) but the HTTP response
 
185
        # isn't, so we need to buffer it
 
186
 
 
187
        rev_f = get_url('/.bzr/revision-store/%s' % rev_id,
 
188
                        compressed=True)
 
189
 
 
190
        rev = Revision.read_xml(rev_f)
 
191
        print rev.message
 
192
        inv_id = rev.inventory_id
 
193
        if inv_id not in got_invs:
 
194
            print 'get inventory %s' % inv_id
 
195
            inv_f = get_url('/.bzr/inventory-store/%s' % inv_id,
 
196
                            compressed=True)
 
197
            inv = Inventory.read_xml(inv_f)
 
198
            print '%4d inventory entries' % len(inv)
 
199
 
 
200
            for path, ie in inv.iter_entries():
 
201
                text_id = ie.text_id
 
202
                if text_id == None:
 
203
                    continue
 
204
                if text_id in got_texts:
 
205
                    continue
 
206
                print '  fetch %s text {%s}' % (path, text_id)
 
207
                text_f = get_url('/.bzr/text-store/%s' % text_id,
 
208
                                 compressed=True)
 
209
                got_texts.add(text_id)
 
210
 
 
211
            got_invs.add(inv_id)
 
212
 
 
213
        print '----'
 
214
 
 
215
 
 
216
def try_me():
 
217
    BASE_URL = 'http://bazaar-ng.org/bzr/bzr.dev/'
 
218
    b = RemoteBranch(BASE_URL)
 
219
    ## print '\n'.join(b.revision_history())
 
220
    from log import show_log
 
221
    show_log(b)
 
222
 
 
223
 
 
224
if __name__ == '__main__':
 
225
    try_me()
185
226