~bzr-pqm/bzr/bzr.dev

411 by Martin Pool
- start adding more useful RemoteBranch() class
1
#! /usr/bin/env python
2
188 by mbp at sourcefrog
- experimental remote-branch support
3
# Copyright (C) 2005 Canonical Ltd
4
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
9
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
# GNU General Public License for more details.
14
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
19
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
20
"""Proxy object for access to remote branches.
21
22
At the moment remote branches are only for HTTP and only for read
23
access.
24
"""
25
26
27
import gzip
192 by mbp at sourcefrog
- exercise the network more towards doing a remote clone
28
from cStringIO import StringIO
419 by Martin Pool
- RemoteBranch.__str__ and repr
29
import urllib2
188 by mbp at sourcefrog
- experimental remote-branch support
30
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
31
from errors import BzrError, BzrCheckError
419 by Martin Pool
- RemoteBranch.__str__ and repr
32
from branch import Branch, BZR_BRANCH_FORMAT
417 by Martin Pool
- trace when we fetch a URL
33
from trace import mutter
188 by mbp at sourcefrog
- experimental remote-branch support
34
193 by mbp at sourcefrog
more experiments with http get
35
# velocitynet.com.au transparently proxies connections and thereby
36
# breaks keep-alive -- sucks!
37
38
450 by Martin Pool
- Use urlgrabber by default
39
ENABLE_URLGRABBER = True
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
40
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
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
417 by Martin Pool
- trace when we fetch a URL
48
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
49
if ENABLE_URLGRABBER:
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
50
    import util.urlgrabber
51
    import util.urlgrabber.keepalive
52
    util.urlgrabber.keepalive.DEBUG = 0
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
53
    def get_url(path, compressed=False):
54
        try:
55
            url = path
56
            if compressed:
57
                url += '.gz'
450 by Martin Pool
- Use urlgrabber by default
58
            mutter("grab url %s" % url)
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
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)
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
62
            if not compressed:
63
                return url_f
64
            else:
65
                return gzip.GzipFile(fileobj=StringIO(url_f.read()))
66
        except urllib2.URLError, e:
67
            raise BzrError("remote fetch failed: %r: %s" % (url, e))
450 by Martin Pool
- Use urlgrabber by default
68
else:
69
    def get_url(url, compressed=False):
70
        import urllib2
71
        if compressed:
72
            url += '.gz'
73
        mutter("get_url %s" % url)
74
        url_f = urllib2.urlopen(url)
75
        if compressed:
76
            return gzip.GzipFile(fileobj=StringIO(url_f.read()))
77
        else:
78
            return url_f
188 by mbp at sourcefrog
- experimental remote-branch support
79
192 by mbp at sourcefrog
- exercise the network more towards doing a remote clone
80
416 by Martin Pool
- bzr log and bzr root now accept an http URL
81
82
def _find_remote_root(url):
83
    """Return the prefix URL that corresponds to the branch root."""
84
    orig_url = url
85
    while True:
86
        try:
87
            ff = get_url(url + '/.bzr/branch-format')
419 by Martin Pool
- RemoteBranch.__str__ and repr
88
89
            fmt = ff.read()
416 by Martin Pool
- bzr log and bzr root now accept an http URL
90
            ff.close()
419 by Martin Pool
- RemoteBranch.__str__ and repr
91
92
            fmt = fmt.rstrip('\r\n')
93
            if fmt != BZR_BRANCH_FORMAT.rstrip('\r\n'):
94
                raise BzrError("sorry, branch format %r not supported at url %s"
95
                               % (fmt, url))
96
            
416 by Martin Pool
- bzr log and bzr root now accept an http URL
97
            return url
418 by Martin Pool
- fix typo
98
        except urllib2.URLError:
416 by Martin Pool
- bzr log and bzr root now accept an http URL
99
            pass
100
101
        try:
102
            idx = url.rindex('/')
103
        except ValueError:
104
            raise BzrError('no branch root found for URL %s' % orig_url)
105
106
        url = url[:idx]        
107
        
108
109
411 by Martin Pool
- start adding more useful RemoteBranch() class
110
class RemoteBranch(Branch):
580 by Martin Pool
- Use explicit lock methods on a branch, rather than doing it
111
    def __init__(self, baseurl, find_root=True):
411 by Martin Pool
- start adding more useful RemoteBranch() class
112
        """Create new proxy for a remote branch."""
419 by Martin Pool
- RemoteBranch.__str__ and repr
113
        if find_root:
114
            self.baseurl = _find_remote_root(baseurl)
115
        else:
116
            self.baseurl = baseurl
117
            self._check_format()
411 by Martin Pool
- start adding more useful RemoteBranch() class
118
534 by Martin Pool
- new RemoteStore class
119
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
120
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
613 by Martin Pool
- fix locking for RemoteBranch
121
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
534 by Martin Pool
- new RemoteStore class
122
417 by Martin Pool
- trace when we fetch a URL
123
    def __str__(self):
585 by Martin Pool
- make RemoteBranch.__str__ more robust
124
        b = getattr(self, 'baseurl', 'undefined')
125
        return '%s(%r)' % (self.__class__.__name__, b)
417 by Martin Pool
- trace when we fetch a URL
126
127
    __repr__ = __str__
128
411 by Martin Pool
- start adding more useful RemoteBranch() class
129
    def controlfile(self, filename, mode):
130
        if mode not in ('rb', 'rt', 'r'):
131
            raise BzrError("file mode %r not supported for remote branches" % mode)
132
        return get_url(self.baseurl + '/.bzr/' + filename, False)
133
580 by Martin Pool
- Use explicit lock methods on a branch, rather than doing it
134
613 by Martin Pool
- fix locking for RemoteBranch
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)
580 by Martin Pool
- Use explicit lock methods on a branch, rather than doing it
143
144
    def unlock(self):
411 by Martin Pool
- start adding more useful RemoteBranch() class
145
        pass
580 by Martin Pool
- Use explicit lock methods on a branch, rather than doing it
146
    
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
147
416 by Martin Pool
- bzr log and bzr root now accept an http URL
148
    def relpath(self, path):
149
        if not path.startswith(self.baseurl):
150
            raise BzrError('path %r is not under base URL %r'
151
                           % (path, self.baseurl))
152
        pl = len(self.baseurl)
153
        return path[pl:].lstrip('/')
154
802 by Martin Pool
- Remove XMLMixin class in favour of simple pack_xml, unpack_xml functions
155
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
156
    def get_revision(self, revision_id):
802 by Martin Pool
- Remove XMLMixin class in favour of simple pack_xml, unpack_xml functions
157
        from bzrlib.revision import Revision
158
        from bzrlib.xml import unpack_xml
762 by Martin Pool
- RemoteBranch.get_revision now goes through store abstraction
159
        revf = self.revision_store[revision_id]
802 by Martin Pool
- Remove XMLMixin class in favour of simple pack_xml, unpack_xml functions
160
        r = unpack_xml(Revision, revf)
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
161
        if r.revision_id != revision_id:
162
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
163
                                % (revision_id, r.revision_id))
164
        return r
534 by Martin Pool
- new RemoteStore class
165
166
558 by Martin Pool
- All top-level classes inherit from object
167
class RemoteStore(object):
534 by Martin Pool
- new RemoteStore class
168
    def __init__(self, baseurl):
169
        self._baseurl = baseurl
170
        
171
172
    def _path(self, name):
173
        if '/' in name:
174
            raise ValueError('invalid store id', name)
175
        return self._baseurl + '/' + name
176
        
177
    def __getitem__(self, fileid):
178
        p = self._path(fileid)
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
179
        try:
180
            return get_url(p, compressed=True)
181
        except:
182
            raise KeyError(fileid)
188 by mbp at sourcefrog
- experimental remote-branch support
183
    
411 by Martin Pool
- start adding more useful RemoteBranch() class
184
185