Python(连接数据库，对数据库进行增删改查）

Python连接数据库需要安装一个外部库pymysql

接下来我们通过实际代码演示如何进行 数据库连接

#import外部库pymysql，下载安装
import pymysql
#这是我们封装的用来获取数据库的配置文件，ip，port，password，user的一个方法，具体代码不做展示了
from KeJie.com.readYaml import getDatabase



class DbConnect():
    def __init__(self, database="congku"):
        db = getDatabase()
#获取到数据库的ip地址，端口，用户名，密码
        ip = db['dbip']
        port = db['port']
        usr = db['usr']
        pwd = db['pwd']
#以下两行代码就是pymysql为我们提供的连接数据库的方法
        self.db = pymysql.connect(host=ip, port=port, user=usr, password=pwd,database=database)
        self.cursor = self.db.cursor()
#封装一个关闭数据库的方法
    # 关闭数据库
    def close(self):
        self.db.close()
#封装数据库查询方法
    # 封装select
    def select(self, sql):
        self.cursor.execute(sql)
        res = self.cursor.fetchall()
        # print(res)
        return res
#封装修改，删除，插入方法，
    # 修改，删除，插入
    def execute(self, sql):
        try:
            self.cursor.execute(sql)
            self.db.commit()
        except:
            self.db.rollback()
            self.db.close()

#通过main对类进行实例化并使用其中的方法。

if __name__=='__main__':
    dc = DbConnect()
    #查询create_time
    sql1 = "select create_time from dtmoban_warehouse_in_warehouse_chuku WHERE czc_number='PO2022442768' ORDER BY ordersn DESC"
    unix = dc.select(sql1)[0][0]
    print(unix)

有了python连接数据库的方法，我们就可以对数据库进行一些操作，辅助我们对自动化测试进行断言，例如当你通过接口创建了一个数据，如果你做的接口自动化测试的要求很高需要到“数据库确实有创建出该数据”，那么就可以通过以上这个方法封装一个数据库连接进行使用