用pymysql和Flask搭建后端，响应前端POST和GET请求，实现登录和注册功能

Posted on | | Visitors:
Words count in article: 1,672 字 | Reading time ≈ 8 分

前言

这次作业不仅需要我建立一个数据库(详情请点击这里)，还需要我基于这个数据库写后端接口（注册和登录）供前端访问，接收前端的POST和GET请求，并将登录、注册是否成功传给前端。

本文介绍如何用Flask搭建后端，其中使用了pymysql操作mysql数据库，也会做这个部分的介绍。

正文

需要为前端提供的接口有两个：注册和登录，为此我定义了四个函数，分别是

  • select_user(userid, password)
  • insert_user(userid, password, phone, email, company)
  • on_register()
  • on_login()

前两个函数是操作数据库，被后两个函数调用；后两个函数是给前端的接口。

后端说明

整个后端的代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
from flask import Flask, request
import json
import pymysql
from flask_cors import CORS


# 定义app
app = Flask(__name__)
# 设置跨域
CORS(app, supports_credentials=True)

# 连接数据库，账号是root，密码是000，数据库名称是shopdata
db = pymysql.connect("localhost", "root", "000", "shopdata")  # 连接数据库
cursor = db.cursor()  # 定义游标


# select a user，根据userid和password获取一个用户是否存在，即判断登录是否成功
def select_user(userid, password):
    # mysql语句
    select_user_sql = 'select * from userinfo where userid="%s" and password="%s";' % (userid, password)
    # 执行mysql语句
    result = cursor.execute(select_user_sql)
    db.commit()
    # 如果返回了一条数据，则登录成功，否则登录失败
    if 1 == result:
        result = True
    else:
        result = False
        print('there is no user where userid="%s and password="%s"!!' % (userid, password))
    return result


# insert a user，根据userid、password等信息，生成一个元组，将其插入数据库shopdata的userinfo表
def insert_user(userid, password, phone, email, company):
    # mysql语句
    insert_user_sql = 'insert into userinfo(userid, password, phone, email, company)' \
                      'values("%s", "%s", "%s", "%s", "%s");' % (userid, password, phone, email, company)
    # 执行mysql语句，如果插入成功，则注册成功，否则注册失败
    try:
        cursor.execute(insert_user_sql)
        db.commit()
        print('insert user where userid="%s" and password="%s"!!' % (userid, password))
        result = True
    except:
        print('can not insert user where userid="%s" and password="%s"!!' % (userid, password))
        result = False
    finally:
        return result


# on_register，提供给前端的注册接口
@app.route("/api/register", methods=['POST'])  # 路由，响应POST请求
def on_register():
    # 默认的用户名和密码（该用户不存在且非法）
    userid = ''
    password = ''
    phone = ''
    email = ''
    company = ''
    # 判断传入的参数是否为空，并取出前端传来的参数
    data = request.get_data()
    # print(data)
    if data is not None:
        # 将bytes类型转化为字典。对应的，前端发过来的内容应该是JSON.stringify(一个对象)
        data = json.loads(data)  # 转化为字典对象
        # print(data)
        userid = data.get('userid')
        password = data.get('password')
        phone = data.get('phone')
        email = data.get('email')
        company = data.get('company')
    # 判断参数是否非法。若非法则直接判断出注册失败，若合法则尝试注册，再根据数据库操作结果判断。
    if len(userid) > 0 and len(password) > 0 and len(phone) > 0 and len(email) > 0 and len(password) > 0:
        return_dict = {'success': insert_user(userid=userid, password=password, phone=phone,
                                              email=email, company=company)}
    else:
        return_dict = {'success': False}
    # 返回结果（结果暂时简单点，只返回成功或失败）
    return json.dumps(return_dict)  # 字典转json


# on_login，提供给前端的登录接口
@app.route("/api/login", methods=['GET'])  # 路由，响应GET请求
def on_login():
    # 默认的用户名和密码（数据库中不存在该用户）
    userid = ''
    password = ''
    # 判断传入的参数是否为空，获取前端传来的参数
    if request.args is not None:
        print(request.args)
        data = request.args.to_dict()
        userid = data.get('userid')
        password = data.get('password')
    # 查询数据库
    result = select_user(userid=userid, password=password)
    if not result:
        print('user where userid="%s and password="%s" login!!' % (userid, password))
    # 返回登录结果（暂时简单一点，成功或失败）
    return_dict = {'success': result}
    return json.dumps(return_dict)  # 字典转json


if __name__ == '__main__':
    # 运行app
    app.run()

    # 程序结束时释放数据库资源
    cursor.close()
    db.close()  # 关闭连接

值得注意的是，这里的两个接口分别响应POST请求和GET请求。

可以发现，两个接口获取前端传过来的参数的方式是不同的：

  • GET

    1
    data = request.args.to_dict()  # 需from flask import request

    这样拿到的data是一个python的字典对象

  • POST

    1
    2
    data = request.get_data()  # 需from flask import request
    data = json.loads(data)  # 转化为字典对象。需import json

    这两行代码才拿到一个python的字典对象

接口获取前端传过来的参数的方式与其响应的请求类型（如POST、GET、POST和GET）是对应的。

对应地，前端发过来的参数类型也需要和后端进行匹配。

前端代码

这里给出前端处理用户登录和注册请求的函数。

  • login

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    function login() {
        var url = "http://127.0.0.1:5000/api/login";
        var userid = document.getElementById("userid");
        var password = document.getElementById("password");
        $.ajax({
            url: url,
            type: "GET",
            data: {
                userid: userid.value,
                password: password.value,
            },
            success: function(data) {
                var data1 = JSON.parse(data);
                // console.log(data1);
                if (data1.success) {
                    window.sessionStorage.setItem("userid", userid.value);
                    alert("登录成功！进入主页面！");
                    window.location.href = 'index.html';
                } else {
                    alert("登录失败！请输入正确的账号和密码！");
                }
            },
            error: function(XMLHttpRequest, textStatus, errorThrown) {
                if (textStatus === 'timeout') {
                    alert('请求超时，请重试！');
                }
            }

        })
    }

  • register

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    function register() {
        var password1 = document.getElementById("password1").value;
        var password2 = document.getElementById("password2").value;
        var userid = document.getElementById("userid").value;
        var phone = document.getElementById("phone").value;
        var email = document.getElementById("email").value;
        var company = document.getElementById("company").value;
        if (password1 !== password2) {
            alert("两次输入的密码不相同,请重新输入！");
            return;
        }
        $.ajax({
            url: "http://127.0.0.1:5000/api/register",
            type: "POST",
            // 转成json
            data: JSON.stringify({
                userid: userid,
                password: password1,
                phone: phone,
                email: email,
                company: company
            }),
            success: function(data) {
                var data1 = JSON.parse(data);
                if (data1.success) {
                    alert("注册成功，请登录！");
                    window.location.href = 'login.html';
                } else {
                    alert("注册失败，请检查您输入的信息是否正确！");
                }
            },
            error: function(XMLHttpRequest, textStatus, errorThrown) {
                if (textStatus === 'timeout') {
                    alert("请求超时！");
                }
            }
        })
    }

可以注意到，用ajax发送GET请求和PSOT请求时，我发的数据类型是不一样的。

  • login是GET请求，发送了json字符串

    1
    2
    3
    4
    5
    6
    7
    JSON.stringify({
        userid: userid,
        password: password1,
        phone: phone,
        email: email,
        company: company
    })

  • register是POST请求，发送了javascript的类的对象

    1
    2
    3
    4
    {
        userid: userid.value,
        password: password.value,
    }

至此，本文对这次作业中后端搭建的介绍就结束了。

这次作业算是我第一次写后端，也是我第一次接触Flask，如果我有写错的地方，请在评论区指正！

作者：@臭咸鱼

转载请注明出处：https://www.cnblogs.com/chouxianyu/

欢迎讨论和交流!