/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/remotebranch.py

  • Committer: Lalo Martins
  • Date: 2005-09-07 10:50:03 UTC
  • mto: (1185.1.5)
  • mto: This revision was merged to the branch mainline in revision 1390.
  • Revision ID: lalo@exoweb.net-20050907105002-4e37d43b7d4060fe
unifying 'base' (from Branch) and 'baseurl' (from RemoteBranch) attributes;
this allows us to remove quite a few "dead chickens" from the code.

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 os
 
30
import urllib2
 
31
 
 
32
from bzrlib.errors import BzrError, BzrCheckError
 
33
from bzrlib.branch import Branch, BZR_BRANCH_FORMAT
 
34
from bzrlib.trace import mutter
 
35
from bzrlib.xml import serializer_v4
 
36
 
 
37
 
 
38
ENABLE_URLGRABBER = False
 
39
 
 
40
from bzrlib.errors import BzrError, NoSuchRevision
 
41
 
 
42
class GetFailed(BzrError):
 
43
    def __init__(self, url, status):
 
44
        BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
 
45
        self.url = url
 
46
        self.status = status
 
47
 
 
48
if ENABLE_URLGRABBER:
 
49
    import util.urlgrabber
 
50
    import util.urlgrabber.keepalive
 
51
    util.urlgrabber.keepalive.DEBUG = 0
 
52
    def get_url(path, compressed=False):
 
53
        try:
 
54
            url = path
 
55
            if compressed:
 
56
                url += '.gz'
 
57
            mutter("grab url %s" % url)
 
58
            url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
 
59
            if url_f.status != 200:
 
60
                raise GetFailed(url, url_f.status)
 
61
            if not compressed:
 
62
                return url_f
 
63
            else:
 
64
                return gzip.GzipFile(fileobj=StringIO(url_f.read()))
 
65
        except urllib2.URLError, e:
 
66
            raise BzrError("remote fetch failed: %r: %s" % (url, e))
 
67
else:
 
68
    def get_url(url, compressed=False):
 
69
        import urllib2
 
70
        if compressed:
 
71
            url += '.gz'
 
72
        mutter("get_url %s" % url)
 
73
        url_f = urllib2.urlopen(url)
 
74
        if compressed:
 
75
            return gzip.GzipFile(fileobj=StringIO(url_f.read()))
 
76
        else:
 
77
            return url_f
 
78
 
 
79
 
 
80
 
 
81
def _find_remote_root(url):
 
82
    """Return the prefix URL that corresponds to the branch root."""
 
83
    orig_url = url
 
84
    while True:
 
85
        try:
 
86
            ff = get_url(url + '/.bzr/branch-format')
 
87
 
 
88
            fmt = ff.read()
 
89
            ff.close()
 
90
 
 
91
            fmt = fmt.rstrip('\r\n')
 
92
            if fmt != BZR_BRANCH_FORMAT.rstrip('\r\n'):
 
93
                raise BzrError("sorry, branch format %r not supported at url %s"
 
94
                               % (fmt, url))
 
95
            
 
96
            return url
 
97
        except urllib2.URLError:
 
98
            pass
 
99
 
 
100
        try:
 
101
            idx = url.rindex('/')
 
102
        except ValueError:
 
103
            raise BzrError('no branch root found for URL %s' % orig_url)
 
104
 
 
105
        url = url[:idx]        
 
106
        
 
107
 
 
108
 
 
109
class RemoteBranch(Branch):
 
110
    def __init__(self, baseurl, find_root=True):
 
111
        """Create new proxy for a remote branch."""
 
112
        if find_root:
 
113
            self.base = _find_remote_root(baseurl)
 
114
        else:
 
115
            self.base = baseurl
 
116
            self._check_format()
 
117
 
 
118
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
 
119
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
 
120
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
 
121
 
 
122
    def __str__(self):
 
123
        b = getattr(self, 'baseurl', 'undefined')
 
124
        return '%s(%r)' % (self.__class__.__name__, b)
 
125
 
 
126
    __repr__ = __str__
 
127
 
 
128
    def controlfile(self, filename, mode):
 
129
        if mode not in ('rb', 'rt', 'r'):
 
130
            raise BzrError("file mode %r not supported for remote branches" % mode)
 
131
        return get_url(self.base + '/.bzr/' + filename, False)
 
132
 
 
133
 
 
134
    def lock_read(self):
 
135
        # no locking for remote branches yet
 
136
        pass
 
137
 
 
138
    def lock_write(self):
 
139
        from errors import LockError
 
140
        raise LockError("write lock not supported for remote branch %s"
 
141
                        % self.base)
 
142
 
 
143
    def unlock(self):
 
144
        pass
 
145
    
 
146
 
 
147
    def relpath(self, path):
 
148
        if not path.startswith(self.base):
 
149
            raise BzrError('path %r is not under base URL %r'
 
150
                           % (path, self.base))
 
151
        pl = len(self.base)
 
152
        return path[pl:].lstrip('/')
 
153
 
 
154
 
 
155
    def get_revision(self, revision_id):
 
156
        try:
 
157
            revf = self.revision_store[revision_id]
 
158
        except KeyError:
 
159
            raise NoSuchRevision(self, revision_id)
 
160
        r = serializer_v4.read_revision(revf)
 
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
 
165
 
 
166
 
 
167
class RemoteStore(object):
 
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)
 
179
        try:
 
180
            return get_url(p, compressed=True)
 
181
        except:
 
182
            raise KeyError(fileid)
 
183
    
 
184
 
 
185