在自动化过程中,我们需要查询数据库,校验结果是否正确,比如充值完成之后,需要查询数据库,查看充值是否成功。
以下主要介绍,pyMysqL安装、操作流程、语法基础及封装操作数据库类。
一
pyMysqL介绍及安装
01 pyMysqL介绍
MysqL应该说是如今使用最为普遍的数据库了,没有之一,而Python作为最为流行的语言之一,自然少不了与MysqL打交道,其中PyMysqL就是使用最多的工具库。
02 pyMysqL安装
方式一:使用命令安装
pip install pyMysqL
方式二:PyCharm内部安装
导入模块:
二import pyMysqL
pyMysqL流程及模块说明
01 pyMysqL操作流程
02 pyMysqL模块说明
▌Connection对象
表示:conn=connect(参数列表)
作用:用于建立与数据库的连接;
参数列表:
▌对象的方法
对象方法如下:
close():关闭连接;
commit():提交;
cursor():返回Cursor对象,用于执行sql语句并获得结果;
execute(operation [, parameters ]):执行语句,返回受影响的行数,主要用于执行insert、update、delete语句,也可以执行create、alter、drop等语句;
▌Cursor对象
游标(cursor)就是游动的标识,通俗的说,一条sql取出对应n条结果资源的接口/句柄,就是游标,沿着游标可以一次取出一行。
用于执行sql语句,使用频度最高的语句为select、insert、update、delete;
▌对象的属性
rowcount只读属性,表示最近一次execute()执行后受影响的行数;
connection获得当前连接对象;
pyMysqL语法基础
01 代码示例
import pyMysqL # 连接数据库 conn = pyMysqL.connect(host='127.0.0.1', user='ITester', password='123456', database='ITester', charset='utf8') # 创建游标 cursor = conn.cursor() # 执行sql语句 sql = 'select * from user limit 3;' res = cursor.execute(sql) # 获取查询结果的1条数据 data = cursor.fetchone() print(data) # 关闭游标连接 cursor.close() # 关闭数据库连接 conn.close()
02 语法总结
1.连接数据库,需要host、user、password、database、charset等信息;
2.操作数据库先创建游标;
3.执行指定的sql语句,如果涉及到增、删、改数据库必须要conn.commit(),提交事务
4.查询获取数据条数有三种方法fetchone、fetchmany、fetchall。
5.需要注意的是,fetch获取的数据默认是元组,如果想要字典类型,
cursor=pyMysqL.cursors.DictCursor;
封装数据库类
01 封装说明
在实际项目中,很多地方都有用到数据库的操作,所以需要将数据库相关操作进行封装,方便其他模块调用。如下,在common目录下,新建文件db_handler.py 用于封装数据库操作。
db_handler.py
import pyMysqL class DBHandler: def __init__(self,host,port,user,password, database,charset,**kwargs): # 连接数据库服务器 self.conn = pyMysqL.connect(host=host, port=port, user=user,password=password, database=database,cursorclass=pyMysqL.cursors.DictCursor, charset=charset,**kwargs) # 获取游标 self.cursor = self.conn.cursor() def query(self, sql, args=None,one=True): self.cursor.execute(sql, args) # 提交事务 self.conn.commit() if one: return self.cursor.fetchone() else: return self.cursor.fetchall() def close(self): self.cursor.close() self.conn.close() if __name__ == "__main__": db = DBHandler(host='127.0.0.1', port=3306, user='ITester', password='123456', database='ITester', charset='utf8') sql = 'select * from user limit 1;' data = db.query(sql) print(data)