python写mysql函数

pip install mysql-connector-python

import mysql.connector

def query_mysql(host, user, password, database, query):
    try:
        # 连接到MySQL数据库
        connection = mysql.connector.connect(
            host=host,
            user=user,
            password=password,
            database=database
        )
        
        # 创建一个cursor对象
        cursor = connection.cursor()
        
        # 执行SQL查询
        cursor.execute(query)
        
        # 获取查询结果
        results = cursor.fetchall()
        
        # 关闭cursor和连接
        cursor.close()
        connection.close()
        
        return results
    
    except mysql.connector.Error as err:
        print(f"Error: {err}")
        return None

# 使用函数
if __name__ == "__main__":
    host = 'localhost'
    user = 'your_username'
    password = 'your_password'
    database = 'your_database'
    query = 'SELECT * FROM your_table'
    
    results = query_mysql(host, user, password, database, query)
    if results:
        for row in results:
            print(row)