pymysql连接数据库后查询是否连接成功

原创

mob64ca12e5502a 2024-12-16 12:17:01 ©著作权

文章标签 mysql 数据库连接 Python 文章分类 MySQL 数据库

©著作权归作者所有：来自51CTO博客作者mob64ca12e5502a的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用pymysql连接数据库并查询连接是否成功

在学习Python的过程中，连接数据库是一项非常重要的技能。接下来我会教你如何使用pymysql库连接数据库，并检查连接是否成功。我们将分步骤进行，每一步都会包含代码示例和解释。

流程概述

以下是连接数据库并验证连接的步骤：

步骤	说明
1	安装pymysql库
2	导入pymysql模块
3	创建数据库连接对象
4	检查连接是否成功
5	关闭数据库连接

步骤详解

步骤1：安装pymysql库

在使用pymysql之前，首先需要确保库已经安装。可以通过pip安装。

pip install pymysql

步骤2：导入pymysql模块

导入pymysql库以便我们可以使用它的方法。

import pymysql  # 导入pymysql模块

步骤3：创建数据库连接对象

使用pymysql提供的方法来连接到数据库。你需要提供数据库的主机、用户名、密码和数据库名称。

# 创建连接对象
connection = pymysql.connect(
    host='localhost',     # 数据库主机地址
    user='your_username', # 用户名
    password='your_password', # 密码
    database='your_database'  # 数据库名称
)

步骤4：检查连接是否成功

要验证连接是否成功，可以使用connection.open属性或尝试进行一次查询。如果连接成功，则会返回True。

try:
    if connection.open:
        print("数据库连接成功！")  # 打印连接成功的信息

        # 可以选择执行一个简单的查询测试
        with connection.cursor() as cursor:
            cursor.execute("SELECT VERSION()")  # 查询数据库版本
            version = cursor.fetchone()  # 获取结果
            print(f"数据库版本: {version[0]}")  # 打印数据库版本
        
except pymysql.MySQLError as e:
    print(f"连接数据库失败：{e}")  # 打印错误信息
finally:
    connection.close()  # 关闭数据库连接

步骤5：关闭数据库连接

记得在完成操作后关闭数据库连接，避免占用资源。

connection.close()  # 关闭数据库连接

代码示例

结合以上步骤，完整代码如下：

import pymysql  # 导入pymysql模块

# 创建连接对象
connection = pymysql.connect(
    host='localhost',     
    user='your_username', 
    password='your_password', 
    database='your_database'  
)

try:
    if connection.open:
        print("数据库连接成功！")

        # 执行一个简单查询
        with connection.cursor() as cursor:
            cursor.execute("SELECT VERSION()")
            version = cursor.fetchone()
            print(f"数据库版本: {version[0]}")

except pymysql.MySQLError as e:
    print(f"连接数据库失败：{e}")
finally:
    connection.close()  # 关闭数据库连接

序列图

sequenceDiagram
    participant User
    participant Python
    participant MySQL
    User->>Python: 请求连接数据库
    Python->>MySQL: 发起连接请求
    MySQL-->>Python: 返回连接成功
    Python-->>User: 数据库连接成功
    Python->>MySQL: 查询数据库版本
    MySQL-->>Python: 返回数据库版本
    Python-->>User: 输出数据库版本
    Python->>MySQL: 关闭连接
    MySQL-->>Python: 返回关闭确认

类图

classDiagram
    class Connection {
        +connect()
        +cursor()
        +close()
    }
    class Cursor {
        +execute(query)
        +fetchone()
    }
    Connection o-- Cursor : uses