blob: 182dc1df0e773025297bf97870405619545fc72b (
plain)
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
|
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()
|
