← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/remotebranch.py

  • Committer: aaron.bentley at utoronto
  • Date: 2005-09-05 01:31:37 UTC
  • mto: (1185.3.4)
  • mto: This revision was merged to the branch mainline in revision 1178.
  • Revision ID: aaron.bentley@utoronto.ca-20050905013137-c8b598b5a90a52a9
Fixed the revno bug in log

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
36
36
# breaks keep-alive -- sucks!
37
37
 
38
38
 
39
 
ENABLE_URLGRABBER = True
40
 
 
 
39
ENABLE_URLGRABBER = False
 
40
 
 
41
from bzrlib.errors import BzrError, NoSuchRevision
 
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
41
48
 
42
49
if ENABLE_URLGRABBER:
43
 
    import urlgrabber
44
 
    import urlgrabber.keepalive
45
 
    urlgrabber.keepalive.DEBUG = 0
 
50
    import util.urlgrabber
 
51
    import util.urlgrabber.keepalive
 
52
    util.urlgrabber.keepalive.DEBUG = 0
46
53
    def get_url(path, compressed=False):
47
54
        try:
48
55
            url = path
49
56
            if compressed:
50
57
                url += '.gz'
51
58
            mutter("grab url %s" % url)
52
 
            url_f = urlgrabber.urlopen(url, keepalive=1, close_connection=0)
 
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)
53
62
            if not compressed:
54
63
                return url_f
55
64
            else:
99
108
 
100
109
 
101
110
class RemoteBranch(Branch):
102
 
    def __init__(self, baseurl, find_root=True, lock_mode='r'):
 
111
    def __init__(self, baseurl, find_root=True):
103
112
        """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
 
 
108
113
        if find_root:
109
114
            self.baseurl = _find_remote_root(baseurl)
110
115
        else:
113
118
 
114
119
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
115
120
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
 
121
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
116
122
 
117
123
    def __str__(self):
118
 
        return '%s(%r)' % (self.__class__.__name__, self.baseurl)
 
124
        b = getattr(self, 'baseurl', 'undefined')
 
125
        return '%s(%r)' % (self.__class__.__name__, b)
119
126
 
120
127
    __repr__ = __str__
121
128
 
124
131
            raise BzrError("file mode %r not supported for remote branches" % mode)
125
132
        return get_url(self.baseurl + '/.bzr/' + filename, False)
126
133
 
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")
 
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
    
133
147
 
134
148
    def relpath(self, path):
135
149
        if not path.startswith(self.baseurl):
138
152
        pl = len(self.baseurl)
139
153
        return path[pl:].lstrip('/')
140
154
 
 
155
 
141
156
    def get_revision(self, revision_id):
142
 
        from revision import Revision
143
 
        revf = get_url(self.baseurl + '/.bzr/revision-store/' + revision_id,
144
 
                       True)
145
 
        r = Revision.read_xml(revf)
 
157
        from bzrlib.revision import Revision
 
158
        from bzrlib.xml import unpack_xml
 
159
        try:
 
160
            revf = self.revision_store[revision_id]
 
161
        except KeyError:
 
162
            raise NoSuchRevision(self, revision_id)
 
163
        r = unpack_xml(Revision, revf)
146
164
        if r.revision_id != revision_id:
147
165
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
148
166
                                % (revision_id, r.revision_id))
161
179
        
162
180
    def __getitem__(self, fileid):
163
181
        p = self._path(fileid)
164
 
        return get_url(p, compressed=True)
 
182
        try:
 
183
            return get_url(p, compressed=True)
 
184
        except:
 
185
            raise KeyError(fileid)
165
186
    
166
187
 
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()
225
188