timeout_on_seek.py (1799B)
1 import os 2 import re 3 4 def parse_range(header_value, file_size): 5 if header_value is None: 6 # HTTP Range header range end is inclusive 7 return 0, file_size - 1 8 9 match = re.match(r"bytes=(\d*)-(\d*)", header_value.decode()) 10 start = int(match.group(1)) if match.group(1).strip() != "" else 0 11 last = int(match.group(2)) if match.group(2).strip() != "" else file_size - 1 12 return start, last 13 14 def main(request, response): 15 file_extension = request.GET.first(b"extension").decode() 16 with open(f"media/movie_300.{file_extension}", "rb") as f: 17 f.seek(0, os.SEEK_END) 18 file_size = f.tell() 19 20 range_header = request.headers.get("range") 21 req_start, req_last = parse_range(range_header, file_size) 22 f.seek(req_start, os.SEEK_SET) 23 24 response.add_required_headers = False 25 response.writer.write_status(206 if range_header else 200) 26 response.writer.write_header("Accept-Ranges", "bytes") 27 response.writer.write_header("Content-Type", f"video/{file_extension}") 28 if range_header: 29 response.writer.write_header("Content-Range", f"bytes {req_start}-{req_last}/{file_size}") 30 response.writer.write_header("Content-Length", str(req_last - req_start + 1)) 31 response.writer.end_headers() 32 33 gap_start = int(file_size * 0.5) 34 gap_last = int(file_size * 0.95) 35 36 if gap_start < req_start < gap_last: 37 # If the start position is part of the gap, don't send any data 38 return 39 40 if req_start < gap_start: 41 # If the position is before of the gap, only send data until the 42 # gap is reached 43 req_last = min(req_last, gap_start) 44 45 size = req_last - req_start + 1 46 response.writer.write(f.read(size))