当前位置：首页 > article >正文

如何使用 Python 连接 MySQL 数据库？

article 2024/12/23 2:15:58

在Python开发中，连接MySQL数据库是一个常见的需求。

我们可以使用多种库来实现这一功能，其中最常用的是mysql-connector-python和PyMySQL。

下面我将详细介绍如何使用这两个库来连接MySQL数据库，并提供一些实际开发中的建议和注意事项。

1. 使用 `mysql-connector-python` 连接MySQL数据库

mysql-connector-python 是MySQL官方提供的Python连接器，支持Python 3.x版本。

安装

首先，你需要安装 mysql-connector-python 库：

pip install mysql-connector-python

连接数据库

下面是一个简单的示例，展示如何连接到MySQL数据库：

import mysql.connector

# 创建数据库连接
mydb = mysql.connector.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="yourdatabase"
)

# 创建游标对象
mycursor = mydb.cursor()

# 执行SQL查询
mycursor.execute("SELECT * FROM yourtable")

# 获取查询结果
myresult = mycursor.fetchall()

for x in myresult:
  print(x)

注意事项

错误处理：在实际开发中，应该添加错误处理机制，以防止数据库连接失败或其他异常情况。

try:
    mydb = mysql.connector.connect(
        host="localhost",
        user="yourusername",
        password="yourpassword",
        database="yourdatabase"
    )
except mysql.connector.Error as err:
    print(f"Error: {err}")

关闭连接：在完成数据库操作后，应该关闭游标和数据库连接，以释放资源。

mycursor.close()
mydb.close()

2. 使用 `PyMySQL` 连接MySQL数据库

PyMySQL 是一个纯Python实现的MySQL客户端库，兼容Python 2.x和3.x版本。

安装

首先，你需要安装 PyMySQL 库：

pip install PyMySQL

连接数据库

下面是一个简单的示例，展示如何连接到MySQL数据库：

import pymysql

# 创建数据库连接
mydb = pymysql.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="yourdatabase"
)

# 创建游标对象
mycursor = mydb.cursor()

# 执行SQL查询
mycursor.execute("SELECT * FROM yourtable")

# 获取查询结果
myresult = mycursor.fetchall()

for x in myresult:
  print(x)

注意事项

错误处理：同样，应该添加错误处理机制。

try:
    mydb = pymysql.connect(
        host="localhost",
        user="yourusername",
        password="yourpassword",
        database="yourdatabase"
    )
except pymysql.MySQLError as err:
    print(f"Error: {err}")

关闭连接：在完成数据库操作后，应该关闭游标和数据库连接。

mycursor.close()
mydb.close()

实际开发中的建议

使用上下文管理器：为了确保资源被正确释放，可以使用上下文管理器（with语句）来管理数据库连接和游标。

import mysql.connector

with mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
) as mydb:
    with mydb.cursor() as mycursor:
        mycursor.execute("SELECT * FROM yourtable")
        myresult = mycursor.fetchall()
        for x in myresult:
            print(x)

参数化查询：为了避免SQL注入攻击，应该使用参数化查询。

import mysql.connector

with mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
) as mydb:
    with mydb.cursor() as mycursor:
        sql = "SELECT * FROM yourtable WHERE id = %s"
        val = (1,)
        mycursor.execute(sql, val)
        myresult = mycursor.fetchall()
        for x in myresult:
            print(x)

连接池：在高并发环境下，可以使用连接池来管理数据库连接，提高性能。

from mysql.connector.pooling import MySQLConnectionPool

pool = MySQLConnectionPool(
    pool_name="mypool",
    pool_size=5,
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)

with pool.get_connection() as mydb:
    with mydb.cursor() as mycursor:
        mycursor.execute("SELECT * FROM yourtable")
        myresult = mycursor.fetchall()
        for x in myresult:
            print(x)

在Python中连接MySQL数据库，可以使用 mysql-connector-python 或 PyMySQL 库。无论使用哪个库，都应该注意以下几点：