简单http和https服务器python脚本

欢迎加入python学习交流群 667279387

工作经常要用到测试http和https协议，这里写了两个简单的脚本实现简单的http服务器和https服务器。

http服务器代码

import sys
import BaseHTTPServer
from SimpleHTTPServer import SimpleHTTPRequestHandler
HandlerClass = SimpleHTTPRequestHandler
ServerClass = BaseHTTPServer.HTTPServer
Protocol = "HTTP/1.0"

if sys.argv[1:]:
  port = int(sys.argv[1])
else:
  port = 8000
server_address = ('127.0.0.1', port) 

HandlerClass.protocol_version = Protocol
httpd = ServerClass(server_address, HandlerClass) 

sa = httpd.socket.getsockname()
print "Serving HTTP on", sa[0], "port", sa[1], "..."
httpd.serve_forever()

https服务器代码

# taken from https://gist.github.com/dergachev/7028596
#
# generate server.xml with the following command:
#   openssl req -new -x509 -keyout https_svr_key.pem -out https_svr_key.pem -days 3650 -nodes
#
import BaseHTTPServer
import SimpleHTTPServer
import os
import socket
import ssl  

script_home = os.path.dirname(os.path.abspath(__file__))
ip = [(s.connect(('8.8.8.8', 53)), s.getsockname()[0], s.close())
      for s in [socket.socket(socket.AF_INET, socket.SOCK_DGRAM)]][0][1]
port = 8888  

def main():
    print ("simple https server, address:%s:%d, document root:%s" % (ip, port, script_home))  

    httpd = BaseHTTPServer.HTTPServer(('127.0.0.1', port), SimpleHTTPServer.SimpleHTTPRequestHandler)
    httpd.socket = ssl.wrap_socket(httpd.socket, certfile='./https_svr_key.pem', server_side=True)
    httpd.serve_forever()  

if __name__ == '__main__':
    os.chdir(script_home)
    main()

欢迎加入python学习交流群 667279387