Python高手进阶：pymysql vs mysql-connector-python，哪种方式连接MySQL更胜一筹

在Python中连接MySQL数据库，pymysql和mysql-connector-python都是常用的选择，但它们的应用场景和流行程度有所不同。

pymysql

pymysql是一个纯Python实现的MySQL驱动库，它提供了一个简单的接口来与MySQL数据库交互。它的优势在于它是纯Python实现的，不需要额外安装MySQL的C语言客户端库。这意味着在一些环境中更容易部署，例如某些类型的容器化环境或者Windows系统上。

mysql-connector-python

mysql-connector-python是MySQL官方提供的Python驱动库，它支持Python 2和Python 3，并且提供了丰富的功能和良好的性能。由于它是官方支持的产品，因此在文档和支持方面可能会更好一些。

使用情况

根据社区的反馈和趋势来看，pymysql在社区中似乎更为常见和广泛使用。这可能是因为它简单易用，并且在很多情况下性能足够满足需求。然而，如果你正在构建一个大型项目或企业级应用，并希望得到官方的支持和维护，那么mysql-connector-python可能是更好的选择。

在Python中连接MySQL数据库通常使用mysql-connector-python或pymysql库。下面是使用这两种库的基本示例。

1. 使用 mysql-connector-python

首先需要安装mysql-connector-python，可以通过pip来安装：

pip install mysql-connector-python

然后可以使用以下代码来建立连接：

import mysql.connector
from mysql.connector import Error

def create_connection():
    """ 创建到MySQL的连接 """
    try:
        connection = mysql.connector.connect(
            host='localhost',
            database='your_database',
            user='your_username',
            password='your_password'
        )
        if connection.is_connected():
            db_info = connection.get_server_info()
            print("Connected to MySQL Server version ", db_info)
            return connection
    except Error as e:
        print("Error while connecting to MySQL", e)
    return None

# 调用函数创建连接
conn = create_connection()
if conn:
    # 连接成功后可以执行查询等操作
    cursor = conn.cursor()
    query = "SELECT * FROM your_table"
    cursor.execute(query)
    records = cursor.fetchall()
    for row in records:
        print(row)
    
    # 关闭游标和连接
    cursor.close()
    conn.close()

2. 使用 pymysql

对于pymysql，同样需要先安装：

pip install pymysql

接着使用如下示例代码来建立连接：

import pymysql

def create_connection():
    """ 创建到MySQL的连接 """
    try:
        connection = pymysql.connect(
            host='localhost',
            user='your_username',
            password='your_password',
            database='your_database',
            charset='utf8mb4',
            cursorclass=pymysql.cursors.DictCursor
        )
        print("Connected to MySQL")
        return connection
    except Exception as e:
        print("Error while connecting to MySQL", e)
    return None

# 调用函数创建连接
conn = create_connection()
if conn:
    try:
        with conn.cursor() as cursor:
            sql = "SELECT * FROM your_table"
            cursor.execute(sql)
            result = cursor.fetchall()
            for row in result:
                print(row)
    finally:
        conn.close()

总结

• 如果考虑兼容性和稳定性 ：建议使用mysql-connector-python，因为它是由MySQL官方维护的。
• 如果考虑简单性和易用性 ：可以选择pymysql，因为它易于安装和使用，并且在许多场景下性能也足够好。
• 如果考虑部署的便利性 ：pymysql不需要MySQL C客户端库，所以在一些特定环境下（如Docker容器）更容易部署。

综上所述，两者都有广泛的使用基础，但pymysql在社区中的使用似乎更加普遍。不过，最终的选择应该基于项目的具体需求和个人偏好。

pymysql完整的mysql操作代码

import pymysql


def create_mysql_connection(host, user, password, database):
    """ 创建到MySQL数据库的连接 """
    try:
        connection = pymysql.connect(
            host=host,
            user=user,
            password=password,
            database=database,
            charset='utf8mb4',
            cursorclass=pymysql.cursors.DictCursor
        )
        print("Connected to MySQL")
        return connection
    except Exception as e:
        print("Error while connecting to MySQL", e)
    return None


def execute_sql(connection, sql, values=None):
    """ 执行SQL语句并返回结果 """
    try:
        with connection.cursor() as cursor:
            if values is not None:
                cursor.execute(sql, values)
            else:
                cursor.execute(sql)
            results = cursor.fetchall()
        connection.commit()
        return results
    except Exception as e:
        print("Error executing SQL:", e)
        connection.rollback()
        return None


def insert_user(connection, name, email, age):
    """ 向 users 表中插入一条新记录 """
    sql = "INSERT INTO users (name, email, age) VALUES (%s, %s, %s)"
    execute_sql(connection, sql, (name, email, age))


def delete_user(connection, user_id):
    """ 从 users 表中删除指定ID的记录 """
    sql = "DELETE FROM users WHERE id = %s"
    execute_sql(connection, sql, (user_id,))


def update_user(connection, user_id, new_name, new_email, new_age):
    """ 更新 users 表中指定ID的记录 """
    sql = "UPDATE users SET name=%s, email=%s, age=%s WHERE id=%s"
    execute_sql(connection, sql, (new_name, new_email, new_age, user_id))


def select_users(connection):
    """ 查询 users 表中的所有记录 """
    sql = "SELECT * FROM alerts limit 3"
    return execute_sql(connection, sql)


# 创建连接
connection = create_mysql_connection(
    host='127.0.0.1',
    user='zabreport',
    password='tFGZ2nzEGw5SmRBf',
    database='zabreport',
)

# # 插入数据
# insert_user(connection, 'John Doe', 'john@example.com', 30)
#
# # 删除数据
# delete_user(connection, 1)
#
# # 更新数据
# update_user(connection, 2, 'Jane Smith', 'jane@example.com', 28)

# 查询数据
users = select_users(connection)
for user in users:
    print(user)

# 关闭连接
connection.close()