mysql 保存用户名密码

import hashlib
import mysql.connector
import os

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

# 创建用户表
cursor.execute("CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, username VARCHAR(255), salt VARCHAR(255), password_hash VARCHAR(255))")

# 注册新用户
def register_user(username, password):
    salt = os.urandom(16).hex()
    password_hash = hashlib.sha256((salt + password).encode()).hexdigest()
    cursor.execute("INSERT INTO users (username, salt, password_hash) VALUES (%s, %s, %s)", (username, salt, password_hash))
    db.commit()

# 用户登录
def login_user(username, password):
    cursor.execute("SELECT salt, password_hash FROM users WHERE username = %s", (username,))
    result = cursor.fetchone()
    if result:
        salt, stored_password_hash = result
        password_hash = hashlib.sha256((salt + password).encode()).hexdigest()
        if password_hash == stored_password_hash:
            return True
    return False

# 示例：注册新用户
register_user("testuser", "testpassword")

# 示例：用户登录
if login_user("testuser", "testpassword"):
    print("登录成功")
else:
    print("登录失败")

# 关闭数据库连接
cursor.close()
db.close()