前言
pymysql是在 Python3.x 版本中用于连接 MySQL 服务器的一个库。它可以帮助测试人员在Python应用程序中轻松地执行各种数据库操作,如查询数据、插入、更新和删除数据等。
1、连接数据库
步骤:连接数据库 -----> 生成游标对象 -----> 执行SQL语句 -----> 关闭游标 -----> 关闭连接,具体代码如下:
import pymysql
# 连接 MySQL 数据库
conn = pymysql.connect(
host='159.xxx.xxx.216', # 主机名
port=3306, # 端口号,MySQL默认为3306
user='xxxx', # 用户名
password='xxxx', # 密码
database='xx', # 数据库名称
)
# 创建游标对象
cursor = conn.cursor()
# 执行 SQL 查询语句
cursor.execute("SELECT * FROM users WHERE gender='女'")
# 获取查询结果
result = cursor.fetchall()
# 关闭游标连接
cursor.close()
# 关闭数据库连接
conn.close()
connect():pymysql模块里一个用来与数据库建立连接的方法。
cursor() :在使用pymysql.connect() 方法与数据库建立连接后,想要操作数据库时,就需要使用游标 cursor通过连接好的数据库(此处为conn)。调用 cursor() 方法即可返回一个新的游标对象,在连接没有关闭之前,游标对象可以反复使用。
cursor.execute():执行sql语句的方法,execute("这里填sql语句")。
cursor.fetchall():获取查询的结果集的一个方法,它返回的是一个元组,其元素是每一行的记录封装的一个元组。
cursor.close():关闭游标。先关闭游标,才能关闭数据库连接。
conn.close():关闭数据库连接。这里要注意如果要连接另外一个库,必须得先把上次的数据库连接关闭,不然无法连接另一个库。
3.列名参数化
将上文中的 执行sql查询语句 部分替换成下面的代码
#1、列名参数化
column1 = 'name'
column2 = 'age'
#2、使用{}作为占位符,表示等待参数传入;然后通过.format(变量名)传参数给{}
sql = """
SELECT
{},{}
FROM
users
WHERE
gender = '女';
""".format(column1,column2)
cursor.execute(sql)
4.where查询条件参数化
#创建sql语句
sql = "SELECT * FROM users WHERE gender=%s and age=%s"
#定义参数化查询
params = ('女',18)
#执行sql语句
cursor.execute(sql,params)
5.表名参数化
#1、表名参数化
table_name = 'users'
#2、创建sql语句
sql = "SELECT * FROM " +table_name + "WHERE gender=%s and age=%s"
#3、where后查询条件参数化
params = ('女',18)
#4、执行sql语句
cursor.execute(sql,params)
6.全部参数化
最后,如果要将列名、表名、查询条件全部参数化,参考如下
#1、表名参数化
table_name = 'users'
#2、列名参数化
column1 = 'name'
column2 = 'age'
#3、创建sql语句
sql = "SELECT {},{} FROM ".format(column1,column2) +table_name + "WHERE gender=%s and age=%s"
#4、where后查询条件参数化
params = ('女',18)
#5、执行sql语句
cursor.execute(sql,params)