前言:
本例使用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)