WSGI服务器

Posted 2021-02-10 sure-feng

  1 import re
  2 from pymysql import connect
  3 import urllib.parse
  4 import logging
  5 
  6 
  7 URL_FUNC_DICT = dict()
  8 
  9 def open_mysql():
 10     # 创建connect连接
 11     conn = connect(host=‘localhost‘,port=3306,user=‘root‘,password=‘m‘,database=‘stock‘,charset=‘utf8‘)
 12     # 获得cursor对象
 13     cs = conn.cursor()
 14     return conn, cs
 15 
 16 def close_mysql(conn, cs):
 17     # 关闭连接、对象
 18     cs.close()
 19     conn.close()
 20 
 21 def route(path_info):
 22     ‘‘‘装饰器：添加方法-路径到字典‘‘‘
 23     def set_func(func):
 24         URL_FUNC_DICT[path_info] = func
 25         def call_func(*args, **kwargs):
 26             return func(*args, **kwargs)
 27         return call_func
 28     return set_func
 29 
 30 @route(r"/index.html")
 31 def index(ret):
 32     # 打开、读取网页
 33     with open("./templates/index.html", encoding="utf-8") as f:
 34         content = f.read()
 35     # 打开数据库
 36     conn, cs = open_mysql()
 37     # 执行sql语句
 38     cs.execute("select * from info;")
 39     # 读取数据库数据
 40     stock_infos = cs.fetchall()
 41     # 关闭数据库
 42     close_mysql(conn, cs)
 43     # 数据库数据替换模板数据
 44     tr_template = """<tr>
 45         <td>%s</td>
 46         <td>%s</td>
 47         <td>%s</td>
 48         <td>%s</td>
 49         <td>%s</td>
 50         <td>%s</td>
 51         <td>%s</td>
 52         <td>%s</td>
 53         <td>
 54             <input type="button" value="添加" id="toAdd" name="toAdd" systemidvaule="%s">
 55         </td>
 56         </tr>
 57     """
 58     html = ""
 59     for line_info in stock_infos:
 60         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])
 61     return re.sub(r"{%content%}", html, content)
 62 
 63 @route(r"/center.html")
 64 def center(ret):
 65     # 打开读取网页
 66     with open("./templates/center.html", encoding="utf-8") as f:
 67         content = f.read()
 68     # 打开数据库
 69     conn, cs = open_mysql()
 70     # 执行sql语句
 71     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;")
 72     # 读取sql语句结果
 73     stock_infos = cs.fetchall()
 74     # 关闭数据库
 75     close_mysql(conn, cs)
 76     tr_template = """
 77         <tr>
 78             <td>%s</td>
 79             <td>%s</td>
 80             <td>%s</td>
 81             <td>%s</td>
 82             <td>%s</td>
 83             <td>%s</td>
 84             <td>%s</td>
 85             <td>
 86                 <a type="button" class="btn btn-default btn-xs" href="/update/%s.html"> <span class="glyphicon glyphicon-star" aria-hidden="true"></span> 修改 </a>
 87             </td>
 88             <td>
 89                 <input type="button" value="删除" id="toDel" name="toDel" systemidvaule="%s">
 90             </td>
 91         </tr>
 92     """
 93     # 用数据库内容替换模板数据
 94     html = ""
 95     for stock_info in stock_infos:
 96         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])
 97     return re.sub(r"{%content%}", html, content)
 98 
 99 @route(r"^/add/(d+).html$")
100 def add_focus(ret):
101     ‘‘‘添加股票‘‘‘
102     # 获取股票代码
103     stock_code = ret.group(1)
104     # 打开数据库，查询股票代码
105     conn, cs = open_mysql()
106     sql = "select * from info where code = %s;"
107     cs.execute(sql, (stock_code,))
108     # 股票代码不在数据库中
109     if not cs.fetchall():
110         # 返回提示
111         close_mysql(conn, cs)
112         return "没有这只股票代码，请收下留情。。"
113     # 判断股票代码是否已存在于focus表数据库中
114     sql = "select * from info as i inner join focus as f on i.id = f.info_id where i.code = %s"
115     cs.execute(sql, (stock_code,))
116     # 已存在于focus表
117     if cs.fetchall():
118         close_mysql(conn, cs)
119         return "已成功关注，请勿重复关注。。"
120     # 不存在于focu表
121     # 执行sql语句添加股票代码到focus表
122     sql = "insert into focus(info_id) select id from info where code = %s;"
123     cs.execute(sql, (stock_code))
124     conn.commit()
125     # 关闭数据库
126     close_mysql(conn, cs)
127     return "关注成功。。"
128 
129 @route(r"^/del/(d+).html$")
130 def del_focus(ret):
131     ‘‘‘取消关注股票‘‘‘
132     stock_code = ret.group(1)
133     print(stock_code)
134     conn, cs = open_mysql()
135     sql = "select * from focus as f inner join info as i on i.id = f.info_id where i.code = %s;"
136     cs.execute(sql, (stock_code,))
137     # 股票代码不存在于focus表
138     if not cs.fetchall():
139         close_mysql(conn, cs)
140         return "未曾关注，谈何取消。。"
141     sql = "delete from focus where info_id = (select id from info where code = %s);"
142     cs.execute(sql, (stock_code,))
143     conn.commit()
144     close_mysql(conn, cs)
145     return "已成功取关注。。"
146 
147 @route(r"/update/(d+).html")
148 def update_info(ret):
149     ‘‘‘更新数据‘‘‘
150     # 获取股票代码
151     stock_code = ret.group(1)
152     # 打开数据库，查询股票代码
153     conn, cs = open_mysql()
154     sql = "select * from info where code = %s;"
155     cs.execute(sql, (stock_code,))
156     # 股票代码不在数据库中
157     if not cs.fetchall():
158         # 返回提示
159         close_mysql(conn, cs)
160         return "没有这只股票代码，请收下留情。。"
161     # 判断股票代码是否已存在于focus表数据库中
162     sql = "select * from info as i inner join focus as f on i.id = f.info_id where i.code = %s"
163     cs.execute(sql, (stock_code,))
164     # 已存在于focus表
165     if not cs.fetchall():
166         close_mysql(conn, cs)
167         return "没有关注，请勿输入备注。。"
168     # 打开网页
169     with open("./templates/update.html", encoding="utf-8") as f:
170         content = f.read()
171     # 替换模板,回传数据
172     content = re.sub(r"{%note_info%}","请输入修改内容", content)
173     return re.sub(r"{%code%}", stock_code, content)
174     # return "更新数据成功。。"
175 
176 @route(r"/update/(d+)/(.*).html")
177 def save_update(ret):
178     stock_code = ret.group(1)
179     new_info = ret.group(2)
180     new_info = urllib.parse.unquote(new_info)
181     # 打开数据库，查询股票代码
182     conn, cs = open_mysql()
183     sql = "select * from info where code = %s;"
184     cs.execute(sql, (stock_code,))
185     # 股票代码不在数据库中
186     if not cs.fetchall():
187         # 返回提示
188         close_mysql(conn, cs)
189         return "没有这只股票代码，请收下留情。。"
190     # 判断股票代码是否已存在于focus表数据库中
191     sql = "select * from info as i inner join focus as f on i.id = f.info_id where i.code = %s"
192     cs.execute(sql, (stock_code,))
193     # 不存在于focu表
194     if not cs.fetchall():
195         close_mysql(conn, cs)
196         return "未曾关注，请勿修改。。"
197     # 已存在于focus表
198     # 执行sql语句添加股票代码到focus表
199     sql = "update focus set note_info=%s where info_id = (select id from info where code=%s);"
200     cs.execute(sql, (new_info, stock_code))
201     conn.commit()
202     # 关闭数据库
203     close_mysql(conn, cs)
204     return "修改成功。。"
205 
206 def application(env, start_header):
207     start_header("200 OK", [("Content-type", "text/html;charset=utf-8")])
208     # 获取请求文件信息
209     file_name = env["PATH_INFO"]
210     file_name = urllib.parse.unquote(file_name)
211     # 日志设定
212     # 创建logger
213     logger = logging.getLogger()
214     logger.setLevel(logging.DEBUG)
215     # 创建一个handler，用于写入文件
216     logfile = "./log.txt"
217     fh = logging.FileHandler(logfile,mode=‘a‘)
218     fh.setLevel(logging.INFO)
219     # 创建一个handler，用于输出控制台
220     ch = logging.StreamHandler()
221     ch.setLevel(logging.WARNING)
222     # 定义日志输出格式
223     formatter = logging.Formatter("%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s")
224     fh.setFormatter(formatter)
225     ch.setFormatter(formatter)
226     # 将logger添加到handler中
227     logger.addHandler(fh)
228     logger.addHandler(ch)
229     logger.info("客户访问：%s" % file_name)
230     try:
231         for url, func in URL_FUNC_DICT.items():
232             ret = re.match(url, file_name)
233             if ret:
234                 return func(ret)
235         else:
236             logger.warning("请求错误。file(%s) not found" % file_name)
237             return "请求错误。file(%s) not found" % file_name
238     except Exception as e:
239         logger.warning("程序异常。错误信息：%s" % e)
240         return "请求错误。logger错误信息：%s" % e

  1 import sys
  2 import re
  3 import socket
  4 import multiprocessing
  5 
  6 class WSGIServer(object):
  7     ‘‘‘服务器运行对象‘‘‘
  8     def __init__(self, port, app, static_path):
  9         ‘‘‘对象初始化‘‘‘
 10         # 创建套接字
 11         self.tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 12         self.tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
 13         # 绑定端口 IP
 14         self.tcp_server_socket.bind(("", port))
 15         # 设置监听套接字
 16         self.tcp_server_socket.listen(128)
 17         # web框架调用方法
 18         self.application = app
 19         # 静态网页存放地址
 20         self.static_path = static_path
 21 
 22     def client_serve(self, client_socket):
 23         ‘‘‘为浏览器实现具体服务‘‘‘
 24         # 获取浏览器请求
 25         request = client_socket.recv(1024).decode("utf-8")
 26         request_lines = request.splitlines()
 27         # 解析浏览器请求
 28         ret = re.match(r"[^/]+(/[^ ]*)", request_lines[0])
 29         file_name = ""
 30         if ret:
 31             file_name = ret.group(1)
 32             if file_name == "/":
 33                 file_name = "/index.html"
 34         # print("file_nema:", file_name)
 35         # 如果为静态页面请求（非html请求）
 36         if not file_name.endswith(".html"):
 37             # 打开请求页面
 38             try:
 39                 with open(self.static_path + file_name, "rb") as f:
 40                     response_body = f.read()
 41             # 打开页面失败
 42             except:
 43                 response_body = "file not found"
 44                 response_header = "HTTP/1.1 404 FILE NOT FOUND
"
 45                 response_header += "
"
 46                 response = response_header + response_body
 47                 client_socket.send(response.encode("utf-8"))
 48             # 打开页面成功
 49             else:
 50                 response_header = "HTTP/1.1 200 OK
"
 51                 response_header += "
"
 52                 response = response_header.encode("utf-8") + response_body
 53                 client_socket.send(response)
 54         # 如果为动态页面请求
 55         else:
 56             env = dict()
 57             env["PATH_INFO"] = file_name
 58             # print(file_name)
 59             # 调用web框架方法，处理页面
 60             response_body = self.application(env, self.set_headers)
 61 
 62             response_header = "HTTP/1.1 %s 
" % self.status
 63             for header in self.header_list:
 64                 response_header += "%s:%s
" %(header[0],header[1])
 65             response_header += "
"
 66 
 67             response = response_header + response_body
 68             client_socket.send(response.encode("utf-8"))
 69         # 关闭套接字
 70         client_socket.close()
 71 
 72     def set_headers(self, status, header_list):
 73         ‘‘‘存放web框架请求头信息‘‘‘
 74         self.status = status
 75         self.header_list = header_list
 76 
 77     def fun_forever(self):
 78         ‘‘‘服务器对象流程控制‘‘‘
 79         while True:
 80             # print("等待浏览器连接")
 81             client_socket, client_addr = self.tcp_server_socket.accept()
 82             # 利用多进程实现多任务操作
 83             # print("创建子线程")
 84             p = multiprocessing.Process(target=self.client_serve, args=(client_socket,))
 85             p.start()
 86             client_socket.close()
 87 
 88 def error_msg():
 89     print("服务器开启失败，请按以下方式输入")
 90     print("python xxx.py 8080 mini_frame:application")
 91 
 92 def main():
 93     ‘‘‘实现主体流程控制‘‘‘
 94     # # 获取服务器端口、web_frame框架地址、方法名
 95     # # 如果程序调用参数为3，输入正确
 96     # if len(sys.argv) == 3:
 97     #     # 参数输入方式正确
 98     #     try:
 99     #         port = int(sys.argv[1])
100     #         frame_app_name = sys.argv[2]
101     #     # 参数输入方式错误
102     #     except:
103     #         error_msg()
104     # # 如果不为3，调用方法错误，输出提示
105     # else:
106     #     error_msg()
107     port = 8080
108     frame_app_name = "mini_frame:application"
109     # 打开配置文件，获取静态、动态文件夹地址
110     with open("web_server.conf") as f:
111         conf_info = eval(f.read())
112     dynamic_path = conf_info["dynamic_path"]
113     sys.path.append(dynamic_path)
114     # 获取框架名、方法名
115     ret = re.match(r"([^:]+):(.*)", frame_app_name)
116     if ret:
117         frame_name = ret.group(1)
118         app_name = ret.group(2)
119         # print("app",app_name)
120     else:
121         error_msg()
122     # 导入web_frame框架
123     frame = __import__(frame_name)
124     # 导入app方法
125     app = getattr(frame, app_name)
126     # print("创建WSGI对象")
127     # 创建WSGI对象
128     wsgi_server = WSGIServer(port, app, conf_info["static_path"])
129     # 调用其run_forever方法，使流程不断运行
130     wsgi_server.fun_forever()
131 
132 if __name__ == "__main__":
133     main()