UI自动化测试常用操作函数（2）

在做断言的时候，有时候需要查询数据库

创建数据库连接的时候，可以将连接信息参数化

import pymysql
class Service:
    # 创建数据库连接
    @classmethod
    def ConnectMysql(cls，host,uname,pwd,database):
        db = pymysql.connect(host=host, user=uname, password=pwd, database=database)
        cur = db.cursor()
        return db, cur

    # 断开数据库连接
    @classmethod
    def connectOff(cls, db, cur):
        db.commit()
        cur.close()
        db.close()

    # mysql查询得到的结果函数，供调用,这里的sql参数是sql语句
    @classmethod
    def select(cls, sql):
        db, cur = cls.ConnectMysql()
        try:
            cur.execute(sql)
            result = cur.fetchall()
            return result
        except Exception as e:
            print(str(e))
        finally:
            cls.connectOff(db, cur)