← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/remotebranch.py

  • Committer: Robert Collins
  • Date: 2005-08-23 10:44:48 UTC
  • mto: (974.1.50) (1185.1.10) (1092.3.1)
  • mto: This revision was merged to the branch mainline in revision 1139.
  • Revision ID: robertc@robertcollins.net-20050823104448-fb5d448e7a5a8ee3
relace runTest with test_foo in blackbox tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
 
1
#! /usr/bin/env python
 
2
 
1
3
# Copyright (C) 2005 Canonical Ltd
2
4
 
3
5
# This program is free software; you can redistribute it and/or modify
24
26
 
25
27
import gzip
26
28
from cStringIO import StringIO
27
 
import os
28
29
import urllib2
29
 
import urlparse
30
 
 
31
 
from bzrlib.errors import BzrError, BzrCheckError
32
 
from bzrlib.branch import Branch, LocalBranch, BZR_BRANCH_FORMAT
33
 
from bzrlib.trace import mutter
34
 
from bzrlib.xml import serializer_v4
35
 
 
36
 
 
37
 
ENABLE_URLGRABBER = False
38
 
 
39
 
from bzrlib.errors import BzrError, NoSuchRevision
 
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
 
 
41
from bzrlib.errors import BzrError
40
42
 
41
43
class GetFailed(BzrError):
42
44
    def __init__(self, url, status):
82
84
    orig_url = url
83
85
    while True:
84
86
        try:
85
 
            fmt_url = url + '/.bzr/branch-format'
86
 
            ff = get_url(fmt_url)
 
87
            ff = get_url(url + '/.bzr/branch-format')
 
88
 
87
89
            fmt = ff.read()
88
90
            ff.close()
89
91
 
96
98
        except urllib2.URLError:
97
99
            pass
98
100
 
99
 
        scheme, host, path = list(urlparse.urlparse(url))[:3]
100
 
        # discard params, query, fragment
101
 
        
102
 
        # strip off one component of the path component
103
 
        idx = path.rfind('/')
104
 
        if idx == -1 or path == '/':
105
 
            raise BzrError('no branch root found for URL %s'
106
 
                           ' or enclosing directories'
107
 
                           % orig_url)
108
 
        path = path[:idx]
109
 
        url = urlparse.urlunparse((scheme, host, path, '', '', ''))
110
 
        
111
 
 
112
 
 
113
 
class RemoteBranch(LocalBranch):
 
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):
114
111
    def __init__(self, baseurl, find_root=True):
115
112
        """Create new proxy for a remote branch."""
116
113
        if find_root:
117
 
            self.base = _find_remote_root(baseurl)
 
114
            self.baseurl = _find_remote_root(baseurl)
118
115
        else:
119
 
            self.base = baseurl
 
116
            self.baseurl = baseurl
120
117
            self._check_format()
121
118
 
122
119
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
129
126
 
130
127
    __repr__ = __str__
131
128
 
132
 
    def setup_caching(self, cache_root):
133
 
        """Set up cached stores located under cache_root"""
134
 
        from bzrlib.meta_store import CachedStore
135
 
        for store_name in ('inventory_store', 'text_store', 'revision_store'):
136
 
            if not isinstance(getattr(self, store_name), CachedStore):
137
 
                cache_path = os.path.join(cache_root, store_name)
138
 
                os.mkdir(cache_path)
139
 
                new_store = CachedStore(getattr(self, store_name), cache_path)
140
 
                setattr(self, store_name, new_store)
141
 
 
142
129
    def controlfile(self, filename, mode):
143
130
        if mode not in ('rb', 'rt', 'r'):
144
131
            raise BzrError("file mode %r not supported for remote branches" % mode)
145
 
        return get_url(self.base + '/.bzr/' + filename, False)
 
132
        return get_url(self.baseurl + '/.bzr/' + filename, False)
146
133
 
147
134
 
148
135
    def lock_read(self):
152
139
    def lock_write(self):
153
140
        from errors import LockError
154
141
        raise LockError("write lock not supported for remote branch %s"
155
 
                        % self.base)
 
142
                        % self.baseurl)
156
143
 
157
144
    def unlock(self):
158
145
        pass
159
146
    
160
147
 
161
148
    def relpath(self, path):
162
 
        if not path.startswith(self.base):
 
149
        if not path.startswith(self.baseurl):
163
150
            raise BzrError('path %r is not under base URL %r'
164
 
                           % (path, self.base))
165
 
        pl = len(self.base)
 
151
                           % (path, self.baseurl))
 
152
        pl = len(self.baseurl)
166
153
        return path[pl:].lstrip('/')
167
154
 
168
155
 
169
156
    def get_revision(self, revision_id):
170
 
        try:
171
 
            revf = self.revision_store[revision_id]
172
 
        except KeyError:
173
 
            raise NoSuchRevision(self, revision_id)
174
 
        r = serializer_v4.read_revision(revf)
 
157
        from bzrlib.revision import Revision
 
158
        from bzrlib.xml import unpack_xml
 
159
        revf = self.revision_store[revision_id]
 
160
        r = unpack_xml(Revision, revf)
175
161
        if r.revision_id != revision_id:
176
162
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
177
163
                                % (revision_id, r.revision_id))
192
178
        p = self._path(fileid)
193
179
        try:
194
180
            return get_url(p, compressed=True)
195
 
        except urllib2.URLError:
 
181
        except:
196
182
            raise KeyError(fileid)
197
183
    
198
184