summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/speculation-rules/prerender/resources/exec.py
blob: de0636117db6ef6bc7069c6afbae6a99c85eaf8b (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
from wptserve.utils import isomorphic_decode
import os

def main(request, response):
    purpose = request.headers.get(b"purpose")
    if (purpose == b'prefetch' and b"code" in request.GET):
        code = int(request.GET.first(b"code"))
    else:
        code = 200

    if b"uuid" in request.GET:
        path = '/speculation-rules/prerender/resources/exec.py'
        uuid = request.GET.first(b"uuid")
        with request.server.stash.lock:
            count = request.server.stash.take(uuid, path) or 0
            if b"get-fetch-count" in request.GET:
                response.status = 200
                response.content = '%d' % count
                request.server.stash.put(uuid, count, path)
                return
            count += 1
            request.server.stash.put(uuid, count, path)

    with open(os.path.join(os.path.dirname(isomorphic_decode(__file__)), "exec.html"), u"r") as fn:
        response.content = fn.read()
    response.status = code