用pymysql和Flask搭建后端，响应前端POST和GET请求
- 2020 年 4 月 5 日
- 笔记
前言
这次作业不仅需要我建立一个数据库(详情请点击这里)，还需要我基于这个数据库写后端接口（注册和登录）供前端访问，接收前端的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/
欢迎讨论和交流!
