用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/