blob: 32cb9a9ba30638bf0e5c7e54655ae87e7085978d (
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
|
"""An HTTP request handler for WPT that handles /get_beacon.py requests."""
import json
_BEACON_ID_KEY = b"uuid"
_BEACON_DATA_PATH = "beacon_data"
def main(request, response):
"""Retrieves the beacon data keyed by the given uuid from server storage.
The response content is a JSON string in one of the following formats:
- "{'data': ['abc', null, '123',...]}"
- "{'data': []}" indicates that no data has been set for this uuid.
"""
if _BEACON_ID_KEY not in request.GET:
response.status = 400
return "Must provide a UUID to store beacon data"
uuid = request.GET.first(_BEACON_ID_KEY)
with request.server.stash.lock:
body = {'data': []}
data = request.server.stash.take(key=uuid, path=_BEACON_DATA_PATH)
if data:
body['data'] = data
# The stash is read-once/write-once, so it has to be put back after
# reading if `data` is not None.
request.server.stash.put(
key=uuid, value=data, path=_BEACON_DATA_PATH)
return [(b'Content-Type', b'text/plain')], json.dumps(body)
|