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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
import http.server
import time
import os
import pyroute2
import socket
import socketserver
import threading
from functools import partial
from pathlib import Path
class ThreadingHTTPServer(socketserver.ThreadingMixIn, http.server.HTTPServer):
pass
class HTTPServer:
def __init__(self, path, remote_ip):
self.path = path
self.port = None
self.old_path = None
self.httpd = None
with pyroute2.IPRoute() as ipr:
r = ipr.route('get', dst=remote_ip)
for attr in r[0]['attrs']:
if attr[0] == 'RTA_PREFSRC':
self.ip = attr[1]
with socket.socket() as s:
s.bind(('', 0))
self.port = s.getsockname()[1]
def get_url(self, filename):
path = Path(self.path) / filename
assert path.exists()
return f"http://{self.ip}:{self.port}/{filename}"
def __enter__(self):
def start_server():
Handler = http.server.SimpleHTTPRequestHandler
self.httpd = ThreadingHTTPServer(("", self.port), Handler)
self.httpd.serve_forever()
# Kind of annoying, but to work with older pythons where
# SimpleHTTPRequestHandler doesn't take a directory parameter but only
# serves the current directory:
self.old_path = os.getcwd()
os.chdir(self.path)
self.thread = threading.Thread(target=start_server)
self.thread.start()
return self
def __exit__(self, type, value, exc):
if self.httpd is not None:
self.httpd.shutdown()
self.httpd.server_close()
if self.old_path is not None:
os.chdir(self.old_path)
if __name__ == '__main__':
with HTTPServer("/tmp", "127.0.0.1") as server:
print("server ip", server.ip)
print("server port", server.port)
time.sleep(300)
|
