mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-23 06:48:16 +00:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
||
|
import http.server
|
||
|
import multiprocessing
|
||
|
import os
|
||
|
import posixpath
|
||
|
import time
|
||
|
import urllib
|
||
|
import webbrowser
|
||
|
from pathlib import Path
|
||
|
|
||
|
import click
|
||
|
|
||
|
ROOT = Path(__file__).parent.parent.resolve()
|
||
|
TEST_RESULT_PATH = ROOT / "tests" / "ui_tests" / "reporting" / "reports" / "test"
|
||
|
|
||
|
|
||
|
class NoCacheRequestHandler(http.server.SimpleHTTPRequestHandler):
|
||
|
def end_headers(self):
|
||
|
self.send_header("Cache-Control", "no-cache, no-store, must-revalidate")
|
||
|
self.send_header("Pragma", "no-cache")
|
||
|
self.send_header("Expires", "0")
|
||
|
return super().end_headers()
|
||
|
|
||
|
def log_message(self, format, *args):
|
||
|
pass
|
||
|
|
||
|
def translate_path(self, path):
|
||
|
# XXX
|
||
|
# Copy-pasted from Python 3.8 BaseHTTPRequestHandler so that we can inject
|
||
|
# the `directory` parameter.
|
||
|
# Otherwise, to keep compatible with 3.6, we'd need to mess with CWD. Which is
|
||
|
# unstable when we expect it to be erased and recreated under us.
|
||
|
path = path.split("?", 1)[0]
|
||
|
path = path.split("#", 1)[0]
|
||
|
# Don't forget explicit trailing slash when normalizing. Issue17324
|
||
|
trailing_slash = path.rstrip().endswith("/")
|
||
|
try:
|
||
|
path = urllib.parse.unquote(path, errors="surrogatepass")
|
||
|
except UnicodeDecodeError:
|
||
|
path = urllib.parse.unquote(path)
|
||
|
path = posixpath.normpath(path)
|
||
|
words = path.split("/")
|
||
|
words = filter(None, words)
|
||
|
path = str(TEST_RESULT_PATH) # XXX this is the only modified line
|
||
|
for word in words:
|
||
|
if os.path.dirname(word) or word in (os.curdir, os.pardir):
|
||
|
# Ignore components that are not a simple file/directory name
|
||
|
continue
|
||
|
path = os.path.join(path, word)
|
||
|
if trailing_slash:
|
||
|
path += "/"
|
||
|
return path
|
||
|
|
||
|
|
||
|
def launch_http_server(port):
|
||
|
http.server.test(HandlerClass=NoCacheRequestHandler, bind="localhost", port=port)
|
||
|
|
||
|
|
||
|
@click.command()
|
||
|
@click.option("-p", "--port", type=int, default=8000)
|
||
|
def main(port):
|
||
|
httpd = multiprocessing.Process(target=launch_http_server, args=(port,))
|
||
|
httpd.start()
|
||
|
time.sleep(0.5)
|
||
|
webbrowser.open(f"http://localhost:{port}/")
|
||
|
httpd.join()
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
main()
|