Source code

Revision control

Copy as Markdown

Other Tools

import os
from wptserve.utils import isomorphic_decode
# Returns a valid image response when request's |referrer| matches
# |expected_referrer|.
def main(request, response):
referrer = request.headers.get(b"referer", b"")
expected_referrer = request.GET.first(b"expected_referrer", b"")
response_headers = [(b"Content-Type", b"image/png")]
if referrer == expected_referrer:
image_path = os.path.join(os.path.dirname(isomorphic_decode(__file__)), u"image.png")
return (200, response_headers, open(image_path, mode='rb').read())
return (404, response_headers, u"Not found")