preload-finished-while-receiving-final-response-body.h2.py (1133B)
1 import importlib 2 import os 3 4 utils = importlib.import_module("loading.early-hints.resources.utils") 5 6 7 def handle_headers(frame, request, response): 8 resource_url = request.GET.first(b"resource-url").decode() 9 link_header_value = "<{}>; rel=preload; as=script".format(resource_url) 10 early_hints = [ 11 (b":status", b"103"), 12 (b"link", link_header_value), 13 ] 14 response.writer.write_raw_header_frame(headers=early_hints, 15 end_headers=True) 16 17 response.status = 200 18 response.headers[b"content-type"] = "text/html" 19 response.write_status_headers() 20 21 22 def main(request, response): 23 # Wait for preload to finish before sending the response body. 24 resource_id = request.GET.first(b"resource-id").decode() 25 utils.wait_for_preload_to_finish(request, resource_id) 26 27 current_dir = os.path.dirname(os.path.realpath(__file__)) 28 file_path = os.path.join(current_dir, "preload-finished-while-receiving-final-response-body.html") 29 with open(file_path, "r") as f: 30 test_content = f.read() 31 response.writer.write_data(item=test_content, last=True)