← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/remotebranch.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/remotebranch.py
28
28
from cStringIO import StringIO
29
29
import urllib2
30
30
 
31
 
from bzrlib.errors import BzrError, BzrCheckError
32
 
from bzrlib.branch import Branch, BZR_BRANCH_FORMAT_5
33
 
from bzrlib.trace import mutter
34
 
from bzrlib.xml5 import serializer_v5
 
31
from errors import BzrError, BzrCheckError
 
32
from branch import Branch, BZR_BRANCH_FORMAT
 
33
from trace import mutter
35
34
 
36
35
# velocitynet.com.au transparently proxies connections and thereby
37
36
# breaks keep-alive -- sucks!
38
37
 
39
38
 
40
 
ENABLE_URLGRABBER = False
41
 
 
42
 
from bzrlib.errors import BzrError, NoSuchRevision
43
 
 
44
 
class GetFailed(BzrError):
45
 
    def __init__(self, url, status):
46
 
        BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
47
 
        self.url = url
48
 
        self.status = status
 
39
ENABLE_URLGRABBER = True
 
40
 
49
41
 
50
42
if ENABLE_URLGRABBER:
51
 
    import util.urlgrabber
52
 
    import util.urlgrabber.keepalive
53
 
    util.urlgrabber.keepalive.DEBUG = 0
 
43
    import urlgrabber
 
44
    import urlgrabber.keepalive
 
45
    urlgrabber.keepalive.DEBUG = 0
54
46
    def get_url(path, compressed=False):
55
47
        try:
56
48
            url = path
57
49
            if compressed:
58
50
                url += '.gz'
59
51
            mutter("grab url %s" % url)
60
 
            url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
61
 
            if url_f.status != 200:
62
 
                raise GetFailed(url, url_f.status)
 
52
            url_f = urlgrabber.urlopen(url, keepalive=1, close_connection=0)
63
53
            if not compressed:
64
54
                return url_f
65
55
            else:
90
80
            fmt = ff.read()
91
81
            ff.close()
92
82
 
93
 
            if fmt != BZR_BRANCH_FORMAT_5:
 
83
            fmt = fmt.rstrip('\r\n')
 
84
            if fmt != BZR_BRANCH_FORMAT.rstrip('\r\n'):
94
85
                raise BzrError("sorry, branch format %r not supported at url %s"
95
86
                               % (fmt, url))
96
87
            
152
143
        pl = len(self.baseurl)
153
144
        return path[pl:].lstrip('/')
154
145
 
155
 
 
156
146
    def get_revision(self, revision_id):
157
 
        from bzrlib.revision import Revision
158
 
        try:
159
 
            revf = self.revision_store[revision_id]
160
 
        except KeyError:
161
 
            raise NoSuchRevision(self, revision_id)
162
 
        r = serializer_v5.read_revision(revf)
 
147
        from revision import Revision
 
148
        revf = get_url(self.baseurl + '/.bzr/revision-store/' + revision_id,
 
149
                       True)
 
150
        r = Revision.read_xml(revf)
163
151
        if r.revision_id != revision_id:
164
152
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
165
153
                                % (revision_id, r.revision_id))
178
166
        
179
167
    def __getitem__(self, fileid):
180
168
        p = self._path(fileid)
181
 
        try:
182
 
            return get_url(p, compressed=True)
183
 
        except:
184
 
            raise KeyError(fileid)
 
169
        return get_url(p, compressed=True)
185
170
    
186
171
 
 
172
def simple_walk():
 
173
    """For experimental purposes, traverse many parts of a remote branch"""
 
174
    from revision import Revision
 
175
    from branch import Branch
 
176
    from inventory import Inventory
 
177
 
 
178
    got_invs = {}
 
179
    got_texts = {}
 
180
 
 
181
    print 'read history'
 
182
    history = get_url('/.bzr/revision-history').readlines()
 
183
    num_revs = len(history)
 
184
    for i, rev_id in enumerate(history):
 
185
        rev_id = rev_id.rstrip()
 
186
        print 'read revision %d/%d' % (i, num_revs)
 
187
 
 
188
        # python gzip needs a seekable file (!!) but the HTTP response
 
189
        # isn't, so we need to buffer it
 
190
 
 
191
        rev_f = get_url('/.bzr/revision-store/%s' % rev_id,
 
192
                        compressed=True)
 
193
 
 
194
        rev = Revision.read_xml(rev_f)
 
195
        print rev.message
 
196
        inv_id = rev.inventory_id
 
197
        if inv_id not in got_invs:
 
198
            print 'get inventory %s' % inv_id
 
199
            inv_f = get_url('/.bzr/inventory-store/%s' % inv_id,
 
200
                            compressed=True)
 
201
            inv = Inventory.read_xml(inv_f)
 
202
            print '%4d inventory entries' % len(inv)
 
203
 
 
204
            for path, ie in inv.iter_entries():
 
205
                text_id = ie.text_id
 
206
                if text_id == None:
 
207
                    continue
 
208
                if text_id in got_texts:
 
209
                    continue
 
210
                print '  fetch %s text {%s}' % (path, text_id)
 
211
                text_f = get_url('/.bzr/text-store/%s' % text_id,
 
212
                                 compressed=True)
 
213
                got_texts[text_id] = True
 
214
 
 
215
            got_invs.add[inv_id] = True
 
216
 
 
217
        print '----'
 
218
 
 
219
 
 
220
def try_me():
 
221
    BASE_URL = 'http://bazaar-ng.org/bzr/bzr.dev/'
 
222
    b = RemoteBranch(BASE_URL)
 
223
    ## print '\n'.join(b.revision_history())
 
224
    from log import show_log
 
225
    show_log(b)
 
226
 
 
227
 
 
228
if __name__ == '__main__':
 
229
    try_me()
187
230