Commit 369ce905 authored by Madhur Tandon's avatar Madhur Tandon

Adhere to flake8

parent 0a72c4ef
...@@ -8,23 +8,31 @@ import pathlib ...@@ -8,23 +8,31 @@ import pathlib
TEST_PATH = pathlib.Path(__file__).parents[0].resolve() TEST_PATH = pathlib.Path(__file__).parents[0].resolve()
BUILD_PATH = TEST_PATH / '..' / 'build' BUILD_PATH = TEST_PATH / '..' / 'build'
class Handler(http.server.SimpleHTTPRequestHandler): class Handler(http.server.SimpleHTTPRequestHandler):
def end_headers(self): def end_headers(self):
self.send_header('Access-Control-Allow-Origin', '*') self.send_header('Access-Control-Allow-Origin', '*')
super().end_headers() super().end_headers()
Handler.extensions_map['.wasm'] = 'application/wasm' Handler.extensions_map['.wasm'] = 'application/wasm'
def make_parser(parser): def make_parser(parser):
parser.description = ('Start a server with the supplied build_dir and port.') parser.description = ('Start a server with the supplied '
parser.add_argument('--build_dir', action='store', type=str, default=BUILD_PATH, help='set the build directory') 'build_dir and port.')
parser.add_argument('--port', action='store', type=int, default=8000, help='set the PORT number') parser.add_argument('--build_dir', action='store', type=str,
default=BUILD_PATH, help='set the build directory')
parser.add_argument('--port', action='store', type=int,
default=8000, help='set the PORT number')
return parser return parser
def server(port): def server(port):
httpd = socketserver.TCPServer(('', port), Handler) httpd = socketserver.TCPServer(('', port), Handler)
return httpd return httpd
def main(args): def main(args):
build_dir = args.build_dir build_dir = args.build_dir
port = args.port port = args.port
...@@ -38,7 +46,8 @@ def main(args): ...@@ -38,7 +46,8 @@ def main(args):
httpd.shutdown() httpd.shutdown()
sys.exit() sys.exit()
if __name__ == "__main__": if __name__ == "__main__":
parser = make_parser(argparse.ArgumentParser()) parser = make_parser(argparse.ArgumentParser())
args = parser.parse_args() args = parser.parse_args()
main(args) main(args)
\ No newline at end of file
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment