用faker和xlwt创建一个假信息库导入excel用于测试

Posted 2021-07-10 帅气的黑桃J

前言

有需求，学了一下，可以直接调用。其中faker用来生成假信息，xlwt操作excel，当然用openpyxl也可以操作。

不多说，直接放上函数

import xlwt
from faker import Faker

def initFaker():
    f = Faker(locale='zh_CN')
    return f

#返回一个数组
def getInfo():
    f = initFaker()
    name  =f.name()
    print(name)

    number = f.phone_number()
    print(number)

    city = f.city()
    print(city)

    birth = f.date()
    print(birth)

    postcode = f.postcode()
    print(postcode)

    company = f.company()
    print(company)
    return [name,number,city,birth,postcode,company]



#创建count条数据
def insertInfo(count):
    # 创建workbook和sheet对象
    workbook = xlwt.Workbook()  # 注意Workbook的开头W要大写
    sheet1 = workbook.add_sheet('sheet1', cell_overwrite_ok=True)
    for row in range(count):
        info = getInfo()
        for col in range(6):
            sheet1.write(row,col,info[col])

    # 保存该excel文件,有同名文件时直接覆盖
    workbook.save('myexcel.xlsx')

    print('创建excel文件完成!')

if __name__ == '__main__':
    insertInfo(100)

运行结果：

完美~

参考文献：
faker库的使用
openpyxl操作手册