← 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-05-26 02:13:57 UTC
  • Revision ID: mbp@sourcefrog.net-20050526021357-9a56decdca0c174f
- All top-level classes inherit from object
  (Python new-style classes)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
38
38
 
39
39
ENABLE_URLGRABBER = True
40
40
 
41
 
from bzrlib.errors import BzrError
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
48
41
 
49
42
if ENABLE_URLGRABBER:
50
 
    import util.urlgrabber
51
 
    import util.urlgrabber.keepalive
52
 
    util.urlgrabber.keepalive.DEBUG = 0
 
43
    import urlgrabber
 
44
    import urlgrabber.keepalive
 
45
    urlgrabber.keepalive.DEBUG = 0
53
46
    def get_url(path, compressed=False):
54
47
        try:
55
48
            url = path
56
49
            if compressed:
57
50
                url += '.gz'
58
51
            mutter("grab url %s" % url)
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)
 
52
            url_f = urlgrabber.urlopen(url, keepalive=1, close_connection=0)
62
53
            if not compressed:
63
54
                return url_f
64
55
            else:
108
99
 
109
100
 
110
101
class RemoteBranch(Branch):
111
 
    def __init__(self, baseurl, find_root=True):
 
102
    def __init__(self, baseurl, find_root=True, lock_mode='r'):
112
103
        """Create new proxy for a remote branch."""
 
104
        if lock_mode not in ('', 'r'):
 
105
            raise BzrError('lock mode %r is not supported for remote branches'
 
106
                           % lock_mode)
 
107
 
113
108
        if find_root:
114
109
            self.baseurl = _find_remote_root(baseurl)
115
110
        else:
118
113
 
119
114
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
120
115
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
121
 
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
122
116
 
123
117
    def __str__(self):
124
 
        b = getattr(self, 'baseurl', 'undefined')
125
 
        return '%s(%r)' % (self.__class__.__name__, b)
 
118
        return '%s(%r)' % (self.__class__.__name__, self.baseurl)
126
119
 
127
120
    __repr__ = __str__
128
121
 
131
124
            raise BzrError("file mode %r not supported for remote branches" % mode)
132
125
        return get_url(self.baseurl + '/.bzr/' + filename, False)
133
126
 
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
 
    
 
127
    def _need_readlock(self):
 
128
        # remote branch always safe for read
 
129
        pass
 
130
 
 
131
    def _need_writelock(self):
 
132
        raise BzrError("cannot get write lock on HTTP remote branch")
147
133
 
148
134
    def relpath(self, path):
149
135
        if not path.startswith(self.baseurl):
152
138
        pl = len(self.baseurl)
153
139
        return path[pl:].lstrip('/')
154
140
 
155
 
 
156
141
    def get_revision(self, revision_id):
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)
 
142
        from revision import Revision
 
143
        revf = get_url(self.baseurl + '/.bzr/revision-store/' + revision_id,
 
144
                       True)
 
145
        r = Revision.read_xml(revf)
161
146
        if r.revision_id != revision_id:
162
147
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
163
148
                                % (revision_id, r.revision_id))
176
161
        
177
162
    def __getitem__(self, fileid):
178
163
        p = self._path(fileid)
179
 
        try:
180
 
            return get_url(p, compressed=True)
181
 
        except:
182
 
            raise KeyError(fileid)
 
164
        return get_url(p, compressed=True)
183
165
    
184
166
 
 
167
def simple_walk():
 
168
    """For experimental purposes, traverse many parts of a remote branch"""
 
169
    from revision import Revision
 
170
    from branch import Branch
 
171
    from inventory import Inventory
 
172
 
 
173
    got_invs = {}
 
174
    got_texts = {}
 
175
 
 
176
    print 'read history'
 
177
    history = get_url('/.bzr/revision-history').readlines()
 
178
    num_revs = len(history)
 
179
    for i, rev_id in enumerate(history):
 
180
        rev_id = rev_id.rstrip()
 
181
        print 'read revision %d/%d' % (i, num_revs)
 
182
 
 
183
        # python gzip needs a seekable file (!!) but the HTTP response
 
184
        # isn't, so we need to buffer it
 
185
 
 
186
        rev_f = get_url('/.bzr/revision-store/%s' % rev_id,
 
187
                        compressed=True)
 
188
 
 
189
        rev = Revision.read_xml(rev_f)
 
190
        print rev.message
 
191
        inv_id = rev.inventory_id
 
192
        if inv_id not in got_invs:
 
193
            print 'get inventory %s' % inv_id
 
194
            inv_f = get_url('/.bzr/inventory-store/%s' % inv_id,
 
195
                            compressed=True)
 
196
            inv = Inventory.read_xml(inv_f)
 
197
            print '%4d inventory entries' % len(inv)
 
198
 
 
199
            for path, ie in inv.iter_entries():
 
200
                text_id = ie.text_id
 
201
                if text_id == None:
 
202
                    continue
 
203
                if text_id in got_texts:
 
204
                    continue
 
205
                print '  fetch %s text {%s}' % (path, text_id)
 
206
                text_f = get_url('/.bzr/text-store/%s' % text_id,
 
207
                                 compressed=True)
 
208
                got_texts[text_id] = True
 
209
 
 
210
            got_invs.add[inv_id] = True
 
211
 
 
212
        print '----'
 
213
 
 
214
 
 
215
def try_me():
 
216
    BASE_URL = 'http://bazaar-ng.org/bzr/bzr.dev/'
 
217
    b = RemoteBranch(BASE_URL)
 
218
    ## print '\n'.join(b.revision_history())
 
219
    from log import show_log
 
220
    show_log(b)
 
221
 
 
222
 
 
223
if __name__ == '__main__':
 
224
    try_me()
185
225