~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/remotebranch.py

  • Committer: Martin Pool
  • Date: 2005-07-07 10:31:36 UTC
  • Revision ID: mbp@sourcefrog.net-20050707103135-9b4d911d8df6e880
- fix pwk help

Show diffs side-by-side

added added

removed removed

Lines of Context:
25
25
 
26
26
 
27
27
import gzip
28
 
from sets import Set
29
28
from cStringIO import StringIO
30
29
import urllib2
31
30
 
100
99
 
101
100
 
102
101
class RemoteBranch(Branch):
103
 
    def __init__(self, baseurl, find_root=True, lock_mode='r'):
 
102
    def __init__(self, baseurl, find_root=True):
104
103
        """Create new proxy for a remote branch."""
105
 
        if lock_mode not in ('', 'r'):
106
 
            raise BzrError('lock mode %r is not supported for remote branches'
107
 
                           % lock_mode)
108
 
 
109
104
        if find_root:
110
105
            self.baseurl = _find_remote_root(baseurl)
111
106
        else:
112
107
            self.baseurl = baseurl
113
108
            self._check_format()
114
109
 
 
110
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
 
111
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
 
112
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
 
113
 
115
114
    def __str__(self):
116
 
        return '%s(%r)' % (self.__class__.__name__, self.baseurl)
 
115
        b = getattr(self, 'baseurl', 'undefined')
 
116
        return '%s(%r)' % (self.__class__.__name__, b)
117
117
 
118
118
    __repr__ = __str__
119
119
 
122
122
            raise BzrError("file mode %r not supported for remote branches" % mode)
123
123
        return get_url(self.baseurl + '/.bzr/' + filename, False)
124
124
 
125
 
    def _need_readlock(self):
126
 
        # remote branch always safe for read
127
 
        pass
128
 
 
129
 
    def _need_writelock(self):
130
 
        raise BzrError("cannot get write lock on HTTP remote branch")
 
125
 
 
126
    def lock_read(self):
 
127
        # no locking for remote branches yet
 
128
        pass
 
129
 
 
130
    def lock_write(self):
 
131
        from errors import LockError
 
132
        raise LockError("write lock not supported for remote branch %s"
 
133
                        % self.baseurl)
 
134
 
 
135
    def unlock(self):
 
136
        pass
 
137
    
131
138
 
132
139
    def relpath(self, path):
133
140
        if not path.startswith(self.baseurl):
136
143
        pl = len(self.baseurl)
137
144
        return path[pl:].lstrip('/')
138
145
 
 
146
 
139
147
    def get_revision(self, revision_id):
140
 
        from revision import Revision
141
 
        revf = get_url(self.baseurl + '/.bzr/revision-store/' + revision_id,
142
 
                       True)
143
 
        r = Revision.read_xml(revf)
 
148
        from bzrlib.revision import Revision
 
149
        from bzrlib.xml import unpack_xml
 
150
        revf = self.revision_store[revision_id]
 
151
        r = unpack_xml(Revision, revf)
144
152
        if r.revision_id != revision_id:
145
153
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
146
154
                                % (revision_id, r.revision_id))
147
155
        return r
 
156
 
 
157
 
 
158
class RemoteStore(object):
 
159
    def __init__(self, baseurl):
 
160
        self._baseurl = baseurl
 
161
        
 
162
 
 
163
    def _path(self, name):
 
164
        if '/' in name:
 
165
            raise ValueError('invalid store id', name)
 
166
        return self._baseurl + '/' + name
 
167
        
 
168
    def __getitem__(self, fileid):
 
169
        p = self._path(fileid)
 
170
        return get_url(p, compressed=True)
148
171
    
149
172
 
150
 
def simple_walk():
151
 
    from revision import Revision
152
 
    from branch import Branch
153
 
    from inventory import Inventory
154
 
 
155
 
    got_invs = Set()
156
 
    got_texts = Set()
157
 
 
158
 
    print 'read history'
159
 
    history = get_url('/.bzr/revision-history').readlines()
160
 
    num_revs = len(history)
161
 
    for i, rev_id in enumerate(history):
162
 
        rev_id = rev_id.rstrip()
163
 
        print 'read revision %d/%d' % (i, num_revs)
164
 
 
165
 
        # python gzip needs a seekable file (!!) but the HTTP response
166
 
        # isn't, so we need to buffer it
167
 
 
168
 
        rev_f = get_url('/.bzr/revision-store/%s' % rev_id,
169
 
                        compressed=True)
170
 
 
171
 
        rev = Revision.read_xml(rev_f)
172
 
        print rev.message
173
 
        inv_id = rev.inventory_id
174
 
        if inv_id not in got_invs:
175
 
            print 'get inventory %s' % inv_id
176
 
            inv_f = get_url('/.bzr/inventory-store/%s' % inv_id,
177
 
                            compressed=True)
178
 
            inv = Inventory.read_xml(inv_f)
179
 
            print '%4d inventory entries' % len(inv)
180
 
 
181
 
            for path, ie in inv.iter_entries():
182
 
                text_id = ie.text_id
183
 
                if text_id == None:
184
 
                    continue
185
 
                if text_id in got_texts:
186
 
                    continue
187
 
                print '  fetch %s text {%s}' % (path, text_id)
188
 
                text_f = get_url('/.bzr/text-store/%s' % text_id,
189
 
                                 compressed=True)
190
 
                got_texts.add(text_id)
191
 
 
192
 
            got_invs.add(inv_id)
193
 
 
194
 
        print '----'
195
 
 
196
 
 
197
 
def try_me():
198
 
    BASE_URL = 'http://bazaar-ng.org/bzr/bzr.dev/'
199
 
    b = RemoteBranch(BASE_URL)
200
 
    ## print '\n'.join(b.revision_history())
201
 
    from log import show_log
202
 
    show_log(b)
203
 
 
204
 
 
205
 
if __name__ == '__main__':
206
 
    try_me()
207
173