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
|
import argparse
import os
import http.server
import socketserver
import socket
from threading import Thread
class Server(socketserver.ThreadingTCPServer):
allow_reuse_address = True
class FileServer(object):
def __init__(self, port, root=os.path.abspath(os.curdir), run=False):
if isinstance(port, str):
port = int(port)
self.root = root
self.port = port
self.handler = http.server.SimpleHTTPRequestHandler
self.httpd = None
if run:
self.run()
def run(self, forever=False):
os.chdir(self.root)
socketserver.TCPServer.allow_reuse_address = True
self.httpd = Server(('localhost', self.port), self.handler, True)
print('{0} active on port {1} ({2})'.format(self.__class__.__name__, self.port, self.root))
if not forever:
self.httpd.handle_request()
else:
self.httpd.serve_forever()
self.close()
def close(self):
self.httpd.server_close()
def main():
parser = argparse.ArgumentParser()
parser.add_argument('-r', '--root', default=os.path.abspath(os.curdir), help='Path to files')
parser.add_argument('-p', '--port', type=int, default=8888, help='TCP port')
parser.add_argument('-s', '--single', action='store_false')
args = parser.parse_args()
fileserver = FileServer(args.port, args.root)
fileserver.run(forever=args.single)
if __name__ == '__main__':
main()
|
