blob: bb2071cb9769e7d2358344fe1782295b8a63e023 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
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")
|
