使用Python中的pymysql模块连接数据库

小蓝 • 2024-12-12 13:32 • 编程

一、背景介绍

在当今互联网时代，数据是非常重要的资源，大量数据需要存储、处理和管理。对于程序员来说，操作数据库是每个人必备的技能之一。Python有很多数据库模块可供使用，其中pymysql模块是一个非常常用的数据库模块，它可以方便地连接MySQL数据库，进行增删改查等操作。

二、pymysql模块的安装

在使用pymysql模块之前，需要先安装。在命令行中输入以下命令：

pip install pymysql

输入后等待安装完成即可。

三、连接MySQL数据库

使用pymysql模块连接MySQL数据库需要以下几个步骤：

1、导入pymysql模块和数据库信息：

import pymysql

# 打开数据库连接
db = pymysql.Connect(host='localhost', user='root', password='123456', database='test', port=3306, charset='utf8')

2、创建游标对象：

# 使用cursor()方法创建一个游标对象
cursor = db.cursor()

3、执行SQL语句：

# 编写SQL查询语句
sql = "SELECT * FROM user"

# 执行SQL语句
cursor.execute(sql)

# 获取查询结果
result = cursor.fetchall()
print(result)

4、提交事务和关闭连接：

# 提交事务
db.commit()

# 关闭游标和数据库连接
cursor.close()
db.close()

四、常用操作示例

1、插入数据

# 编写SQL插入语句
sql = "INSERT INTO user(username, password, email) VALUES('Tom', '123456', 'tom@qq.com')"

# 执行插入操作
cursor.execute(sql)

# 提交事务
db.commit()

2、更新数据

# 编写SQL更新语句
sql = "UPDATE user SET username='Jerry' WHERE id=1"

# 执行更新操作
cursor.execute(sql)

# 提交事务
db.commit()

3、删除数据

# 编写SQL删除语句
sql = "DELETE FROM user WHERE id=1"

# 执行删除操作
cursor.execute(sql)

# 提交事务
db.commit()

4、查询数据

# 编写SQL查询语句
sql = "SELECT * FROM user"

# 执行查询操作
cursor.execute(sql)

# 获取查询结果
result = cursor.fetchall()
print(result)

五、错误处理

在进行数据库操作时，可能会存在一些错误，例如连接失败、查询失败等。因此，我们需要对这些错误进行合理的处理，防止程序崩溃。

try:
    # 打开数据库连接
    db = pymysql.Connect(host='localhost', user='root', password='123456', database='test', port=3306, charset='utf8')

    # 使用cursor()方法创建一个游标对象
    cursor = db.cursor()

    # 编写SQL查询语句
    sql = "SELECT * FROM user"

    # 执行SQL语句
    cursor.execute(sql)

    # 获取查询结果
    result = cursor.fetchall()
    print(result)

    # 提交事务
    db.commit()

except Exception as e:
    # 发生异常，回滚事务
    db.rollback()

finally:
    # 关闭游标和数据库连接
    cursor.close()
    db.close()

六、总结

pymysql模块是Python中连接MySQL数据库的一种重要方式，它可以轻松实现对数据库的增删改查等操作。使用pymysql模块需要注意一些细节问题，例如SQL语句的编写、事务的提交等。同时，对于错误的处理也是非常重要的。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/249299.html