用pymysql和Flask搭建后端，响应前端POST和GET请求，实现登录和注册功能
前言
这次作业不仅需要我建立一个数据库(详情请点击这里)，还需要我基于这个数据库写后端接口（注册和登录）供前端访问，接收前端的POST和GET请求，并将登录、注册是否成功传给前端。
本文介绍如何用Flask搭建后端，其中使用了pymysql操作mysql数据库，也会做这个部分的介绍。
正文
需要为前端提供的接口有两个：注册和登录，为此我定义了四个函数，分别是
- select_user(userid, password)
- insert_user(userid, password, phone, email, company)
- on_register()
- on_login()
前两个函数是操作数据库，被后两个函数调用；后两个函数是给前端的接口。
后端说明
整个后端的代码如下：
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
data = request.args.to_dict() # 需from flask import request这样拿到的data是一个python的字典对象
-
POST
data = request.get_data() # 需from flask import request data = json.loads(data) # 转化为字典对象。需import json这两行代码才拿到一个python的字典对象
接口获取前端传过来的参数的方式与其响应的请求类型（如POST、GET、POST和GET）是对应的。
对应地，前端发过来的参数类型也需要和后端进行匹配。
前端代码
这里给出前端处理用户登录和注册请求的函数。
-
login
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
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字符串
JSON.stringify({ userid: userid, password: password1, phone: phone, email: email, company: company }) -
register是POST请求，发送了javascript的类的对象
{ userid: userid.value, password: password.value, }
至此，本文对这次作业中后端搭建的介绍就结束了。
这次作业算是我第一次写后端，也是我第一次接触Flask，如果我有写错的地方，请在评论区指正！
作者：@臭咸鱼
转载请注明出处：https://www.cnblogs.com/chouxianyu/
欢迎讨论和交流!
