Flask框架实现登录注册功能（mysql数据库）

前言：

本例使用Flask框架完成登录和注册操作，包括前端（index.html，regist.html）和后端（app.py）两部分，前端页面不过多介绍，直接进入后端部分：

1、逻辑思路：

登录部分：运行程序默认进入登录界面，输入账号和密码点击登录，系统进入”获取登录请求参数及处理“阶段（@app.route(‘/login’)），验证账号密码正确后跳转到目标界面。

注册部分：运行程序输入注册部分的路由进入注册界面，输入账号、密码和确认密码，系统进入”获取注册请求及处理“阶段（@app.route(‘/registuser’)），判断两次输入密码是否一致，一致则跳转到登录界面，否则弹出警告框提示用户重新输入。

2、主要步骤：

• 导入需要的扩展包

#导入数据库模块
import pymysql

from flask import Flask
from flask import render_template

#导入前台请求的request模块
from flask import request
import traceback
#导入弹出警告框模块
import win32api,win32con

• 定义登录界面路由

#默认登录页面
@app.route('/')
def login():
    return render_template('login.html')

• 定义注册界面路由

#访问注册页面
@app.route('/regist')
def regist():
    return render_template('regist.html')

• 获取登录请求参数及处理

@app.route('/login')
def getLoginRequest():
    # 查询用户名及密码是否匹配及存在
    # 连接数据库,此前在数据库中创建数据库TESTDB
    db = pymysql.connect(host="localhost", user="root", password="417020", database="flask",charset="utf8")
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    # SQL 查询语句
    sql = "select * from user where user=" + request.args.get('user') + " and password=" + request.args.get(
        'password') + ""
    try:
        # 执行sql语句
        cursor.execute(sql)
        results = cursor.fetchall()
        print(len(results))
        if len(results) == 1:
            return '登录成功'       #返回需要跳转的页面或需要显示的字符串
        else:
            return '用户名或密码不正确'
        # 提交到数据库执行
        db.commit()
    except:
        # 如果发生错误则回滚
        traceback.print_exc()
        db.rollback()
    # 关闭数据库连接
    db.close()

• 获取注册请求及处理

@app.route('/registuser')
def getRigistRequest():
#把用户名和密码注册到数据库中

    #连接数据库,此前在数据库中创建数据库flask
    db = pymysql.connect(host="localhost", user="root", password="417020", database="flask",charset="utf8")
    #使用cursor()方法获取操作游标
    cursor = db.cursor()
    #获取输入框内容
    username=request.args.get('user')
    password=request.args.get('password')
    password2=request.args.get('password2')
    print(password)
    print(password2)
    #判断两次输入密码是否一致，一致则跳转到登录界面，不一致则弹出警告，要求用户重新输入
    if password==password2:
        # SQL 插入语句
        sql = "INSERT INTO user(user, password) VALUES ("+username+", "+password+")"
        try:
            # 执行sql语句
            cursor.execute(sql)
            # 提交到数据库执行
            db.commit()
             #注册成功之后跳转到登录页面
            return render_template('login.html')
        except:
            #抛出错误信息
            traceback.print_exc()
            # 如果发生错误则回滚
            db.rollback()
            return '注册失败'
        # 关闭数据库连接
        db.close()
    else:
        win32api.MessageBox(0,"两次输入密码不一致，请重新输入！","提醒",win32con.MB_ICONWARNING)
        return render_template('regist.html')

3、完整代码：

#导入数据库模块
import pymysql

from flask import Flask
from flask import render_template

#导入前台请求的request模块
from flask import request
import traceback
#导入弹出警告框模块
import win32api,win32con

#创建Flask应用实例
app = Flask(__name__)


#默认登录页面
@app.route('/')
def login():
    return render_template('login.html')

#访问注册页面
@app.route('/regist')
def regist():
    return render_template('regist.html')

#获取注册请求及处理
@app.route('/registuser')
def getRigistRequest():
#把用户名和密码注册到数据库中

    #连接数据库,此前在数据库中创建数据库flask
    db = pymysql.connect(host="localhost", user="root", password="417020", database="flask",charset="utf8")
    #使用cursor()方法获取操作游标
    cursor = db.cursor()
    #获取输入框内容
    username=request.args.get('user')
    password=request.args.get('password')
    password2=request.args.get('password2')
    print(password)
    print(password2)
    #判断两次输入密码是否一致，一致则跳转到登录界面，不一致则弹出警告，要求用户重新输入
    if password==password2:
        # SQL 插入语句
        sql = "INSERT INTO user(user, password) VALUES ("+username+", "+password+")"
        try:
            # 执行sql语句
            cursor.execute(sql)
            # 提交到数据库执行
            db.commit()
             #注册成功之后跳转到登录页面
            return render_template('login.html')
        except:
            #抛出错误信息
            traceback.print_exc()
            # 如果发生错误则回滚
            db.rollback()
            return '注册失败'
        # 关闭数据库连接
        db.close()
    else:
        win32api.MessageBox(0,"两次输入密码不一致，请重新输入！","提醒",win32con.MB_ICONWARNING)
        return render_template('regist.html')

# 获取登录参数及处理
@app.route('/login')
def getLoginRequest():
    # 查询用户名及密码是否匹配及存在
    # 连接数据库,此前在数据库中创建数据库TESTDB
    db = pymysql.connect(host="localhost", user="root", password="417020", database="flask",charset="utf8")
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    # SQL 查询语句
    sql = "select * from user where user=" + request.args.get('user') + " and password=" + request.args.get(
        'password') + ""
    try:
        # 执行sql语句
        cursor.execute(sql)
        results = cursor.fetchall()
        print(len(results))
        if len(results) == 1:
            return '登录成功'       #返回需要跳转的页面或需要显示的字符串
        else:
            return '用户名或密码不正确'
        # 提交到数据库执行
        db.commit()
    except:
        # 如果发生错误则回滚
        traceback.print_exc()
        db.rollback()
    # 关闭数据库连接
    db.close()


# 使用__name__ == '__main__'是 Python 的惯用法，确保直接执行此脚本时才
# 启动服务器，若其他程序调用该脚本可能父级程序会启动不同的服务器
if __name__ == '__main__':
    app.run(debug=True)