数据库登录系统代码

import mysql.connector

def login(username, password):
    try:
        # 建立数据库连接
        connection = mysql.connector.connect(
            host="localhost",
            user="db_user",
            password="db_password",
            database="mydatabase"
        )
        cursor = connection.cursor()
        
        # 查询数据库验证用户
        query = "SELECT * FROM users WHERE username = %s AND password = %s"
        cursor.execute(query, (username, password))
        result = cursor.fetchone()
        
        if result:
            print("登录成功")
        else:
            print("用户名或密码错误")
    except mysql.connector.Error as err:
        print(f"数据库错误: {err}")
    finally:
        # 关闭游标和连接
        if cursor:
            cursor.close()
        if connection.is_connected():
            connection.close()

# 用户输入
username = input("请输入用户名: ")
password = input("请输入密码: ")

login(username, password)