Python基础教程（十八）：MySQL - mysql-connector 驱动

用户11147438

发布于 2024-06-15 12:34:29

880

发布于 2024-06-15 12:34:29

引言

在Web开发和数据分析领域，数据库的操作是一项核心技能。Python，作为一种流行的编程语言，提供了多种库来与MySQL数据库进行交互，其中mysql-connector-python是官方推荐的驱动之一。本文将详细介绍如何使用mysql-connector-python来创建数据库、插入、删除、查询数据以及进行排序等操作。

一、安装 mysql-connector-python

在开始之前，确保你已经安装了mysql-connector-python库。如果没有，可以通过pip安装：

pip install mysql-connector-python

二、连接数据库

首先，我们需要建立与MySQL数据库的连接。以下是连接代码的基本结构：

import mysql.connector
from mysql.connector import Error

try:
    connection = mysql.connector.connect(host='localhost',
                                         database='testdb',
                                         user='root',
                                         password='password')
    if connection.is_connected():
        db_Info = connection.get_server_info()
        print("Connected to MySQL Server version ", db_Info)
except Error as e:
    print("Error while connecting to MySQL", e)
finally:
    if (connection.is_connected()):
        connection.close()
        print("MySQL connection is closed")

三、创建数据库

要使用mysql-connector-python创建数据库，我们首先需要连接到MySQL服务器，然后执行SQL语句来创建数据库：

def create_database(connection, query):
    cursor = connection.cursor()
    try:
        cursor.execute(query)
        print("Database created successfully")
    except Error as e:
        print(f"The error '{e}' occurred")

create_db_query = "CREATE DATABASE testdb"
create_database(connection, create_db_query)

四、插入数据

一旦数据库创建完成，我们就可以在其中创建表并插入数据：

def insert_data(connection, query):
    cursor = connection.cursor()
    try:
        cursor.execute(query)
        connection.commit()
        print("Data inserted successfully")
    except Error as e:
        print(f"The error '{e}' occurred")

insert_query = """
INSERT INTO students (id, name, age)
VALUES (1, 'John Doe', 22),
       (2, 'Jane Doe', 25);
"""
insert_data(connection, insert_query)

五、查询数据

查询数据是数据库操作中最常见的需求之一。使用mysql-connector-python，我们可以轻松地执行SELECT语句：

def select_data(connection, query):
    cursor = connection.cursor()
    try:
        cursor.execute(query)
        result = cursor.fetchall()
        for row in result:
            print(row)
    except Error as e:
        print(f"The error '{e}' occurred")

select_query = "SELECT * FROM students"
select_data(connection, select_query)

六、更新数据

更新数据涉及到修改现有记录：

def update_data(connection, query):
    cursor = connection.cursor()
    try:
        cursor.execute(query)
        connection.commit()
        print("Data updated successfully")
    except Error as e:
        print(f"The error '{e}' occurred")

update_query = "UPDATE students SET age = 26 WHERE name = 'Jane Doe'"
update_data(connection, update_query)

七、删除数据

删除数据是另一个基本操作，我们可以通过以下方式实现：

def delete_data(connection, query):
    cursor = connection.cursor()
    try:
        cursor.execute(query)
        connection.commit()
        print("Data deleted successfully")
    except Error as e:
        print(f"The error '{e}' occurred")

delete_query = "DELETE FROM students WHERE id = 1"
delete_data(connection, delete_query)

八、排序数据

对数据进行排序可以使用ORDER BY子句：

def sort_data(connection, query):
    cursor = connection.cursor()
    try:
        cursor.execute(query)
        result = cursor.fetchall()
        for row in result:
            print(row)
    except Error as e:
        print(f"The error '{e}' occurred")

sort_query = "SELECT * FROM students ORDER BY age DESC"
sort_data(connection, sort_query)

九、结论

通过mysql-connector-python，我们可以方便地与MySQL数据库进行交互，无论是创建数据库、插入数据、查询、更新还是删除数据，都能轻松实现。掌握这些基本操作，将大大增强你在项目中的数据库管理能力。以上就是使用Python和mysql-connector-python操作MySQL数据库的完整指南。希望这篇文章能帮助你更好地理解和使用数据库操作。