python使用Flask操作mysql实现登录功能

下面是使用Flask操作mysql实现登录功能的完整攻略。

1. 安装依赖

首先需要安装Flask和MySQL驱动。可以使用以下命令安装：

pip install flask pymysql

2. 创建数据库

在MySQL中创建一个名为user的数据库，然后在该数据库中创建一个名为user_info的表。

CREATE DATABASE user;
USE user;
CREATE TABLE user_info (
  id INT AUTO_INCREMENT PRIMARY KEY,
  username VARCHAR(20) NOT NULL UNIQUE,
  password VARCHAR(20) NOT NULL
);

3. 创建Flask应用程序

创建一个名为app.py的Python文件，然后编写Flask应用程序。以下是一个简单的示例：

from flask import Flask, request, session, redirect, url_for, render_template
import pymysql

app = Flask(__name__)
app.secret_key = 'your_secret_key'

# MySQL数据库配置
db = pymysql.connect(
  host='localhost',
  user='root',
  password='root',
  db='user',
  cursorclass=pymysql.cursors.DictCursor
)

@app.route('/')
def home():
  return render_template('home.html')

@app.route('/login', methods=['GET', 'POST'])
def login():
  if request.method == 'POST':
    # 获取表单数据
    username = request.form['username']
    password = request.form['password']

    # 查询数据库中是否存在该用户
    with db.cursor() as cursor:
      sql = "SELECT * FROM user_info WHERE username=%s AND password=%s"
      cursor.execute(sql, (username, password))
      user = cursor.fetchone()

    # 如果存在该用户，保存用户信息到session中并跳转到欢迎页面
    if user:
      session['username'] = user['username']
      return redirect(url_for('welcome'))
    else:
      # 用户没有被查询到，显示错误消息
      error = 'Invalid username or password!'
      return render_template('login.html', error=error)

  # GET请求返回登录页面
  return render_template('login.html')

@app.route('/welcome')
def welcome():
  # 检查是否有用户会话
  if 'username' in session:
    username = session['username']
    return f'Welcome, {username}!'
  else:
    return redirect(url_for('login'))

@app.route('/logout')
def logout():
  # 从会话中删除用户的登录信息
  session.pop('username', None)
  return redirect(url_for('home'))

if __name__ == '__main__':
  app.run(debug=True)

4. 创建模板文件

Flask在templates文件夹中查找HTML模板文件。需要创建三个模板文件：home.html、login.html、welcome.html。这里我们只提供一个简单的home.html示例：

<!DOCTYPE html>
<html>
<head>
  <title>Home</title>
</head>
<body>
  <h1>Welcome to my website!</h1>
  <a href="{{ url_for('login') }}">Login</a>
</body>
</html>

5. 运行程序

在终端中运行Flask应用程序：

python app.py

访问网站首页：http://localhost:5000/，点击登录链接进行登录。

以上为使用Flask操作MySQL实现登录功能的完整攻略。

示例二：

以下是第二个示例：使用SQLAlchemy作为ORM和Flask-Login作为身份验证库实现登录功能。

1. 安装依赖

需要安装以下依赖：

pip install flask flask-login sqlalchemy pymysql

2. 创建数据库

在MySQL中创建一个名为user的数据库，然后在该数据库中创建一个名为user_info的表。

CREATE DATABASE user;
USE user;
CREATE TABLE user_info (
  id INT AUTO_INCREMENT PRIMARY KEY,
  username VARCHAR(20) NOT NULL UNIQUE,
  password VARCHAR(20) NOT NULL
);

3. 创建Flask应用程序

创建一个名为app.py的Python文件，然后编写Flask应用程序。以下是一个简单的示例：

from flask import Flask, request, session, redirect, url_for, render_template
from flask_login import LoginManager, login_user, current_user, logout_user
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from werkzeug.security import generate_password_hash, check_password_hash

app = Flask(__name__)
app.secret_key = 'your_secret_key'

# MySQL数据库配置
engine = create_engine('mysql+pymysql://root:root@localhost/user')
Session = sessionmaker(bind=engine)
Base = declarative_base()

# Flask-Login身份验证
login_manager = LoginManager(app)
login_manager.login_view = 'login'

class User(Base):
  __tablename__ = 'user_info'

  id = Column(Integer, primary_key=True)
  username = Column(String(20), unique=True)
  password = Column(String(128))

  @property
  def is_authenticated(self):
    return True

  @property
  def is_active(self):
    return True

  @property
  def is_anonymous(self):
    return False

  def get_id(self):
    return str(self.id)

@login_manager.user_loader
def load_user(user_id):
  session = Session()
  return session.query(User).get(int(user_id))

@app.route('/')
def home():
  return render_template('home.html')

@app.route('/login', methods=['GET', 'POST'])
def login():
  if request.method == 'POST':
    # 获取表单数据
    username = request.form['username']
    password = request.form['password']

    # 查询数据库中是否存在该用户
    session = Session()
    user = session.query(User).filter_by(username=username).first()

    # 如果存在该用户，验证密码并保存用户信息到session中并跳转到欢迎页面
    if user and check_password_hash(user.password, password):
      login_user(user)
      return redirect(url_for('welcome'))
    else:
      # 用户没有被查询到或密码错误，显示错误消息
      error = 'Invalid username or password!'
      return render_template('login.html', error=error)

  # GET请求返回登录页面
  return render_template('login.html')

@app.route('/register', methods=['GET', 'POST'])
def register():
  if request.method == 'POST':
    # 获取表单数据
    username = request.form['username']
    password = request.form['password']

    # 检查用户名是否已被使用
    session = Session()
    user = session.query(User).filter_by(username=username).first()
    if user:
      error = 'This username is already taken!'
      return render_template('register.html', error=error)

    # 在数据库中创建新用户
    password_hash = generate_password_hash(password)
    new_user = User(username=username, password=password_hash)
    session.add(new_user)
    session.commit()

    # 登录新用户并跳转到欢迎页面
    login_user(new_user)
    return redirect(url_for('welcome'))

  # GET请求返回注册页面
  return render_template('register.html')

@app.route('/welcome')
def welcome():
  # 获取当前用户
  if current_user.is_authenticated:
    username = current_user.username
    return f'Welcome, {username}!'
  else:
    return redirect(url_for('login'))

@app.route('/logout')
def logout():
  logout_user()
  return redirect(url_for('home'))

if __name__ == '__main__':
  Base.metadata.create_all(engine)
  app.run(debug=True)

4. 创建模板文件

Flask在templates文件夹中查找HTML模板文件。需要创建五个模板文件：home.html、login.html、register.html、welcome.html、base.html。这里我们只提供一个简单的home.html示例：

{% extends 'base.html' %}

{% block content %}
  <h1>Welcome to my website!</h1>
  <a href="{{ url_for('login') }}">Login</a> or <a href="{{ url_for('register') }}">Register</a>
{% endblock %}

5. 运行程序

在终端中运行Flask应用程序：

python app.py

访问网站首页：http://localhost:5000/，点击登录链接进行登录，点击注册链接进行注册。

以上为使用SQLAlchemy作为ORM和Flask-Login作为身份验证库实现登录功能的完整攻略。

本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：python使用Flask操作mysql实现登录功能 - Python技术站

python使用Flask操作mysql实现登录功能

1. 安装依赖

2. 创建数据库

3. 创建Flask应用程序

4. 创建模板文件

5. 运行程序

1. 安装依赖

2. 创建数据库

3. 创建Flask应用程序

4. 创建模板文件

5. 运行程序

相关文章