← 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-09-15 08:37:41 UTC
  • Revision ID: mbp@sourcefrog.net-20050915083741-70d7550b97c7b580
- some updates for fetch/update function

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
 
1
#! /usr/bin/env python
 
2
 
 
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
 
 
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
 
28
from cStringIO import StringIO
 
29
import urllib2
 
30
 
 
31
from bzrlib.errors import BzrError, BzrCheckError
 
32
from bzrlib.branch import Branch, BZR_BRANCH_FORMAT_5
 
33
from bzrlib.trace import mutter
 
34
from bzrlib.xml5 import serializer_v5
 
35
 
 
36
# velocitynet.com.au transparently proxies connections and thereby
 
37
# breaks keep-alive -- sucks!
 
38
 
 
39
 
 
40
ENABLE_URLGRABBER = False
 
41
 
 
42
from bzrlib.errors import BzrError, NoSuchRevision
 
43
 
 
44
class GetFailed(BzrError):
 
45
    def __init__(self, url, status):
 
46
        BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
 
47
        self.url = url
 
48
        self.status = status
 
49
 
 
50
if ENABLE_URLGRABBER:
 
51
    import util.urlgrabber
 
52
    import util.urlgrabber.keepalive
 
53
    util.urlgrabber.keepalive.DEBUG = 0
 
54
    def get_url(path, compressed=False):
 
55
        try:
 
56
            url = path
 
57
            if compressed:
 
58
                url += '.gz'
 
59
            mutter("grab url %s" % url)
 
60
            url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
 
61
            if url_f.status != 200:
 
62
                raise GetFailed(url, url_f.status)
 
63
            if not compressed:
 
64
                return url_f
 
65
            else:
 
66
                return gzip.GzipFile(fileobj=StringIO(url_f.read()))
 
67
        except urllib2.URLError, e:
 
68
            raise BzrError("remote fetch failed: %r: %s" % (url, e))
 
69
else:
 
70
    def get_url(url, compressed=False):
 
71
        import urllib2
 
72
        if compressed:
 
73
            url += '.gz'
 
74
        mutter("get_url %s" % url)
 
75
        url_f = urllib2.urlopen(url)
 
76
        if compressed:
 
77
            return gzip.GzipFile(fileobj=StringIO(url_f.read()))
 
78
        else:
 
79
            return url_f
 
80
 
 
81
 
 
82
 
 
83
def _find_remote_root(url):
 
84
    """Return the prefix URL that corresponds to the branch root."""
 
85
    orig_url = url
 
86
    while True:
 
87
        try:
 
88
            ff = get_url(url + '/.bzr/branch-format')
 
89
 
 
90
            fmt = ff.read()
 
91
            ff.close()
 
92
 
 
93
            if fmt != BZR_BRANCH_FORMAT_5:
 
94
                raise BzrError("sorry, branch format %r not supported at url %s"
 
95
                               % (fmt, url))
 
96
            
 
97
            return url
 
98
        except urllib2.URLError:
 
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
 
 
110
class RemoteBranch(Branch):
 
111
    def __init__(self, baseurl, find_root=True):
 
112
        """Create new proxy for a remote branch."""
 
113
        if find_root:
 
114
            self.baseurl = _find_remote_root(baseurl)
 
115
        else:
 
116
            self.baseurl = baseurl
 
117
            self._check_format()
 
118
 
 
119
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
 
120
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
 
121
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
 
122
 
 
123
    def __str__(self):
 
124
        b = getattr(self, 'baseurl', 'undefined')
 
125
        return '%s(%r)' % (self.__class__.__name__, b)
 
126
 
 
127
    __repr__ = __str__
 
128
 
 
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
 
 
134
 
 
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)
 
143
 
 
144
    def unlock(self):
 
145
        pass
 
146
    
 
147
 
 
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
 
 
155
 
 
156
    def get_revision(self, revision_id):
 
157
        from bzrlib.revision import Revision
 
158
        try:
 
159
            revf = self.revision_store[revision_id]
 
160
        except KeyError:
 
161
            raise NoSuchRevision(self, revision_id)
 
162
        r = serializer_v5.read_revision(revf)
 
163
        if r.revision_id != revision_id:
 
164
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
 
165
                                % (revision_id, r.revision_id))
 
166
        return r
 
167
 
 
168
 
 
169
class RemoteStore(object):
 
170
    def __init__(self, baseurl):
 
171
        self._baseurl = baseurl
 
172
        
 
173
 
 
174
    def _path(self, name):
 
175
        if '/' in name:
 
176
            raise ValueError('invalid store id', name)
 
177
        return self._baseurl + '/' + name
 
178
        
 
179
    def __getitem__(self, fileid):
 
180
        p = self._path(fileid)
 
181
        try:
 
182
            return get_url(p, compressed=True)
 
183
        except:
 
184
            raise KeyError(fileid)
 
185
    
 
186
 
 
187