mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-28 16:26:10 +03:00
53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
|
# Copyright 2018 The Chromium Authors. All rights reserved.
|
||
|
|
# Use of this source code is governed by a BSD-style license that can be
|
||
|
|
# found in the LICENSE file.
|
||
|
|
|
||
|
|
"""Runs a server to let the user interact with supersize using a web UI."""
|
||
|
|
|
||
|
|
import BaseHTTPServer
|
||
|
|
import logging
|
||
|
|
import os
|
||
|
|
import SimpleHTTPServer
|
||
|
|
|
||
|
|
|
||
|
|
class SupersizeHTTPRequestHandler(SimpleHTTPServer.SimpleHTTPRequestHandler,
|
||
|
|
object):
|
||
|
|
# Directory to serve files from
|
||
|
|
serve_from = None
|
||
|
|
# Path to data file
|
||
|
|
data_file_path = None
|
||
|
|
|
||
|
|
#override
|
||
|
|
def translate_path(self, path):
|
||
|
|
f = super(SupersizeHTTPRequestHandler, self).translate_path(path)
|
||
|
|
relative_path = os.path.relpath(f, os.getcwd())
|
||
|
|
if relative_path == 'data.ndjson':
|
||
|
|
return SupersizeHTTPRequestHandler.data_file_path
|
||
|
|
else:
|
||
|
|
return os.path.join(SupersizeHTTPRequestHandler.serve_from, relative_path)
|
||
|
|
|
||
|
|
|
||
|
|
def AddArguments(parser):
|
||
|
|
parser.add_argument('report_file',
|
||
|
|
help='Path to a custom html_report data file to load.')
|
||
|
|
parser.add_argument('-p', '--port', type=int, default=8000,
|
||
|
|
help='Port for the HTTP server')
|
||
|
|
parser.add_argument('-a', '--address', default='localhost',
|
||
|
|
help='Address for the HTTP server')
|
||
|
|
|
||
|
|
|
||
|
|
def Run(args, _parser):
|
||
|
|
logging.info('Starting server')
|
||
|
|
server_addr = (args.address, args.port)
|
||
|
|
|
||
|
|
static_files = os.path.join(os.path.dirname(__file__), 'static')
|
||
|
|
|
||
|
|
SupersizeHTTPRequestHandler.serve_from = static_files
|
||
|
|
SupersizeHTTPRequestHandler.data_file_path = args.report_file
|
||
|
|
httpd = BaseHTTPServer.HTTPServer(server_addr, SupersizeHTTPRequestHandler)
|
||
|
|
|
||
|
|
sa = httpd.socket.getsockname()
|
||
|
|
logging.warning('Server ready at http://%s:%d?data_url=data.ndjson',
|
||
|
|
sa[0], sa[1])
|
||
|
|
httpd.serve_forever()
|