~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
974.1.53 by aaron.bentley at utoronto
Disabled urlgrabber
39
ENABLE_URLGRABBER = False
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
40
974.1.51 by aaron.bentley at utoronto
Caused remotebranch to throw a NoSuchRevision when appropriate
41
from bzrlib.errors import BzrError, NoSuchRevision
974.1.26 by aaron.bentley at utoronto
merged mbp@sourcefrog.net-20050817233101-0939da1cf91f2472
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
411 by Martin Pool
- start adding more useful RemoteBranch() class
109
class RemoteBranch(Branch):
1092.2.2 by Robert Collins
move RemoteStore to store.py
110
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."""
1092.2.2 by Robert Collins
move RemoteStore to store.py
113
        # circular import protection
114
        from bzrlib.store import RemoteStore
419 by Martin Pool
- RemoteBranch.__str__ and repr
115
        if find_root:
116
            self.baseurl = _find_remote_root(baseurl)
117
        else:
118
            self.baseurl = baseurl
119
            self._check_format()
534 by Martin Pool
- new RemoteStore class
120
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
121
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
613 by Martin Pool
- fix locking for RemoteBranch
122
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
534 by Martin Pool
- new RemoteStore class
123
417 by Martin Pool
- trace when we fetch a URL
124
    def __str__(self):
585 by Martin Pool
- make RemoteBranch.__str__ more robust
125
        b = getattr(self, 'baseurl', 'undefined')
126
        return '%s(%r)' % (self.__class__.__name__, b)
417 by Martin Pool
- trace when we fetch a URL
127
128
    __repr__ = __str__
129
411 by Martin Pool
- start adding more useful RemoteBranch() class
130
    def controlfile(self, filename, mode):
131
        if mode not in ('rb', 'rt', 'r'):
132
            raise BzrError("file mode %r not supported for remote branches" % mode)
133
        return get_url(self.baseurl + '/.bzr/' + filename, False)
134
580 by Martin Pool
- Use explicit lock methods on a branch, rather than doing it
135
613 by Martin Pool
- fix locking for RemoteBranch
136
    def lock_read(self):
137
        # no locking for remote branches yet
138
        pass
139
140
    def lock_write(self):
141
        from errors import LockError
142
        raise LockError("write lock not supported for remote branch %s"
143
                        % self.baseurl)
580 by Martin Pool
- Use explicit lock methods on a branch, rather than doing it
144
145
    def unlock(self):
411 by Martin Pool
- start adding more useful RemoteBranch() class
146
        pass
580 by Martin Pool
- Use explicit lock methods on a branch, rather than doing it
147
    
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
148
416 by Martin Pool
- bzr log and bzr root now accept an http URL
149
    def relpath(self, path):
150
        if not path.startswith(self.baseurl):
151
            raise BzrError('path %r is not under base URL %r'
152
                           % (path, self.baseurl))
153
        pl = len(self.baseurl)
154
        return path[pl:].lstrip('/')
155
802 by Martin Pool
- Remove XMLMixin class in favour of simple pack_xml, unpack_xml functions
156
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
157
    def get_revision(self, revision_id):
802 by Martin Pool
- Remove XMLMixin class in favour of simple pack_xml, unpack_xml functions
158
        from bzrlib.revision import Revision
159
        from bzrlib.xml import unpack_xml
974.1.51 by aaron.bentley at utoronto
Caused remotebranch to throw a NoSuchRevision when appropriate
160
        try:
161
            revf = self.revision_store[revision_id]
162
        except KeyError:
163
            raise NoSuchRevision(self, revision_id)
802 by Martin Pool
- Remove XMLMixin class in favour of simple pack_xml, unpack_xml functions
164
        r = unpack_xml(Revision, revf)
412 by Martin Pool
- RemoteBranch displays the log of a remote branch.
165
        if r.revision_id != revision_id:
166
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
167
                                % (revision_id, r.revision_id))
168
        return r
534 by Martin Pool
- new RemoteStore class
169