Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 1 | import http.server |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 2 | import multiprocessing |
| 3 | import os |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 4 | from socketserver import ThreadingMixIn |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 5 | |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 6 | class HTTPServer(ThreadingMixIn, http.server.HTTPServer): |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 7 | |
| 8 | def server_start(self, root_dir): |
| 9 | import signal |
| 10 | signal.signal(signal.SIGTERM, signal.SIG_DFL) |
| 11 | os.chdir(root_dir) |
| 12 | self.serve_forever() |
| 13 | |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 14 | class HTTPRequestHandler(http.server.SimpleHTTPRequestHandler): |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 15 | |
| 16 | def log_message(self, format_str, *args): |
| 17 | pass |
| 18 | |
| 19 | class HTTPService(object): |
| 20 | |
| 21 | def __init__(self, root_dir, host=''): |
| 22 | self.root_dir = root_dir |
| 23 | self.host = host |
| 24 | self.port = 0 |
| 25 | |
| 26 | def start(self): |
| 27 | self.server = HTTPServer((self.host, self.port), HTTPRequestHandler) |
| 28 | if self.port == 0: |
| 29 | self.port = self.server.server_port |
| 30 | self.process = multiprocessing.Process(target=self.server.server_start, args=[self.root_dir]) |
| 31 | self.process.start() |
| 32 | |
| 33 | def stop(self): |
| 34 | self.server.server_close() |
| 35 | self.process.terminate() |
| 36 | self.process.join() |