← 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-20 02:16:01 UTC
  • Revision ID: mbp@sourcefrog.net-20050520021601-d931cce5b66227a3
- bzr diff finds a branch from the first parameter,
  if any are given

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
26
26
 
27
27
import gzip
28
28
from cStringIO import StringIO
29
 
import os
30
29
import urllib2
31
 
import urlparse
32
 
 
33
 
from bzrlib.errors import BzrError, BzrCheckError
34
 
from bzrlib.branch import Branch, BZR_BRANCH_FORMAT
35
 
from bzrlib.trace import mutter
36
 
from bzrlib.xml import serializer_v4
37
 
 
38
 
 
39
 
ENABLE_URLGRABBER = False
40
 
 
41
 
from bzrlib.errors import BzrError, NoSuchRevision
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
 
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
 
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:
84
75
    orig_url = url
85
76
    while True:
86
77
        try:
87
 
            fmt_url = url + '/.bzr/branch-format'
88
 
            ff = get_url(fmt_url)
 
78
            ff = get_url(url + '/.bzr/branch-format')
 
79
 
89
80
            fmt = ff.read()
90
81
            ff.close()
91
82
 
98
89
        except urllib2.URLError:
99
90
            pass
100
91
 
101
 
        scheme, host, path = list(urlparse.urlparse(url))[:3]
102
 
        # discard params, query, fragment
103
 
        
104
 
        # strip off one component of the path component
105
 
        idx = path.rfind('/')
106
 
        if idx == -1 or path == '/':
107
 
            raise BzrError('no branch root found for URL %s'
108
 
                           ' or enclosing directories'
109
 
                           % orig_url)
110
 
        path = path[:idx]
111
 
        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]        
112
98
        
113
99
 
114
100
 
115
101
class RemoteBranch(Branch):
116
 
    def __init__(self, baseurl, find_root=True):
 
102
    def __init__(self, baseurl, find_root=True, lock_mode='r'):
117
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
 
118
108
        if find_root:
119
109
            self.baseurl = _find_remote_root(baseurl)
120
110
        else:
123
113
 
124
114
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
125
115
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
126
 
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
127
116
 
128
117
    def __str__(self):
129
 
        b = getattr(self, 'baseurl', 'undefined')
130
 
        return '%s(%r)' % (self.__class__.__name__, b)
 
118
        return '%s(%r)' % (self.__class__.__name__, self.baseurl)
131
119
 
132
120
    __repr__ = __str__
133
121
 
136
124
            raise BzrError("file mode %r not supported for remote branches" % mode)
137
125
        return get_url(self.baseurl + '/.bzr/' + filename, False)
138
126
 
139
 
 
140
 
    def lock_read(self):
141
 
        # no locking for remote branches yet
142
 
        pass
143
 
 
144
 
    def lock_write(self):
145
 
        from errors import LockError
146
 
        raise LockError("write lock not supported for remote branch %s"
147
 
                        % self.baseurl)
148
 
 
149
 
    def unlock(self):
150
 
        pass
151
 
    
 
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")
152
133
 
153
134
    def relpath(self, path):
154
135
        if not path.startswith(self.baseurl):
157
138
        pl = len(self.baseurl)
158
139
        return path[pl:].lstrip('/')
159
140
 
160
 
 
161
141
    def get_revision(self, revision_id):
162
 
        try:
163
 
            revf = self.revision_store[revision_id]
164
 
        except KeyError:
165
 
            raise NoSuchRevision(self, revision_id)
166
 
        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)
167
146
        if r.revision_id != revision_id:
168
147
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
169
148
                                % (revision_id, r.revision_id))
170
149
        return r
171
150
 
172
151
 
173
 
class RemoteStore(object):
 
152
class RemoteStore:
174
153
    def __init__(self, baseurl):
175
154
        self._baseurl = baseurl
176
155
        
182
161
        
183
162
    def __getitem__(self, fileid):
184
163
        p = self._path(fileid)
185
 
        try:
186
 
            return get_url(p, compressed=True)
187
 
        except urllib2.URLError:
188
 
            raise KeyError(fileid)
 
164
        return get_url(p, compressed=True)
189
165
    
190
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
 
 
173
    got_invs = {}
 
174
    got_texts = {}
 
175
 
 
176
    print 'read history'
 
177
    history = get_url('/.bzr/revision-history').readlines()
 
178
    num_revs = len(history)
 
179
    for i, rev_id in enumerate(history):
 
180
        rev_id = rev_id.rstrip()
 
181
        print 'read revision %d/%d' % (i, num_revs)
 
182
 
 
183
        # python gzip needs a seekable file (!!) but the HTTP response
 
184
        # isn't, so we need to buffer it
 
185
 
 
186
        rev_f = get_url('/.bzr/revision-store/%s' % rev_id,
 
187
                        compressed=True)
 
188
 
 
189
        rev = Revision.read_xml(rev_f)
 
190
        print rev.message
 
191
        inv_id = rev.inventory_id
 
192
        if inv_id not in got_invs:
 
193
            print 'get inventory %s' % inv_id
 
194
            inv_f = get_url('/.bzr/inventory-store/%s' % inv_id,
 
195
                            compressed=True)
 
196
            inv = Inventory.read_xml(inv_f)
 
197
            print '%4d inventory entries' % len(inv)
 
198
 
 
199
            for path, ie in inv.iter_entries():
 
200
                text_id = ie.text_id
 
201
                if text_id == None:
 
202
                    continue
 
203
                if text_id in got_texts:
 
204
                    continue
 
205
                print '  fetch %s text {%s}' % (path, text_id)
 
206
                text_f = get_url('/.bzr/text-store/%s' % text_id,
 
207
                                 compressed=True)
 
208
                got_texts[text_id] = True
 
209
 
 
210
            got_invs.add[inv_id] = True
 
211
 
 
212
        print '----'
 
213
 
 
214
 
 
215
def try_me():
 
216
    BASE_URL = 'http://bazaar-ng.org/bzr/bzr.dev/'
 
217
    b = RemoteBranch(BASE_URL)
 
218
    ## print '\n'.join(b.revision_history())
 
219
    from log import show_log
 
220
    show_log(b)
 
221
 
 
222
 
 
223
if __name__ == '__main__':
 
224
    try_me()
191
225