统计Flask中的登录尝试次数

from flask import Flask, request
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'your_database_uri'
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(80), nullable=False)
    login_attempts = db.Column(db.Integer, default=0)

    def __init__(self, username, password):
        self.username = username
        self.password = password

@app.route('/login', methods=['POST'])
def login():
    username = request.form['username']
    password = request.form['password']

    user = User.query.filter_by(username=username).first()

    if user:
        if user.password == password:
            user.login_attempts = 0
            db.session.commit()
            return 'Login successful'
        else:
            user.login_attempts += 1
            db.session.commit()
            if user.login_attempts >= 3:
                return 'Too many login attempts. Your account has been locked.'
            else:
                return 'Invalid password'
    else:
        return 'User not found'

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