Python写自动化之数据库操作

在软件测试中，涉及到数据存储时，经常会用到数据库，这是就需要使用测试脚本操作数据库了

文件型数据库sqllite读写操作：

def ExecSql(dbFileName,stats):
    """
    | ##@函数目的: 执行sqlite语句
    | ##@参数说明：
    | ##@返回值：  sqlite数据库查询结果
    | ##@函数逻辑:
    | ##@注意：本函数只适用于查询sqlite数据库
    """
    con = sqlite3.connect(dbFileName)
    cur = con.cursor()
    stats = stats.decode("gbk")
    cur.execute(stats)
    con.commit()
    DBExecResult = cur.fetchall()
    con.close()
    return DBExecResult

MySQL数据库读写操作：

def ExecSql(sql, dbName="", host="", user="", passwd="", charset=''):
    import MySQLdb

    conn = MySQLdb.connect(host=host, user=user, passwd=passwd, charset=charset)
    conn.select_db(dbName)
    cursor = conn.cursor()
    cursor.execute(sql)
    conn.commit()
    result = cursor.fetchall()
    cursor.close()
    conn.close()
    return result

当然还有一些其他操作，比如说批量插入多条数据，参数化传入SQL语句等，具体的大家可以在网络上查找

欢迎关注“搜狗测试”公众号，每天一篇测试相关的文章与您分享，共同讨论软件测试的技术与发展

转载请注明：http://blog.csdn.net/sogouauto