← 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-05-19 08:31:06 UTC
  • Revision ID: mbp@sourcefrog.net-20050519083106-ebe71562d3bda4a7
- fix typo

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
 
1
#! /usr/bin/env python
 
2
 
1
3
# Copyright (C) 2005 Canonical Ltd
2
4
 
3
5
# This program is free software; you can redistribute it and/or modify
24
26
 
25
27
import gzip
26
28
from cStringIO import StringIO
27
 
import os
28
29
import urllib2
29
 
import urlparse
30
 
 
31
 
from bzrlib.errors import BzrError, BzrCheckError
32
 
from bzrlib.branch import Branch, BZR_BRANCH_FORMAT
33
 
from bzrlib.trace import mutter
34
 
from bzrlib.xml import serializer_v4
35
 
 
36
 
 
37
 
ENABLE_URLGRABBER = False
38
 
 
39
 
from bzrlib.errors import BzrError, NoSuchRevision
40
 
 
41
 
class GetFailed(BzrError):
42
 
    def __init__(self, url, status):
43
 
        BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
44
 
        self.url = url
45
 
        self.status = status
 
30
 
 
31
from errors import BzrError, BzrCheckError
 
32
from branch import Branch, BZR_BRANCH_FORMAT
 
33
from trace import mutter
 
34
 
 
35
# velocitynet.com.au transparently proxies connections and thereby
 
36
# breaks keep-alive -- sucks!
 
37
 
 
38
 
 
39
ENABLE_URLGRABBER = True
 
40
 
46
41
 
47
42
if ENABLE_URLGRABBER:
48
 
    import util.urlgrabber
49
 
    import util.urlgrabber.keepalive
50
 
    util.urlgrabber.keepalive.DEBUG = 0
 
43
    import urlgrabber
 
44
    import urlgrabber.keepalive
 
45
    urlgrabber.keepalive.DEBUG = 0
51
46
    def get_url(path, compressed=False):
52
47
        try:
53
48
            url = path
54
49
            if compressed:
55
50
                url += '.gz'
56
51
            mutter("grab url %s" % url)
57
 
            url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
58
 
            if url_f.status != 200:
59
 
                raise GetFailed(url, url_f.status)
 
52
            url_f = urlgrabber.urlopen(url, keepalive=1, close_connection=0)
60
53
            if not compressed:
61
54
                return url_f
62
55
            else:
82
75
    orig_url = url
83
76
    while True:
84
77
        try:
85
 
            fmt_url = url + '/.bzr/branch-format'
86
 
            ff = get_url(fmt_url)
 
78
            ff = get_url(url + '/.bzr/branch-format')
 
79
 
87
80
            fmt = ff.read()
88
81
            ff.close()
89
82
 
96
89
        except urllib2.URLError:
97
90
            pass
98
91
 
99
 
        scheme, host, path = list(urlparse.urlparse(url))[:3]
100
 
        # discard params, query, fragment
101
 
        
102
 
        # strip off one component of the path component
103
 
        idx = path.rfind('/')
104
 
        if idx == -1 or path == '/':
105
 
            raise BzrError('no branch root found for URL %s'
106
 
                           ' or enclosing directories'
107
 
                           % orig_url)
108
 
        path = path[:idx]
109
 
        url = urlparse.urlunparse((scheme, host, path, '', '', ''))
 
92
        try:
 
93
            idx = url.rindex('/')
 
94
        except ValueError:
 
95
            raise BzrError('no branch root found for URL %s' % orig_url)
 
96
 
 
97
        url = url[:idx]        
110
98
        
111
99
 
112
100
 
113
101
class RemoteBranch(Branch):
114
 
    def __init__(self, baseurl, find_root=True):
 
102
    def __init__(self, baseurl, find_root=True, lock_mode='r'):
115
103
        """Create new proxy for a remote branch."""
 
104
        if lock_mode not in ('', 'r'):
 
105
            raise BzrError('lock mode %r is not supported for remote branches'
 
106
                           % lock_mode)
 
107
 
116
108
        if find_root:
117
 
            self.base = _find_remote_root(baseurl)
 
109
            self.baseurl = _find_remote_root(baseurl)
118
110
        else:
119
 
            self.base = baseurl
 
111
            self.baseurl = baseurl
120
112
            self._check_format()
121
113
 
122
114
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
123
115
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
124
 
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
125
116
 
126
117
    def __str__(self):
127
 
        b = getattr(self, 'baseurl', 'undefined')
128
 
        return '%s(%r)' % (self.__class__.__name__, b)
 
118
        return '%s(%r)' % (self.__class__.__name__, self.baseurl)
129
119
 
130
120
    __repr__ = __str__
131
121
 
132
122
    def controlfile(self, filename, mode):
133
123
        if mode not in ('rb', 'rt', 'r'):
134
124
            raise BzrError("file mode %r not supported for remote branches" % mode)
135
 
        return get_url(self.base + '/.bzr/' + filename, False)
136
 
 
137
 
 
138
 
    def lock_read(self):
139
 
        # no locking for remote branches yet
140
 
        pass
141
 
 
142
 
    def lock_write(self):
143
 
        from errors import LockError
144
 
        raise LockError("write lock not supported for remote branch %s"
145
 
                        % self.base)
146
 
 
147
 
    def unlock(self):
148
 
        pass
149
 
    
 
125
        return get_url(self.baseurl + '/.bzr/' + filename, False)
 
126
 
 
127
    def _need_readlock(self):
 
128
        # remote branch always safe for read
 
129
        pass
 
130
 
 
131
    def _need_writelock(self):
 
132
        raise BzrError("cannot get write lock on HTTP remote branch")
150
133
 
151
134
    def relpath(self, path):
152
 
        if not path.startswith(self.base):
 
135
        if not path.startswith(self.baseurl):
153
136
            raise BzrError('path %r is not under base URL %r'
154
 
                           % (path, self.base))
155
 
        pl = len(self.base)
 
137
                           % (path, self.baseurl))
 
138
        pl = len(self.baseurl)
156
139
        return path[pl:].lstrip('/')
157
140
 
158
 
 
159
141
    def get_revision(self, revision_id):
160
 
        try:
161
 
            revf = self.revision_store[revision_id]
162
 
        except KeyError:
163
 
            raise NoSuchRevision(self, revision_id)
164
 
        r = serializer_v4.read_revision(revf)
 
142
        from revision import Revision
 
143
        revf = get_url(self.baseurl + '/.bzr/revision-store/' + revision_id,
 
144
                       True)
 
145
        r = Revision.read_xml(revf)
165
146
        if r.revision_id != revision_id:
166
147
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
167
148
                                % (revision_id, r.revision_id))
168
149
        return r
169
150
 
170
151
 
171
 
class RemoteStore(object):
 
152
class RemoteStore:
172
153
    def __init__(self, baseurl):
173
154
        self._baseurl = baseurl
174
155
        
180
161
        
181
162
    def __getitem__(self, fileid):
182
163
        p = self._path(fileid)
183
 
        try:
184
 
            return get_url(p, compressed=True)
185
 
        except urllib2.URLError:
186
 
            raise KeyError(fileid)
 
164
        return get_url(p, compressed=True)
187
165
    
188
166
 
 
167
def simple_walk():
 
168
    """For experimental purposes, traverse many parts of a remote branch"""
 
169
    from revision import Revision
 
170
    from branch import Branch
 
171
    from inventory import Inventory
 
172
    from bzrlib import set
 
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()
189
226