-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathserve.py
39 lines (30 loc) · 1.33 KB
/
serve.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#!/usr/bin/env python3
import http.server
import socketserver
import argparse
import functools
import os
import signal
import sys
class CORSRequestHandler (http.server.SimpleHTTPRequestHandler):
def end_headers (self):
self.send_header('Access-Control-Allow-Origin', '*')
self.send_header('Cross-Origin-Opener-Policy', 'same-origin')
self.send_header('Cross-Origin-Embedder-Policy', 'require-corp')
http.server.SimpleHTTPRequestHandler.end_headers(self)
def signal_handler(sig, frame):
print('You pressed Ctrl+C!')
sys.exit(0)
def main():
parser = argparse.ArgumentParser(description='Simple HTTP Server with CORS and COOP/COEP headers.')
parser.add_argument('-d', '--directory', default='.', help='Directory to serve.')
parser.add_argument('-p', '--port', type=int, default=8000, help='Port to listen on.')
args = parser.parse_args()
handler_class = functools.partial(CORSRequestHandler, directory=os.path.abspath(args.directory))
with socketserver.TCPServer(("", args.port), handler_class) as httpd:
signal.signal(signal.SIGINT, signal_handler)
print(f"Serving HTTP on port {args.port} (http://localhost:{args.port}/) ...")
print(f"Serving directory: {os.path.abspath(args.directory)}")
httpd.serve_forever()
if __name__ == '__main__':
main()