httpServer.py

import logging
import multiprocessing
import os
import time
from http.server import SimpleHTTPRequestHandler, HTTPServer

from config import MEDIA_PATH, MEDIA_SERVER_PORT

logger = logging.getLogger()

class CORSRequestHandler(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(CORSRequestHandler, self).end_headers()

def http_server(path, port):
    try:
        os.chdir(path)
        server_address = ("",port) # 设置服务器地址
        server_obj = HTTPServer(server_address, CORSRequestHandler) # 创建服务器对象
        server_obj.serve_forever() # 启动服务器
    except BlockingIOError as e:
        logger.error(f"BlockingIOError: {e}")
        time.sleep(1)
    except KeyboardInterrupt:
        logger.info(f"Stopping Media Server...")
        server_obj.socket.close()
        server_obj.shutdown()
        logger.info(f"Media Server stopped.")
    except Exception as e:
        logger.exception(e)
        return

def run_http_server(path, port):
    p = multiprocessing.Process(target=http_server, args = (path, port))
    p.start()
    logger.info(f"[PID: {p.pid}] Media server started at {path}:{port}")

if __name__ == "__main__":
    run_http_server(MEDIA_PATH, MEDIA_SERVER_PORT)