WSGI服务器

import re
from pymysql import connect
import urllib.parse
import logging


URL_FUNC_DICT = dict()

def open_mysql():
    # 创建connect连接
    conn = connect(host='localhost',port=3306,user='root',password='m',database='stock',charset='utf8')
    # 获得cursor对象
    cs = conn.cursor()
    return conn, cs

def close_mysql(conn, cs):
    # 关闭连接、对象
    cs.close()
    conn.close()

def route(path_info):
    '''装饰器：添加方法-路径到字典'''
    def set_func(func):
        URL_FUNC_DICT[path_info] = func
        def call_func(*args, **kwargs):
            return func(*args, **kwargs)
        return call_func
    return set_func

@route(r"/index.html")
def index(ret):
    # 打开、读取网页
    with open("./templates/index.html", encoding="utf-8") as f:
        content = f.read()
    # 打开数据库
    conn, cs = open_mysql()
    # 执行sql语句
    cs.execute("select * from info;")
    # 读取数据库数据
    stock_infos = cs.fetchall()
    # 关闭数据库
    close_mysql(conn, cs)
    # 数据库数据替换模板数据
    tr_template = """<tr>
        <td>%s</td>
        <td>%s</td>
        <td>%s</td>
        <td>%s</td>
        <td>%s</td>
        <td>%s</td>
        <td>%s</td>
        <td>%s</td>
        <td>
            <input type="button" value="添加" id="toAdd" name="toAdd" systemidvaule="%s">
        </td>
        </tr>
    """
    html = ""
    for line_info in stock_infos:
        html += tr_template % (line_info[0],line_info[1],line_info[2],line_info[3],line_info[4],line_info[5],line_info[6],line_info[7], line_info[1])
    return re.sub(r"{%content%}", html, content)

@route(r"/center.html")
def center(ret):
    # 打开读取网页
    with open("./templates/center.html", encoding="utf-8") as f:
        content = f.read()
    # 打开数据库
    conn, cs = open_mysql()
    # 执行sql语句
    cs.execute("select i.code, i.short, i.chg, i.turnover, i.price, i.highs, f.note_info from info as i inner join focus as f on i.id = f.info_id;")
    # 读取sql语句结果
    stock_infos = cs.fetchall()
    # 关闭数据库
    close_mysql(conn, cs)
    tr_template = """
        <tr>
            <td>%s</td>
            <td>%s</td>
            <td>%s</td>
            <td>%s</td>
            <td>%s</td>
            <td>%s</td>
            <td>%s</td>
            <td>
                <a type="button" class="btn btn-default btn-xs" href="/update/%s.html"> <span class="glyphicon glyphicon-star" aria-hidden="true"></span> 修改 </a>
            </td>
            <td>
                <input type="button" value="删除" id="toDel" name="toDel" systemidvaule="%s">
            </td>
        </tr>
    """
    # 用数据库内容替换模板数据
    html = ""
    for stock_info in stock_infos:
        html += tr_template % (stock_info[0], stock_info[1], stock_info[2], stock_info[3],stock_info[4], stock_info[5], stock_info[6], stock_info[0], stock_info[0])
    return re.sub(r"{%content%}", html, content)

@route(r"^/add/(d+).html$")
def add_focus(ret):
    '''添加股票'''
    # 获取股票代码
    stock_code = ret.group(1)
    # 打开数据库，查询股票代码
    conn, cs = open_mysql()
    sql = "select * from info where code = %s;"
    cs.execute(sql, (stock_code,))
    # 股票代码不在数据库中
    if not cs.fetchall():
        # 返回提示
        close_mysql(conn, cs)
        return "没有这只股票代码，请收下留情。。"
    # 判断股票代码是否已存在于focus表数据库中
    sql = "select * from info as i inner join focus as f on i.id = f.info_id where i.code = %s"
    cs.execute(sql, (stock_code,))
    # 已存在于focus表
    if cs.fetchall():
        close_mysql(conn, cs)
        return "已成功关注，请勿重复关注。。"
    # 不存在于focu表
    # 执行sql语句添加股票代码到focus表
    sql = "insert into focus(info_id) select id from info where code = %s;"
    cs.execute(sql, (stock_code))
    conn.commit()
    # 关闭数据库
    close_mysql(conn, cs)
    return "关注成功。。"

@route(r"^/del/(d+).html$")
def del_focus(ret):
    '''取消关注股票'''
    stock_code = ret.group(1)
    print(stock_code)
    conn, cs = open_mysql()
    sql = "select * from focus as f inner join info as i on i.id = f.info_id where i.code = %s;"
    cs.execute(sql, (stock_code,))
    # 股票代码不存在于focus表
    if not cs.fetchall():
        close_mysql(conn, cs)
        return "未曾关注，谈何取消。。"
    sql = "delete from focus where info_id = (select id from info where code = %s);"
    cs.execute(sql, (stock_code,))
    conn.commit()
    close_mysql(conn, cs)
    return "已成功取关注。。"

@route(r"/update/(d+).html")
def update_info(ret):
    '''更新数据'''
    # 获取股票代码
    stock_code = ret.group(1)
    # 打开数据库，查询股票代码
    conn, cs = open_mysql()
    sql = "select * from info where code = %s;"
    cs.execute(sql, (stock_code,))
    # 股票代码不在数据库中
    if not cs.fetchall():
        # 返回提示
        close_mysql(conn, cs)
        return "没有这只股票代码，请收下留情。。"
    # 判断股票代码是否已存在于focus表数据库中
    sql = "select * from info as i inner join focus as f on i.id = f.info_id where i.code = %s"
    cs.execute(sql, (stock_code,))
    # 已存在于focus表
    if not cs.fetchall():
        close_mysql(conn, cs)
        return "没有关注，请勿输入备注。。"
    # 打开网页
    with open("./templates/update.html", encoding="utf-8") as f:
        content = f.read()
    # 替换模板,回传数据
    content = re.sub(r"{%note_info%}","请输入修改内容", content)
    return re.sub(r"{%code%}", stock_code, content)
    # return "更新数据成功。。"

@route(r"/update/(d+)/(.*).html")
def save_update(ret):
    stock_code = ret.group(1)
    new_info = ret.group(2)
    new_info = urllib.parse.unquote(new_info)
    # 打开数据库，查询股票代码
    conn, cs = open_mysql()
    sql = "select * from info where code = %s;"
    cs.execute(sql, (stock_code,))
    # 股票代码不在数据库中
    if not cs.fetchall():
        # 返回提示
        close_mysql(conn, cs)
        return "没有这只股票代码，请收下留情。。"
    # 判断股票代码是否已存在于focus表数据库中
    sql = "select * from info as i inner join focus as f on i.id = f.info_id where i.code = %s"
    cs.execute(sql, (stock_code,))
    # 不存在于focu表
    if not cs.fetchall():
        close_mysql(conn, cs)
        return "未曾关注，请勿修改。。"
    # 已存在于focus表
    # 执行sql语句添加股票代码到focus表
    sql = "update focus set note_info=%s where info_id = (select id from info where code=%s);"
    cs.execute(sql, (new_info, stock_code))
    conn.commit()
    # 关闭数据库
    close_mysql(conn, cs)
    return "修改成功。。"

def application(env, start_header):
    start_header("200 OK", [("Content-type", "text/html;charset=utf-8")])
    # 获取请求文件信息
    file_name = env["PATH_INFO"]
    file_name = urllib.parse.unquote(file_name)
    # 日志设定
    # 创建logger
    logger = logging.getLogger()
    logger.setLevel(logging.DEBUG)
    # 创建一个handler，用于写入文件
    logfile = "./log.txt"
    fh = logging.FileHandler(logfile,mode='a')
    fh.setLevel(logging.INFO)
    # 创建一个handler，用于输出控制台
    ch = logging.StreamHandler()
    ch.setLevel(logging.WARNING)
    # 定义日志输出格式
    formatter = logging.Formatter("%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s")
    fh.setFormatter(formatter)
    ch.setFormatter(formatter)
    # 将logger添加到handler中
    logger.addHandler(fh)
    logger.addHandler(ch)
    logger.info("客户访问：%s" % file_name)
    try:
        for url, func in URL_FUNC_DICT.items():
            ret = re.match(url, file_name)
            if ret:
                return func(ret)
        else:
            logger.warning("请求错误。file(%s) not found" % file_name)
            return "请求错误。file(%s) not found" % file_name
    except Exception as e:
        logger.warning("程序异常。错误信息：%s" % e)
        return "请求错误。logger错误信息：%s" % e

import sys
import re
import socket
import multiprocessing

class WSGIServer(object):
    '''服务器运行对象'''
    def __init__(self, port, app, static_path):
        '''对象初始化'''
        # 创建套接字
        self.tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        # 绑定端口 IP
        self.tcp_server_socket.bind(("", port))
        # 设置监听套接字
        self.tcp_server_socket.listen(128)
        # web框架调用方法
        self.application = app
        # 静态网页存放地址
        self.static_path = static_path

    def client_serve(self, client_socket):
        '''为浏览器实现具体服务'''
        # 获取浏览器请求
        request = client_socket.recv(1024).decode("utf-8")
        request_lines = request.splitlines()
        # 解析浏览器请求
        ret = re.match(r"[^/]+(/[^ ]*)", request_lines[0])
        file_name = ""
        if ret:
            file_name = ret.group(1)
            if file_name == "/":
                file_name = "/index.html"
        # print("file_nema:", file_name)
        # 如果为静态页面请求（非html请求）
        if not file_name.endswith(".html"):
            # 打开请求页面
            try:
                with open(self.static_path + file_name, "rb") as f:
                    response_body = f.read()
            # 打开页面失败
            except:
                response_body = "file not found"
                response_header = "HTTP/1.1 404 FILE NOT FOUND
"
                response_header += "
"
                response = response_header + response_body
                client_socket.send(response.encode("utf-8"))
            # 打开页面成功
            else:
                response_header = "HTTP/1.1 200 OK
"
                response_header += "
"
                response = response_header.encode("utf-8") + response_body
                client_socket.send(response)
        # 如果为动态页面请求
        else:
            env = dict()
            env["PATH_INFO"] = file_name
            # print(file_name)
            # 调用web框架方法，处理页面
            response_body = self.application(env, self.set_headers)

            response_header = "HTTP/1.1 %s 
" % self.status
            for header in self.header_list:
                response_header += "%s:%s
" %(header[0],header[1])
            response_header += "
"

            response = response_header + response_body
            client_socket.send(response.encode("utf-8"))
        # 关闭套接字
        client_socket.close()

    def set_headers(self, status, header_list):
        '''存放web框架请求头信息'''
        self.status = status
        self.header_list = header_list

    def fun_forever(self):
        '''服务器对象流程控制'''
        while True:
            # print("等待浏览器连接")
            client_socket, client_addr = self.tcp_server_socket.accept()
            # 利用多进程实现多任务操作
            # print("创建子线程")
            p = multiprocessing.Process(target=self.client_serve, args=(client_socket,))
            p.start()
            client_socket.close()

def error_msg():
    print("服务器开启失败，请按以下方式输入")
    print("python xxx.py 8080 mini_frame:application")

def main():
    '''实现主体流程控制'''
    # # 获取服务器端口、web_frame框架地址、方法名
    # # 如果程序调用参数为3，输入正确
    # if len(sys.argv) == 3:
    #     # 参数输入方式正确
    #     try:
    #         port = int(sys.argv[1])
    #         frame_app_name = sys.argv[2]
    #     # 参数输入方式错误
    #     except:
    #         error_msg()
    # # 如果不为3，调用方法错误，输出提示
    # else:
    #     error_msg()
    port = 8080
    frame_app_name = "mini_frame:application"
    # 打开配置文件，获取静态、动态文件夹地址
    with open("web_server.conf") as f:
        conf_info = eval(f.read())
    dynamic_path = conf_info["dynamic_path"]
    sys.path.append(dynamic_path)
    # 获取框架名、方法名
    ret = re.match(r"([^:]+):(.*)", frame_app_name)
    if ret:
        frame_name = ret.group(1)
        app_name = ret.group(2)
        # print("app",app_name)
    else:
        error_msg()
    # 导入web_frame框架
    frame = __import__(frame_name)
    # 导入app方法
    app = getattr(frame, app_name)
    # print("创建WSGI对象")
    # 创建WSGI对象
    wsgi_server = WSGIServer(port, app, conf_info["static_path"])
    # 调用其run_forever方法，使流程不断运行
    wsgi_server.fun_forever()

if __name__ == "__main__":
    main()