mirror of
https://github.com/sphinx-doc/sphinx.git
synced 2025-02-25 18:55:22 -06:00
28 lines
675 B
Python
28 lines
675 B
Python
|
import contextlib
|
||
|
|
import http.server
|
||
|
|
import threading
|
||
|
|
|
||
|
|
|
||
|
|
class HttpServerThread(threading.Thread):
|
||
|
|
def __init__(self, handler, *args, **kwargs):
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
self.server = http.server.HTTPServer(("localhost", 7777), handler)
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
self.server.serve_forever(poll_interval=0.01)
|
||
|
|
|
||
|
|
def terminate(self):
|
||
|
|
self.server.shutdown()
|
||
|
|
self.server.server_close()
|
||
|
|
self.join()
|
||
|
|
|
||
|
|
|
||
|
|
@contextlib.contextmanager
|
||
|
|
def http_server(handler):
|
||
|
|
server_thread = HttpServerThread(handler, daemon=True)
|
||
|
|
server_thread.start()
|
||
|
|
try:
|
||
|
|
yield server_thread
|
||
|
|
finally:
|
||
|
|
server_thread.terminate()
|