blob: f24638ab3ac1d9a85c8ff802fb08338e079c6631 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
import datetime
import json
import time
def _url_dir(request):
return u'/'.join(request.url_parts.path.split(u'/')[:-1]) + u'/'
def store_request_timing_and_headers(request):
"""Store the current timestamp and request's headers in the stash object of
the server. The request must a GET request and must have the "id" parameter.
"""
id = request.GET.first(b"id")
timestamp = datetime.datetime.now().timestamp()
value = {
"timestamp": timestamp,
"headers": request.raw_headers,
}
url_dir = _url_dir(request)
request.server.stash.put(id, value, url_dir)
def get_request_timing_and_headers(request, id=None):
"""Get previously stored timestamp and request headers associated with the
given "id". When "id" is not given the id is retrieved from "request".
"""
if id is None:
id = request.GET.first(b"id")
url_dir = _url_dir(request)
item = request.server.stash.take(id, url_dir)
if not item:
return None
return json.dumps(item)
def wait_for_preload_to_finish(request, id):
"""Wait until a preload associated with "id" is sent."""
while True:
if get_request_timing_and_headers(request, id):
break
time.sleep(0.1)
time.sleep(0.1)
|
