天天看点

用pymysql和Flask搭建后端,响应前端POST和GET请求定义app设置跨域连接数据库,账号是root,密码是000,数据库名称是shopdataselect a user,根据userid和password获取一个用户是否存在,即判断登录是否成功insert a user,根据userid、password等信息,生成一个元组,将其插入数据库shopdata的userinfo表on_register,提供给前端的注册接口on_login,提供给前端的登录接口

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