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

在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)

注意事项

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

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

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

   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)

注意事项

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

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

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

   mycursor.close()
   mydb.close()

实际开发中的建议

1. 使用上下文管理器 ：为了确保资源被正确释放，可以使用上下文管理器（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)

2. 参数化查询：为了避免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)

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

   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 库。无论使用哪个库，都应该注意以下几点：

1. 错误处理：添加错误处理机制，以防止异常情况。
2. 关闭连接：在完成数据库操作后，关闭游标和数据库连接。
3. 使用上下文管理器：使用上下文管理器来管理资源。
4. 参数化查询：使用参数化查询来防止SQL注入攻击。
5. 连接池：在高并发环境下，使用连接池来提高性能。