2019-04-02 19:10:09 +00:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
|
|
|
|
import argparse
|
|
|
|
|
import http.server
|
|
|
|
|
import socketserver
|
|
|
|
|
import logging
|
2020-09-11 16:09:49 +00:00
|
|
|
import pprint
|
2019-04-02 19:10:09 +00:00
|
|
|
|
2021-03-12 15:44:41 +00:00
|
|
|
#__name__ = "python-web-server"
|
|
|
|
|
__version__ = (1, 1, 1)
|
|
|
|
|
|
2019-04-02 19:10:09 +00:00
|
|
|
# setting up logging for this script
|
|
|
|
|
_level = logging.INFO
|
|
|
|
|
_format = "%(asctime)-15s [%(levelname)-8s] %(lineno)d : %(funcName)s : %(message)s"
|
|
|
|
|
logging.basicConfig(format=_format, level=_level)
|
|
|
|
|
log = logging.getLogger()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SimpleHandler(http.server.BaseHTTPRequestHandler):
|
|
|
|
|
|
2020-09-09 10:43:52 +00:00
|
|
|
def do_HEAD(self):
|
|
|
|
|
log.info(f"server version: {self.server_version}")
|
|
|
|
|
log.info(f"client address: {self.client_address}")
|
|
|
|
|
log.info(f"path: {self.path}")
|
|
|
|
|
log.info(f"headers: {self.headers}")
|
|
|
|
|
log.info(f"request version: {self.request_version}")
|
|
|
|
|
|
|
|
|
|
self.send_response(200)
|
|
|
|
|
self.send_header('Content-type', 'text/html')
|
|
|
|
|
self.end_headers()
|
|
|
|
|
|
2020-09-14 13:21:35 +00:00
|
|
|
headers = "\n".join([f"{k}: {v}" for k, v in self.headers.items()])
|
|
|
|
|
|
2020-09-09 10:43:52 +00:00
|
|
|
if self.path != '/':
|
2020-09-14 13:21:35 +00:00
|
|
|
uri = f"{self.path[1:]} "
|
2020-09-09 10:43:52 +00:00
|
|
|
else:
|
2020-09-14 13:21:35 +00:00
|
|
|
uri = ""
|
|
|
|
|
|
|
|
|
|
message = f"""Hello, {uri}World!
|
2021-03-12 15:44:41 +00:00
|
|
|
App version: {__version__}
|
2020-09-14 13:21:35 +00:00
|
|
|
Server version: {self.server_version}
|
|
|
|
|
Client address: {self.client_address}
|
|
|
|
|
Path: {self.path}
|
|
|
|
|
Headers:
|
|
|
|
|
{pprint.pformat(self.headers)}
|
|
|
|
|
Request version: {self.request_version}
|
|
|
|
|
"""
|
2020-09-09 10:43:52 +00:00
|
|
|
|
|
|
|
|
self.wfile.write(bytes(message, encoding='utf-8'))
|
|
|
|
|
return
|
|
|
|
|
|
2019-04-02 19:10:09 +00:00
|
|
|
def do_GET(self):
|
|
|
|
|
log.info(f"server version: {self.server_version}")
|
|
|
|
|
log.info(f"client address: {self.client_address}")
|
2020-08-24 20:44:28 +00:00
|
|
|
log.info(f"path: {self.path}")
|
|
|
|
|
log.info(f"headers: {self.headers}")
|
|
|
|
|
log.info(f"request version: {self.request_version}")
|
|
|
|
|
|
2019-04-02 19:10:09 +00:00
|
|
|
self.send_response(200)
|
|
|
|
|
self.send_header('Content-type', 'text/html')
|
|
|
|
|
self.end_headers()
|
2020-08-24 20:44:28 +00:00
|
|
|
|
2020-09-11 16:09:49 +00:00
|
|
|
headers = "<br/>".join([f"{k}: {v}" for k, v in self.headers.items()])
|
|
|
|
|
|
2020-08-24 20:44:28 +00:00
|
|
|
if self.path != '/':
|
2020-09-14 13:21:35 +00:00
|
|
|
uri = f"{self.path[1:]} "
|
2020-08-24 20:44:28 +00:00
|
|
|
else:
|
2020-09-14 13:21:35 +00:00
|
|
|
uri = ""
|
|
|
|
|
|
|
|
|
|
message = f"""<html><body>
|
|
|
|
|
<h1>Hello, {uri}World!</h1>
|
2021-03-12 15:44:41 +00:00
|
|
|
<p>App version: <b>{__version__}</b></p>
|
2020-09-14 13:21:35 +00:00
|
|
|
<p>Server version: <b>{self.server_version}</b></p>
|
|
|
|
|
<p>Client address: <b>{self.client_address}</b></p>
|
|
|
|
|
<p>Path: <b>{self.path}</b></p>
|
|
|
|
|
<p>Headers:</p> <p><code>{pprint.pformat(self.headers)}</code></p>
|
|
|
|
|
<p>Request version: <b>{self.request_version}</b></p>
|
|
|
|
|
</body></html>"""
|
2020-08-24 20:44:28 +00:00
|
|
|
|
|
|
|
|
self.wfile.write(bytes(message, encoding='utf-8'))
|
2019-04-02 19:10:09 +00:00
|
|
|
return
|
|
|
|
|
|
|
|
|
|
class SockHandler(socketserver.BaseRequestHandler):
|
|
|
|
|
|
|
|
|
|
def handle(self):
|
|
|
|
|
self.data = self.request.recv(1024).strip()
|
|
|
|
|
log.info(f"client address: {self.client_address}")
|
|
|
|
|
log.info(f"received data:\n{self.data}")
|
|
|
|
|
self.request.sendall(b"Hello, World!\n")
|
|
|
|
|
|
|
|
|
|
|
2021-03-12 15:44:41 +00:00
|
|
|
def _version(ver=__version__):
|
|
|
|
|
"""
|
|
|
|
|
Stringify version value
|
|
|
|
|
"""
|
|
|
|
|
return f"{ver[0]}.{ver[1]}.{ver[2]}"
|
|
|
|
|
|
|
|
|
|
|
2019-04-02 19:10:09 +00:00
|
|
|
def parse_args():
|
|
|
|
|
"""
|
|
|
|
|
Parse args
|
|
|
|
|
"""
|
|
|
|
|
argp = argparse.ArgumentParser()
|
|
|
|
|
argp.add_argument('-i', '--ip-address', default="", help="IP Address to bind the server")
|
|
|
|
|
argp.add_argument('-p', '--port', type=int, default=8080, help="Port on which to listen for traffic")
|
2021-03-12 15:44:41 +00:00
|
|
|
argp.add_argument('-v', '--version', action='version', version=f"%(prog)s {_version()}", help="Print version")
|
2019-04-02 19:10:09 +00:00
|
|
|
argp.add_argument('-d', '--debug', help="Run in debug mode")
|
|
|
|
|
return argp.parse_args()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
args = parse_args()
|
|
|
|
|
|
|
|
|
|
if args.debug:
|
|
|
|
|
log.setLevel(logging.DEBUG)
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
#simple_handler = SimpleHandler() # http.server.SimpleHTTPRequestHandler
|
|
|
|
|
|
|
|
|
|
with http.server.HTTPServer((args.ip_address, args.port), SimpleHandler) as httpd:
|
|
|
|
|
log.info(f"Server started on {args.ip_address}:{args.port}")
|
|
|
|
|
httpd.serve_forever()
|
|
|
|
|
|
|
|
|
|
#with socketserver.TCPServer((args.ip_address, args.port), SockHandler) as tcpd:
|
|
|
|
|
# log.info(f"Server started on {args.ip_address}:{args.port}")
|
|
|
|
|
# tcpd.serve_forever()
|
|
|
|
|
|
|
|
|
|
except KeyboardInterrupt as control_c:
|
|
|
|
|
log.info("Server stopped")
|
|
|
|
|
|
