一、安装pymysql
在Python中使用pymysql模块前,我们首先需要安装它。可以通过以下pip命令进行安装:
pip install pymysql
二、连接到MySQL服务器
安装完成后,我们可以通过以下代码连接到MySQL服务器:
import pymysql
# 连接数据库
db = pymysql.connect(host='localhost', user='root', password='root', database='test')
# 获取操作游标
cursor = db.cursor()
其中,host是MySQL服务器所在的主机名或IP地址,user是登录用户名,password是登录密码,database是需要操作的数据库名。
三、执行SQL语句
连接到数据库后,我们可以通过游标对象执行SQL语句。如下面的代码演示了插入、查询、更新和删除四种基本的SQL操作:
# 插入数据
sql_insert = "INSERT INTO students(name, age) VALUES('Tom', 20)"
cursor.execute(sql_insert)
db.commit() # 提交事务
# 查询数据
sql_select = "SELECT * FROM students"
cursor.execute(sql_select)
results = cursor.fetchall() # 获取所有查询结果
for row in results:
print(row)
# 更新数据
sql_update = "UPDATE students SET age = 21 WHERE name = 'Tom'"
cursor.execute(sql_update)
db.commit() # 提交事务
# 删除数据
sql_delete = "DELETE FROM students WHERE name = 'Tom'"
cursor.execute(sql_delete)
db.commit() # 提交事务
注意,每次执行修改(包括插入、更新、删除)数据库的操作后,都需要调用db.commit()方法来提交事务,否则修改的结果不会被真正保存到数据库中。
四、异常处理
在执行SQL操作时,可能会出现各种各样的错误,如语法错误、操作非法等。为了防止程序因为这些错误而中断,我们可以使用Python的异常处理机制来捕获并处理这些错误:
try:
# 执行SQL语句
cursor.execute(sql_insert)
# 提交事务
db.commit()
except Exception as e:
# 如果出现错误,则回滚事务
db.rollback()
print("发生错误:", e)
五、关闭连接
在完成所有操作后,我们需要关闭游标和数据库连接,以释放资源:
# 关闭游标和数据库连接
cursor.close()
db.close()
以上就是使用pymysql操作MySQL数据库的基本流程和方法。在实际使用过程中,还需要根据具体的需求和场景进行相应的调整和优化。希望本文能帮助你更好地理解和使用pymysql模块,更有效地在Python中操作MySQL数据库。