|
import os |
|
from http.server import HTTPServer, SimpleHTTPRequestHandler |
|
import argparse |
|
|
|
class CORSHTTPRequestHandler(SimpleHTTPRequestHandler): |
|
def end_headers(self): |
|
self.send_header('Access-Control-Allow-Origin', '*') |
|
self.send_header('Access-Control-Allow-Methods', 'GET') |
|
self.send_header('Cache-Control', 'no-store, no-cache, must-revalidate') |
|
return super().end_headers() |
|
|
|
def do_OPTIONS(self): |
|
self.send_response(200) |
|
self.send_header('Access-Control-Allow-Origin', '*') |
|
self.send_header('Access-Control-Allow-Methods', 'GET, OPTIONS') |
|
self.send_header('Access-Control-Allow-Headers', 'X-Requested-With') |
|
self.send_header('Access-Control-Allow-Headers', 'Content-Type') |
|
self.end_headers() |
|
|
|
if __name__ == '__main__': |
|
parser = argparse.ArgumentParser() |
|
parser.add_argument('--port', type=int, default=7860, help='Port to run the server on') |
|
args = parser.parse_args() |
|
|
|
server_address = ('', args.port) |
|
httpd = HTTPServer(server_address, CORSHTTPRequestHandler) |
|
print(f"Starting server on port {args.port}...") |
|
httpd.serve_forever() |
|
|