37
class ResponseFile(object):
38
"""A wrapper around the http socket containing the result of a GET request.
40
Only read() and seek() (forward) are supported.
43
def __init__(self, path, infile):
46
:param path: File url, for error reports.
48
:param infile: File-like socket set at body start.
57
Dummy implementation for consistency with the 'file' API.
60
def read(self, size=-1):
61
"""Read size bytes from the current position in the file.
63
:param size: The number of bytes to read. Leave unspecified or pass
66
data = self._file.read(size)
67
self._pos += len(data)
71
data = self._file.readline()
72
self._pos += len(data)
77
line = self.readline()
85
def seek(self, offset, whence=os.SEEK_SET):
86
if whence == os.SEEK_SET:
87
if offset < self._pos:
89
"Can't seek backwards, pos: %s, offset: %s"
90
% (self._pos, offset))
91
to_discard = offset - self._pos
92
elif whence == os.SEEK_CUR:
95
raise AssertionError("Can't seek backwards")
97
# Just discard the unwanted bytes
35
100
# A RangeFile expects the following grammar (simplified to outline the
36
101
# assumptions we rely upon).
40
104
# | multiple_range
42
# whole_file: [content_length_header] data
44
106
# single_range: content_range_header data
46
108
# multiple_range: boundary_header boundary (content_range_header data boundary)+
48
class RangeFile(object):
110
class RangeFile(ResponseFile):
49
111
"""File-like object that allow access to partial available data.
51
113
All accesses should happen sequentially since the acquisition occurs during
72
134
:param path: File url, for error reports.
73
136
:param infile: File-like socket set at body start.
138
super(RangeFile, self).__init__(path, infile)
77
139
self._boundary = None
78
140
# When using multi parts response, this will be set with the headers
79
141
# associated with the range currently read.
297
359
:return: A file-like object that can seek()+read() the
298
360
ranges indicated by the headers.
300
rfile = RangeFile(url, data)
303
size = msg.getheader('content-length', None)
308
rfile.set_range(0, size)
364
rfile = ResponseFile(url, data)
309
365
elif code == 206:
366
rfile = RangeFile(url, data)
310
367
content_type = msg.getheader('content-type', None)
311
368
if content_type is None:
312
369
# When there is no content-type header we treat the response as