简单介绍:用python中的pymysql模块,提供了对数据库的一些操作。用python连接数据库比较灵活,将sql语句封装好后,可以在测试中轻松调用sql方法,提高工作效率;
数据库查询
def read_database(sql):
# 输入sql,进行数据库查询,返回查询的结果
conn=pymysql.connect(host=XXX, user=XXX, password="XXX,port=XXX,db=XXX,charset='utf8')
cur = conn.cursor()
sql=sql
cur.execute(sql)
result = cur.fetchall()
# print(result)
return result
# 实际查询的过程中,可能参数比较多,需要灵活运用参数化的方式
以student表的查询为例
查询符合条件的学生姓名:select name from student where age = {age} and gender = {gender} and class = {class}
def get_student_name_from_db()
数据库增删改
def new_database_con(sql):
"""用于删除、插入数据"""
conn = pymysql.connect(host=XXX, user=XXX, password=XXX,
port=XXX, db=XXX, charset='utf8')
cursor = conn.cursor()
try:
res = cursor.execute(sql)
print("======",res)
conn.commit()
except Exception as e:
print(e)
conn.rollback()
finally:
conn.close()
return resj
所以当后续要用到sql时,无论是查询还是增删改,调用封装好的方法,将sql参数化,返回想要获取的具体数据再处理;