← 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-22 12:12:53 UTC
  • Revision ID: mbp@sourcefrog.net-20050922121253-eae2a3240ea5e493
- upgrade can no longer be done in current version branches
  so don't test it

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
30
 
31
31
from bzrlib.errors import BzrError, BzrCheckError
32
 
from bzrlib.branch import Branch, LocalBranch, BZR_BRANCH_FORMAT_5
 
32
from bzrlib.branch import Branch, BZR_BRANCH_FORMAT_5
33
33
from bzrlib.trace import mutter
34
 
from bzrlib.weavestore import WeaveStore
35
34
from bzrlib.xml5 import serializer_v5
36
35
 
37
 
 
38
36
# velocitynet.com.au transparently proxies connections and thereby
39
37
# breaks keep-alive -- sucks!
40
38
 
87
85
    orig_url = url
88
86
    while True:
89
87
        try:
90
 
            fmt_url = url + '/.bzr/branch-format'
91
 
            ff = get_url(fmt_url)
 
88
            ff = get_url(url + '/.bzr/branch-format')
 
89
 
92
90
            fmt = ff.read()
93
91
            ff.close()
94
92
 
100
98
        except urllib2.URLError:
101
99
            pass
102
100
 
103
 
        scheme, host, path = list(urlparse.urlparse(url))[:3]
104
 
        # discard params, query, fragment
105
 
        
106
 
        # strip off one component of the path component
107
 
        idx = path.rfind('/')
108
 
        if idx == -1 or path == '/':
109
 
            raise BzrError('no branch root found for URL %s'
110
 
                           ' or enclosing directories'
111
 
                           % orig_url)
112
 
        path = path[:idx]
113
 
        url = urlparse.urlunparse((scheme, host, path, '', '', ''))
114
 
        
115
 
 
116
 
 
117
 
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):
118
111
    def __init__(self, baseurl, find_root=True):
119
112
        """Create new proxy for a remote branch."""
120
113
        if find_root:
121
 
            self.base = _find_remote_root(baseurl)
 
114
            self.baseurl = _find_remote_root(baseurl)
122
115
        else:
123
 
            self.base = baseurl
124
 
        self._check_format(False)
125
 
        # is guaranteed to be a v5 store
 
116
            self.baseurl = baseurl
 
117
            self._check_format()
126
118
 
127
 
        cfn = self.controlfilename
128
 
        assert self._branch_format == 5
129
 
        self.control_weaves = WeaveStore(cfn([]), get_url)
130
 
        self.weave_store = WeaveStore(cfn('weaves'), get_url)
131
 
        self.revision_store = RemoteStore(cfn('revision-store'))
 
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/')
132
122
 
133
123
    def __str__(self):
134
124
        b = getattr(self, 'baseurl', 'undefined')
136
126
 
137
127
    __repr__ = __str__
138
128
 
139
 
    def setup_caching(self, cache_root):
140
 
        """Set up cached stores located under cache_root"""
141
 
        from bzrlib.meta_store import CachedStore
142
 
        for store_name in ('inventory_store', 'text_store', 'revision_store'):
143
 
            if not isinstance(getattr(self, store_name), CachedStore):
144
 
                cache_path = os.path.join(cache_root, store_name)
145
 
                os.mkdir(cache_path)
146
 
                new_store = CachedStore(getattr(self, store_name), cache_path)
147
 
                setattr(self, store_name, new_store)
148
 
 
149
129
    def controlfile(self, filename, mode):
150
130
        if mode not in ('rb', 'rt', 'r'):
151
131
            raise BzrError("file mode %r not supported for remote branches" % mode)
152
 
        return get_url(self.base + '/.bzr/' + filename, False)
 
132
        return get_url(self.baseurl + '/.bzr/' + filename, False)
153
133
 
154
134
 
155
135
    def lock_read(self):
159
139
    def lock_write(self):
160
140
        from errors import LockError
161
141
        raise LockError("write lock not supported for remote branch %s"
162
 
                        % self.base)
 
142
                        % self.baseurl)
163
143
 
164
144
    def unlock(self):
165
145
        pass
166
146
    
167
147
 
168
148
    def relpath(self, path):
169
 
        if not path.startswith(self.base):
 
149
        if not path.startswith(self.baseurl):
170
150
            raise BzrError('path %r is not under base URL %r'
171
 
                           % (path, self.base))
172
 
        pl = len(self.base)
 
151
                           % (path, self.baseurl))
 
152
        pl = len(self.baseurl)
173
153
        return path[pl:].lstrip('/')
174
154
 
175
155
 
176
156
    def get_revision(self, revision_id):
 
157
        from bzrlib.revision import Revision
177
158
        try:
178
159
            revf = self.revision_store[revision_id]
179
160
        except KeyError:
194
175
        if '/' in name:
195
176
            raise ValueError('invalid store id', name)
196
177
        return self._baseurl + '/' + name
197
 
 
198
 
    def __contains__(self, fileid):
199
 
        try:
200
 
            self[fileid]
201
 
            return True
202
 
        except KeyError:
203
 
            return False
204
178
        
205
179
    def __getitem__(self, fileid):
206
180
        p = self._path(fileid)
207
181
        try:
208
182
            return get_url(p, compressed=True)
209
 
        except urllib2.URLError:
210
 
            pass
211
 
        try:
212
 
            return get_url(p, compressed=False)
213
 
        except urllib2.URLError:
 
183
        except:
214
184
            raise KeyError(fileid)
215
185
    
216
186