Python实战MySQL：数据库操作全流程详解

大家好，我是涛哥，本文内容来自 涛哥聊Python ，转载请标原创。

更多Python学习内容：http://ipengtao.com

MySQL是一种广泛使用的关系型数据库管理系统，Python可以通过多种方式与MySQL进行交互。本文将详细介绍如何使用Python操作MySQL数据库，包括安装必要的库、连接数据库、执行基本的CRUD（创建、读取、更新、删除）操作，并包含具体的示例代码，帮助全面掌握这一过程。

准备工作

安装MySQL服务器

首先，需要安装MySQL服务器。如果还没有安装，可以从MySQL官网下载并安装适合操作系统的版本。

安装Python MySQL驱动

为了使Python能够与MySQL数据库进行交互，我们需要安装一个MySQL驱动。常用的MySQL驱动有mysql-connector-python和PyMySQL。这里以mysql-connector-python为例。

使用以下命令安装mysql-connector-python：

pip install mysql-connector-python

连接到MySQL数据库

安装好驱动后，可以使用Python连接到MySQL数据库。

import mysql.connector

def connect_to_mysql():
    try:
        connection = mysql.connector.connect(
            host='localhost',
            user='your_username',
            password='your_password',
            database='your_database'
        )
        if connection.is_connected():
            print("成功连接到数据库")
            return connection
    except mysql.connector.Error as err:
        print(f"连接错误: {err}")
        return None

# 示例
connection = connect_to_mysql()
if connection:
    connection.close()

在这个示例中，定义了一个connect_to_mysql函数，用于连接到MySQL数据库。请根据实际数据库信息替换host、user、password和database参数。

创建数据库和表

创建数据库

在使用数据库之前，需要先创建一个数据库。

def create_database(cursor, database_name):
    try:
        cursor.execute(f"CREATE DATABASE {database_name}")
        print(f"数据库 {database_name} 创建成功")
    except mysql.connector.Error as err:
        print(f"创建数据库错误: {err}")

# 示例
connection = connect_to_mysql()
if connection:
    cursor = connection.cursor()
    create_database(cursor, 'test_db')
    cursor.close()
    connection.close()

创建表

接下来，在数据库中创建一个表。

def create_table(cursor, table_name):
    create_table_query = f"""
    CREATE TABLE {table_name} (
        id INT AUTO_INCREMENT PRIMARY KEY,
        name VARCHAR(255) NOT NULL,
        age INT,
        gender VARCHAR(10)
    )
    """
    try:
        cursor.execute(create_table_query)
        print(f"表 {table_name} 创建成功")
    except mysql.connector.Error as err:
        print(f"创建表错误: {err}")

# 示例
connection = connect_to_mysql()
if connection:
    cursor = connection.cursor()
    cursor.execute("USE test_db")
    create_table(cursor, 'persons')
    cursor.close()
    connection.close()

插入数据

插入单条数据

可以使用INSERT INTO语句向表中插入数据。

def insert_data(cursor, table_name, name, age, gender):
    insert_query = f"INSERT INTO {table_name} (name, age, gender) VALUES (%s, %s, %s)"
    data = (name, age, gender)
    try:
        cursor.execute(insert_query, data)
        print("数据插入成功")
    except mysql.connector.Error as err:
        print(f"插入数据错误: {err}")

# 示例
connection = connect_to_mysql()
if connection:
    cursor = connection.cursor()
    cursor.execute("USE test_db")
    insert_data(cursor, 'persons', 'Alice', 30, 'Female')
    connection.commit()
    cursor.close()
    connection.close()

插入多条数据

也可以一次插入多条数据。

def insert_multiple_data(cursor, table_name, data_list):
    insert_query = f"INSERT INTO {table_name} (name, age, gender) VALUES (%s, %s, %s)"
    try:
        cursor.executemany(insert_query, data_list)
        print("多条数据插入成功")
    except mysql.connector.Error as err:
        print(f"插入数据错误: {err}")

# 示例
data_list = [
    ('Bob', 25, 'Male'),
    ('Charlie', 28, 'Male'),
    ('Diana', 22, 'Female')
]

connection = connect_to_mysql()
if connection:
    cursor = connection.cursor()
    cursor.execute("USE test_db")
    insert_multiple_data(cursor, 'persons', data_list)
    connection.commit()
    cursor.close()
    connection.close()

查询数据

查询所有数据

可以使用SELECT语句查询表中的所有数据。

def query_all_data(cursor, table_name):
    query = f"SELECT * FROM {table_name}"
    try:
        cursor.execute(query)
        results = cursor.fetchall()
        for row in results:
            print(row)
    except mysql.connector.Error as err:
        print(f"查询数据错误: {err}")

# 示例
connection = connect_to_mysql()
if connection:
    cursor = connection.cursor()
    cursor.execute("USE test_db")
    query_all_data(cursor, 'persons')
    cursor.close()
    connection.close()

条件查询

可以使用条件查询从表中获取特定的数据。

def query_data_with_condition(cursor, table_name, condition):
    query = f"SELECT * FROM {table_name} WHERE {condition}"
    try:
        cursor.execute(query)
        results = cursor.fetchall()
        for row in results:
            print(row)
    except mysql.connector.Error as err:
        print(f"查询数据错误: {err}")

# 示例
connection = connect_to_mysql()
if connection:
    cursor = connection.cursor()
    cursor.execute("USE test_db")
    query_data_with_condition(cursor, 'persons', "age > 25")
    cursor.close()
    connection.close()

更新数据

可以使用UPDATE语句更新表中的数据。

def update_data(cursor, table_name, set_clause, condition):
    update_query = f"UPDATE {table_name} SET {set_clause} WHERE {condition}"
    try:
        cursor.execute(update_query)
        print("数据更新成功")
    except mysql.connector.Error as err:
        print(f"更新数据错误: {err}")

# 示例
connection = connect_to_mysql()
if connection:
    cursor = connection.cursor()
    cursor.execute("USE test_db")
    update_data(cursor, 'persons', "age = 31", "name = 'Alice'")
    connection.commit()
    cursor.close()
    connection.close()

删除数据

可以使用DELETE语句删除表中的数据。

def delete_data(cursor, table_name, condition):
    delete_query = f"DELETE FROM {table_name} WHERE {condition}"
    try:
        cursor.execute(delete_query)
        print("数据删除成功")
    except mysql.connector.Error as err:
        print(f"删除数据错误: {err}")

# 示例
connection = connect_to_mysql()
if connection:
    cursor = connection.cursor()
    cursor.execute("USE test_db")
    delete_data(cursor, 'persons', "name = 'Diana'")
    connection.commit()
    cursor.close()
    connection.close()

关闭数据库连接

完成所有操作后，需要关闭数据库连接。

connection.close()
print("数据库连接已关闭")

总结

本文详细介绍了如何使用Python操作MySQL数据库，包括连接数据库、创建数据库和表、插入数据、查询数据、更新数据和删除数据。通过安装MySQL服务器和Python MySQL驱动，可以轻松地在Python中实现与MySQL的交互。文章提供了详细的示例代码，展示了如何执行基本的CRUD操作，帮助大家掌握在实际项目中高效管理和操作数据库的技巧。通过这些示例，可以学会在Python中实现数据库操作，提高数据处理的效率和可靠性。