53 lines
2 KiB
Python
53 lines
2 KiB
Python
"""
|
|
This generates a partial response for a 100-byte text file.
|
|
"""
|
|
import re
|
|
|
|
from wptserve.utils import isomorphic_decode
|
|
|
|
def main(request, response):
|
|
total_length = int(request.GET.first(b'length', b'100'))
|
|
partial_code = int(request.GET.first(b'partial', b'206'))
|
|
content_type = request.GET.first(b'type', b'text/plain')
|
|
range_header = request.headers.get(b'Range', b'')
|
|
|
|
# Send a 200 if there is no range request
|
|
if not range_header:
|
|
to_send = ''.zfill(total_length)
|
|
response.headers.set(b"Content-Type", content_type)
|
|
response.headers.set(b"Cache-Control", b"no-cache")
|
|
response.headers.set(b"Content-Length", total_length)
|
|
response.content = to_send
|
|
return
|
|
|
|
# Simple range parsing, requires specifically "bytes=xxx-xxxx"
|
|
range_header_match = re.search(r'^bytes=(\d*)-(\d*)$', isomorphic_decode(range_header))
|
|
start, end = range_header_match.groups()
|
|
start = int(start)
|
|
end = int(end) if end else total_length
|
|
length = end - start
|
|
|
|
# Error the request if the range goes beyond the length
|
|
if length <= 0 or end > total_length:
|
|
response.set_error(416, u"Range Not Satisfiable")
|
|
# set_error sets the MIME type to application/json, which - for a
|
|
# no-cors media request - will be blocked by ORB. We'll just force
|
|
# the expected MIME type here, whichfixes the test, but doesn't make
|
|
# sense in general.
|
|
response.headers = [(b"Content-Type", content_type)]
|
|
response.write()
|
|
return
|
|
|
|
# Generate a partial response of the requested length
|
|
to_send = ''.zfill(length)
|
|
response.headers.set(b"Content-Type", content_type)
|
|
response.headers.set(b"Accept-Ranges", b"bytes")
|
|
response.headers.set(b"Cache-Control", b"no-cache")
|
|
response.status = partial_code
|
|
|
|
content_range = b"bytes %d-%d/%d" % (start, end, total_length)
|
|
|
|
response.headers.set(b"Content-Range", content_range)
|
|
response.headers.set(b"Content-Length", length)
|
|
|
|
response.content = to_send
|