Source code
Revision control
Copy as Markdown
Other Tools
import os.path
from wptserve.utils import isomorphic_decode
def main(request, response):
token = request.GET.first(b"token", None)
is_query = request.GET.first(b"query", None) != None
with request.server.stash.lock:
value = request.server.stash.take(token)
count = 0
if value != None:
count = int(value)
if is_query:
if count < 2:
request.server.stash.put(token, count)
else:
count = count + 1
request.server.stash.put(token, count)
if is_query:
headers = [(b"Count", count)]
content = b""
return 200, headers, content
else:
filename = u"green-16x16.png"
if count > 1:
filename = u"green-256x256.png"
path = os.path.join(os.path.dirname(isomorphic_decode(__file__)), u"../../../images", filename)
body = open(path, "rb").read()
response.add_required_headers = False
response.writer.write_status(200)
response.writer.write_header(b"content-length", len(body))
response.writer.write_header(b"Cache-Control", b"private, max-age=0, stale-while-revalidate=60")
response.writer.write_header(b"content-type", b"image/png")
response.writer.end_headers()
response.writer.write(body)