Flask框架实现登录注册功能(mysql数据库)

前言:

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

逻辑思路:

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

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

主要步骤:

①导入需要的扩展包

#导入数据库模块
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')

完整代码:

#导入数据库模块
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)
原文地址:https://www.cnblogs.com/zyj3955/p/15504433.html