mysql实现注册登录代码

pip install Flask mysql-connector-python

from flask import Flask, request, jsonify
import mysql.connector

app = Flask(__name__)

# 连接到MySQL数据库
db = mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)

cursor = db.cursor()

@app.route('/register', methods=['POST'])
def register():
    data = request.get_json()
    username = data['username']
    password = data['password']

    # 检查用户名是否已存在
    cursor.execute("SELECT * FROM users WHERE username = %s", (username,))
    result = cursor.fetchone()
    if result:
        return jsonify({"message": "用户名已存在"}), 400

    # 插入新用户
    sql = "INSERT INTO users (username, password) VALUES (%s, %s)"
    cursor.execute(sql, (username, password))
    db.commit()

    return jsonify({"message": "注册成功"}), 201

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

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

    # 验证用户名和密码
    cursor.execute("SELECT * FROM users WHERE username = %s AND password = %s", (username, password))
    result = cursor.fetchone()
    if result:
        return jsonify({"message": "登录成功"}), 200
    else:
        return jsonify({"message": "用户名或密码错误"}), 401

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

cursor.execute("SELECT * FROM users WHERE username = %s", (username,))
result = cursor.fetchone()
if result:
    return jsonify({"message": "用户名已存在"}), 400

cursor.execute("SELECT * FROM users WHERE username = %s AND password = %s", (username, password))
result = cursor.fetchone()
if result:
    return jsonify({"message": "登录成功"}), 200
else:
    return jsonify({"message": "用户名或密码错误"}), 401

db = mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)