Source code

Revision control

Copy as Markdown

Other Tools

"""
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