02 自定义web框架

自定义web框架

web框架
main.py: 启动文件,封装了socket
1 urls.py: 路径与视图函数映射关系  ---- url控制器
2 views.py 视图函数,固定有一个形式参数:environ -----视图函数,
3 templates文件夹: html文件   -----模板
4 models: 在项目启动前,在数据库中创建表结构    ----- 与数据库相关

如图所示：

 

main.py

# -*- coding: utf-8 -*-
# @Time    : 2019/7/9 20:05
# @Author  : Xiao

from wsgiref.simple_server import make_server
from urls import url_datas

def application(environ,start_response):
    start_response('200 OK',[('Content-Type','text/html'),('Charset','utf8')])
    print("PATH", environ.get("PATH_INFO"))
    # 当前请求路径
    path = environ.get("PATH_INFO")
    func = None
    for item in url_datas:
        if path == item[0]:
            func = item[1]
            break
    if func:
        return [func(environ)]
    return [b'404!']

server = make_server('',8080,application)

# 开始监听HTTP请求:
print('starting....')
server.serve_forever()

 

models.py

# -*- coding: utf-8 -*-
# @Time    : 2019/7/16 10:32
# @Author  : Xiao


from tools import Mysql

sql_check_db = "show databases like 'oldboy';"
sql_create_db = 'create database oldboy DEFAULT CHARACTER SET utf8 COLLATE utf8_general_ci;'
sql_check_table = 'show tables;'
sql_create_table ='create table user_info(id int(10) PRIMARY KEY auto_increment,name VARCHAR(20),password VARCHAR(20),age int(10))'
sql_check_data = 'select count(*) from oldboy'
sql_create_data = "INSERT into user_info(name,password,age) values ('xg','a123456',18),('xk','a123456',38),('xh','a123456',28)"


mysql_check = Mysql('localhost', 3306, 'root', 'root', charset="utf8")

if mysql_check.exec_sql(sql_check_db):
    mysql_create = Mysql('localhost', 3306, 'root', 'root',dbname='oldboy', charset="utf8")
    if mysql_create.exec_sql(sql_check_table):
        if not mysql_create.exec_sql(sql_check_data):
            mysql_create.exec_sql(sql_create_data)
    else:
        mysql_create.exec_sql(sql_create_table)
        mysql_create.exec_sql(sql_create_data)
else:
    mysql_check.exec_sql(sql_create_db)
    mysql_create = Mysql('localhost', 3306, 'root', 'root', dbname='oldboy', charset="utf8")
    mysql_create.exec_sql(sql_create_table)
    mysql_create.exec_sql(sql_create_data)

 

tools.py

# -*- coding: utf-8 -*-
# @Time    : 2019/7/16 10:42
# @Author  : Xiao

import pymysql


class Mysql(object):
    def __init__(self, host, port, user, pwd, dbname="", charset="utf8"):
        """初始化数据库信息以及连接"""
        self.host = host
        self.port = port
        self.user = user
        self.pwd = pwd
        self.dbname = dbname
        self.charset = charset
        self.conn = None  # 给析构函数用的，如果存在链接，则需要关闭链接，没有就不用管
        self.conn = self.get_conn
        self.cur = None
        self.cur = self.get_cur

    @property
    def get_conn(self):
        """根据传参判断是否有传数据库名，然后做不同的链接操作"""
        try:
            if self.dbname:
                conn = pymysql.connect(host=self.host, port=self.port, user=self.user,
                                            password=self.pwd, database=self.dbname, charset=self.charset)
                return conn
            else:
                conn = pymysql.connect(host=self.host, port=self.port, user=self.user,
                                            password=self.pwd, charset=self.charset)
                return conn
        except Exception as e:
            print(e)

    @property
    def get_cur(self):
        """获取游标"""
        if self.conn:
            return self.conn.cursor()

    def exec_sql(self, sql, *args):
        """
        执行sql，根据不同的sql语法做不同的操作：
        查询就返回查询结果列表，其他成功就返回受影响的行数整型，报错返回报错信息字符串
        *args是为了防止sql注入，自己拼接的sql语法字符串可能被sql注入,
        这里的用法是sql不需要自己拼接，直接将变量按顺序传进来，pymysql自动拼接，而且避免了sql注入的问题
        如：
        sql = "insert into test.student VALUES (19,%s,'English',%s);"
        res = test_mysql.exec_sql(sql,"yangxga",100)
        """
        try:
            rows = self.cur.execute(sql, args)  # rows记录受影响的行
            if sql.strip().lower().startswith('select') or sql.strip().lower().startswith('show'):
                res = self.cur.fetchall()
                res = self.format_res(res)
                print(res)
                return res
            else:
                self.conn.commit()  # 非查询语句需要提交才能生效
                res = rows
                print(res)
                return res
        except Exception as e:
            print(e)
            return e

    @staticmethod
    def format_res(res):
        """格式化数据库查找到的结果，如果"""
        res_lis = []
        if res:
            for i in res:
                res_lis.append(i)
        return res_lis

    def __del__(self):
        """析构函数，关闭鼠标，断开连接"""
        if self.cur:
            self.cur.close()
        if self.conn:
            self.conn.close()

 

urls.py

# -*- coding: utf-8 -*-
# @Time    : 2019/7/9 20:05
# @Author  : Xiao

from views import *

url_datas = [
    ('/favicon.ico',favicon),
    ('/index',index),
    ('/login',login),
    ('/regist',regist)
]

 

views.py

# -*- coding: utf-8 -*-
# @Time    : 2019/7/9 20:17
# @Author  : Xiao


from tools import Mysql
from urllib.parse import parse_qs

def op_file(filename):
    with open(filename,'rb') as f:
        data = f.read()
    return data

def favicon(environ):
    data = op_file('./templates/favicon.ico')
    return data

def index(environ):
    data = op_file('./templates/index.html')
    return data

def regist(environ):
    method = environ.get('REQUEST_METHOD').lower()
    if method == 'post':
        try:
            request_body_size = int(environ.get('CONTENT_LENGTH',0))
        except (ValueError):
            request_body_size = 0
        request_body = environ['wsgi.input'].read(request_body_size)
        data = parse_qs(request_body)
        user = data.get(b'user')[0].decode('utf8')
        pwd = data.get(b'pwd')[0].decode('utf8')
        if user.strip() and pwd.strip():
            mysql_check = Mysql('localhost', 3306, 'root', 'root',dbname='oldboy', charset="utf8")
            sql = "select * from user_info WHERE  name=%s"
            res = mysql_check.exec_sql(sql, user, pwd)
            if res and isinstance(res, list):
                data = b'username is already exits!'
            else:
                data = op_file('./templates/index.html')
                sql_reg = "insert into user_info(name,password) values(%s,%s);"
                mysql_check.exec_sql(sql_reg,user,pwd)
        else:
            data = b'username and password can not be empty!'
    else:
        data = op_file('./templates/register.html')
    return data

def login(environ):
    method = environ.get('REQUEST_METHOD').lower()
    if method == 'post':
        try:
            request_body_size = int(environ.get('CONTENT_LENGTH', 0))
        except (ValueError):
            request_body_size = 0
        request_body = environ['wsgi.input'].read(request_body_size)
        data = parse_qs(request_body)
        user = data.get(b'user')[0].decode('utf8')
        pwd = data.get(b'pwd')[0].decode('utf8')
        mysql_check = Mysql('localhost', 3306, 'root', 'root', dbname='oldboy', charset="utf8")
        sql = "select * from user_info WHERE  name=%s and password=%s"
        res = mysql_check.exec_sql(sql, user, pwd)
        if res and isinstance(res,list):
            data = op_file('./templates/index.html')
        else:
            data = b'username or password is wrong!'
    else:
        data = op_file('./templates/login.html')
    return data

 

index.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>首页</title>
    <style type="text/css">
        h3{
            margin: 10px 12px;
            float: left;
            overflow: hidden;
        }
        button{
            margin: 12px;
            overflow: hidden;
        }
    </style>
</head>
<body>
<h3>欢迎进入首页</h3>
<button id="btn_reg">注册</button>
<button id="btn_login">登陆</button>
<form action="">
    <img src="http://pic39.nipic.com/20140321/18063302_210604412116_2.jpg">
</form>
</body>
<script type="text/javascript">
    window.onload = function () {
        var tmp_reg = document.getElementById('btn_reg');
        var tmp_login = document.getElementById('btn_login');
        tmp_reg.onclick = function () {
            window.open('http://127.0.0.1:8080/regist',target='_self')
        }
        tmp_login.onclick = function () {
            window.open('http://127.0.0.1:8080/login',target='_self')
        }
    }
</script>
</html>

 

login.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>登录</title>
</head>
<body>
<h3>登录</h3>
<form action="http://127.0.0.1:8080/login" method="post">
    <label for="user">用户名：</label>
    <input type="text" name="user" id="user">
    <label for="pwd">密码：</label>
    <input type="password" name="pwd" id="pwd">
    <input type="submit" value="登录">
</form>
</body>
</html>

 

register.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>注册</title>
</head>
<body>
<h3>注册</h3>
<form action="http://127.0.0.1:8080/regist" method="post">
    <label for="user">用户名：</label>
    <input type="text" name="user" id="user">
    <label for="pwd">密码：</label>
    <input type="password" name="pwd" id="pwd">
    <input type="submit" value="注册">
</form>
</body>
</html>