← 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-27 01:39:26 UTC
  • Revision ID: mbp@sourcefrog.net-20050627013926-49413d5928809350
- import effbot.org http client

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
38
38
 
39
39
ENABLE_URLGRABBER = True
40
40
 
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
41
 
49
42
if ENABLE_URLGRABBER:
50
 
    import util.urlgrabber
51
 
    import util.urlgrabber.keepalive
52
 
    util.urlgrabber.keepalive.DEBUG = 0
 
43
    import urlgrabber
 
44
    import urlgrabber.keepalive
 
45
    urlgrabber.keepalive.DEBUG = 0
53
46
    def get_url(path, compressed=False):
54
47
        try:
55
48
            url = path
56
49
            if compressed:
57
50
                url += '.gz'
58
51
            mutter("grab url %s" % url)
59
 
            url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
60
 
            if url_f.status != 200:
61
 
                raise GetFailed(url, url_f.status)
 
52
            url_f = urlgrabber.urlopen(url, keepalive=1, close_connection=0)
62
53
            if not compressed:
63
54
                return url_f
64
55
            else:
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
 
        from bzrlib.xml import unpack_xml
 
147
        from revision import Revision
159
148
        revf = self.revision_store[revision_id]
160
 
        r = unpack_xml(Revision, revf)
 
149
        r = Revision.read_xml(revf)
161
150
        if r.revision_id != revision_id:
162
151
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
163
152
                                % (revision_id, r.revision_id))
176
165
        
177
166
    def __getitem__(self, fileid):
178
167
        p = self._path(fileid)
179
 
        try:
180
 
            return get_url(p, compressed=True)
181
 
        except:
182
 
            raise KeyError(fileid)
 
168
        return get_url(p, compressed=True)
183
169
    
184
170
 
 
171
def simple_walk():
 
172
    """For experimental purposes, traverse many parts of a remote branch"""
 
173
    from revision import Revision
 
174
    from branch import Branch
 
175
    from inventory import Inventory
 
176
 
 
177
    got_invs = {}
 
178
    got_texts = {}
 
179
 
 
180
    print 'read history'
 
181
    history = get_url('/.bzr/revision-history').readlines()
 
182
    num_revs = len(history)
 
183
    for i, rev_id in enumerate(history):
 
184
        rev_id = rev_id.rstrip()
 
185
        print 'read revision %d/%d' % (i, num_revs)
 
186
 
 
187
        # python gzip needs a seekable file (!!) but the HTTP response
 
188
        # isn't, so we need to buffer it
 
189
 
 
190
        rev_f = get_url('/.bzr/revision-store/%s' % rev_id,
 
191
                        compressed=True)
 
192
 
 
193
        rev = Revision.read_xml(rev_f)
 
194
        print rev.message
 
195
        inv_id = rev.inventory_id
 
196
        if inv_id not in got_invs:
 
197
            print 'get inventory %s' % inv_id
 
198
            inv_f = get_url('/.bzr/inventory-store/%s' % inv_id,
 
199
                            compressed=True)
 
200
            inv = Inventory.read_xml(inv_f)
 
201
            print '%4d inventory entries' % len(inv)
 
202
 
 
203
            for path, ie in inv.iter_entries():
 
204
                text_id = ie.text_id
 
205
                if text_id == None:
 
206
                    continue
 
207
                if text_id in got_texts:
 
208
                    continue
 
209
                print '  fetch %s text {%s}' % (path, text_id)
 
210
                text_f = get_url('/.bzr/text-store/%s' % text_id,
 
211
                                 compressed=True)
 
212
                got_texts[text_id] = True
 
213
 
 
214
            got_invs.add[inv_id] = True
 
215
 
 
216
        print '----'
 
217
 
 
218
 
 
219
def try_me():
 
220
    BASE_URL = 'http://bazaar-ng.org/bzr/bzr.dev/'
 
221
    b = RemoteBranch(BASE_URL)
 
222
    ## print '\n'.join(b.revision_history())
 
223
    from log import show_log
 
224
    show_log(b)
 
225
 
 
226
 
 
227
if __name__ == '__main__':
 
228
    try_me()
185
229