Python登录系统实现

在Web开发中，登录系统是一个非常重要的功能，它允许用户通过输入用户名和密码来访问受保护的资源，在Python中，我们可以使用多种方法来实现登录系统，例如使用Flask框架、Django框架等，本文将介绍如何使用Python的Flask框架实现一个简单的登录系统。

我们需要安装Flask框架，在命令行中输入以下命令来安装Flask：

pip install flask

接下来，我们创建一个简单的Flask应用，在项目目录下创建一个名为app.py的文件，并输入以下代码：

from flask import Flask, render_template, request, redirect, url_for, flash
from flask_sqlalchemy import SQLAlchemy
from werkzeug.security import generate_password_hash, check_password_hash

app = Flask(__name__)
app.config['SECRET_KEY'] = 'mysecretkey'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///login.db'
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, nullable=False)
    password = db.Column(db.String(120), nullable=False)

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

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        username = request.form['username']
        password = request.form['password']
        user = User.query.filter_by(username=username).first()
        if user and check_password_hash(user.password, password):
            flash('登录成功！', 'success')
            return redirect(url_for('index'))
        else:
            flash('用户名或密码错误，请重试。', 'danger')
    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']
        hashed_password = generate_password_hash(password, method='sha256')
        new_user = User(username=username, password=hashed_password)
        db.session.add(new_user)
        db.session.commit()
        flash('注册成功，请登录。', 'success')
        return redirect(url_for('login'))
    return render_template('register.html')

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

在这个代码中，我们首先导入了所需的库和模块，然后创建了一个Flask应用实例，我们定义了一个User模型，用于存储用户的用户名和密码，我们还定义了三个路由函数：index()、login()和register()，分别用于显示主页、处理登录请求和处理注册请求。

在login()函数中，我们首先检查请求方法是否为POST，如果是POST请求，我们从表单中获取用户名和密码，然后查询数据库中的用户，如果找到了匹配的用户并且密码正确，我们将显示一个成功的提示信息并重定向到主页，我们将显示一个错误提示信息。

在register()函数中，我们同样首先检查请求方法是否为POST，如果是POST请求，我们从表单中获取用户名和密码，然后使用generate_password_hash()函数对密码进行哈希处理，我们创建一个新的用户对象并将其添加到数据库中，我们显示一个成功的提示信息并重定向到登录页面。