async def test_response_make_conditional() -> None:
response = Response(b"abcdef")
await response.make_conditional(Range("bytes", [RangeSet(0, 3)]))
assert b"abc" == (await response.get_data()) # type: ignore
assert response.status_code == 206
assert response.accept_ranges == "bytes"
assert response.content_range == ContentRange("bytes", 0, 2, 6)
def test_content_range() -> None:
updated = False
def on_update(_: HeaderSet) -> None:
nonlocal updated
updated = True
content_range = ContentRange.from_header('bytes 0-499/1234', on_update=on_update)
assert content_range.units == 'bytes'
assert content_range.start == 0
assert content_range.stop == 499
assert content_range.length == 1234
content_range.start = 734
content_range.stop = 1233
assert updated
assert content_range.to_header() == 'bytes 734-1233/1234'
def good_response(fileobj, headers=None, download=False):
headers = headers or {}
content_type = fileobj.content_type or 'application/octet-stream'
if not isinstance(content_type, str):
content_type = content_type.decode('utf8')
headers['Content-Type'] = content_type
modified = fileobj.modified or fileobj.created
headers['Last-Modified'] = modified.strftime("%a, %d %b %Y %H:%M:%S GMT")
headers['x-amz-version-id'] = str(fileobj.rev)
for key, value in fileobj.meta.items():
if isinstance(value, bytes):
value = value.decode('utf8')
else:
value = str(value)
headers['x-amz-meta-%s' % key] = value
if download:
headers[
'Content-Disposition'] = 'attachment; filename="%s"' % os.path.basename(
fileobj.path)
status = 200
headers['Accept-Ranges'] = 'bytes'
brange = request.headers.get('range')
body = b''
to_read = -1
ranger = None
if not brange:
etag = fileobj.meta.get('md5') or fileobj.meta.get('sha256') or ''
if etag:
headers['Etag'] = etag
if etag and request.headers.get('if-none-match') == etag:
status = 304
body = b''
to_read = 0
else:
if_date = request.headers.get('If-Modified-Since', '')
if if_date:
ts = datetime.strptime(if_date, "%a, %d %b %Y %H:%M:%S %Z")
if ts >= modified:
status = 304
body = b''
to_read = 0
else:
try:
ranger = Range.from_header(brange)
except ValueError:
print('BAD RANGE HEADER', brange)
else:
fr = ranger.ranges[0]
headers['Content-Range'] = ContentRange(
ranger.units, fr.begin, fr.end, fileobj.length).to_header()
fileobj.seek(fr.begin)
to_read = (fr.end - fr.begin) + 1
status = 206
if request.method == 'HEAD':
resp = Response('', headers=headers)
resp.content_length = fileobj.length
return resp
if status >= 200 and (fileobj.length >= 65535 and
(to_read == -1 or to_read >= 65535)):
if to_read == -1:
to_read = fileobj.length
async def iterator(to_read):
blocksize = getattr(fileobj, 'bs', 8192)
# blocksize = 16384
with fileobj:
while to_read > 0:
chunk = await asynread(fileobj, min(to_read, blocksize))
if chunk:
yield chunk
to_read -= len(chunk)
else:
break
return Response(iterator(to_read),
status=status,
headers=headers,
content_type=content_type)
else:
if to_read:
with fileobj:
body = fileobj.read(to_read)
# print(repr(body))
return Response(body,
headers=headers,
status=status,
content_type=content_type)