Skip to content

Commit

Permalink
Python 3: port tests in content-dpr, mimesniff and upgrade-insecure-r… (
Browse files Browse the repository at this point in the history
  • Loading branch information
ziransun authored Jun 26, 2020
1 parent b4b88e7 commit bc6c6ef
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 16 deletions.
24 changes: 13 additions & 11 deletions content-dpr/resources/dpr.py
Original file line number Diff line number Diff line change
@@ -1,24 +1,26 @@
from wptserve.utils import isomorphic_decode

def main(request, response):
"""
Simple handler that sets a response header based on which client hint
request headers were received.
"""

response.headers.append("Access-Control-Allow-Origin", "*")
response.headers.append(b"Access-Control-Allow-Origin", b"*")
values = request.GET
name = values.first('name')
type = values.first('mimeType')
dpr = values.first('dpr')
name = values.first(b'name')
type = values.first(b'mimeType')
dpr = values.first(b'dpr')
double = None
if 'double' in values:
double = values.first('double')
image_path = request.doc_root + "/".join(request.url_parts[2].split("/")[:-1]) + "/" + name
if b'double' in values:
double = values.first(b'double')
image_path = request.doc_root + u"/".join(request.url_parts[2].split(u"/")[:-1]) + u"/" + isomorphic_decode(name)
f = open(image_path, "rb")
buff = f.read()
f.close()
response.headers.set("Content-Type", type)
response.headers.set("Content-DPR", dpr)
response.headers.set(b"Content-Type", type)
response.headers.set(b"Content-DPR", dpr)
if double:
response.headers.append("Content-DPR", double)
response.headers.set("Content-Length", len(buff))
response.headers.append(b"Content-DPR", double)
response.headers.set(b"Content-Length", len(buff))
response.content = buff
12 changes: 7 additions & 5 deletions mimesniff/mime-types/resources/mime-charset.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,18 @@
from wptserve.utils import isomorphic_encode

def main(request, response):
content = "<meta charset=utf-8>\n<script>document.write(document.characterSet)</script>"
content = b"<meta charset=utf-8>\n<script>document.write(document.characterSet)</script>"

# This uses the following rather than
# response.headers.set("Content-Type", request.GET.first("type"));
# response.content = content
# to work around https://github.com/web-platform-tests/wpt/issues/8372.

response.add_required_headers = False
output = "HTTP/1.1 200 OK\r\n"
output += "Content-Length: " + str(len(content)) + "\r\n"
output += "Content-Type: " + request.GET.first("type") + "\r\n"
output += "\r\n"
output = b"HTTP/1.1 200 OK\r\n"
output += b"Content-Length: " + isomorphic_encode(str(len(content))) + b"\r\n"
output += b"Content-Type: " + request.GET.first(b"type") + b"\r\n"
output += b"\r\n"
output += content
response.writer.write(output)
response.close_connection = True

0 comments on commit bc6c6ef

Please sign in to comment.