利用mysql实现登录

CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    username VARCHAR(50) NOT NULL UNIQUE,
    password VARCHAR(255) NOT NULL
);

INSERT INTO users (username, password) VALUES
('user1', SHA2('password1', 256)),
('user2', SHA2('password2', 256));

import mysql.connector
from mysql.connector import Error

def login(username, password):
    try:
        connection = mysql.connector.connect(host='localhost',
                                             database='your_database',
                                             user='your_username',
                                             password='your_password')
        cursor = connection.cursor()
        
        query = "SELECT * FROM users WHERE username = %s AND password = SHA2(%s, 256)"
        cursor.execute(query, (username, password))
        
        result = cursor.fetchone()
        
        if result:
            print("Login successful!")
        else:
            print("Invalid username or password.")
        
    except Error as e:
        print(f"Error: {e}")
    
    finally:
        if connection.is_connected():
            cursor.close()
            connection.close()

# 示例调用
login('user1', 'password1')