利用Factory-boy来生成实例数据

 

库和版本：
Faker==2.0.0
factory-boy==2.12.0

官方文档：
https://factoryboy.readthedocs.io/en/latest/index.html

1. 新建两个object：User和School

user.py

class User():
    def __init__(self, name, num, age, school, city, phone):
        self.name, self.num, self.age, self.school, self.city, self.phone = 
            name, num, age, school, city, phone

school.py

class School:
    def __init__(self,schoolName):
        self.schoolName=schoolName

2. 给对象创建factory

import factory
import factory.fuzzy
from demo_factory_boy.fbobject.user import User
from demo_factory_boy.fbobject.school import School
from demo_factory_boy.fbprivader.numpv import NumProvider

factory.Faker.add_provider(NumProvider)  #添加自定义的NumProvider


class SchoolFactory(factory.Factory):
    class Meta:
        model = School

    schoolName = factory.sequence(lambda n: 'school%04d' % n)#factory.sequence


class UserFactory(factory.Factory):
    class Meta:
        model = User

    name = factory.Faker("name", locale="zh_CN")#factory.Faker 
    num = factory.Faker("num")
    age = factory.fuzzy.FuzzyInteger(42)#factory.fuzzy.FuzzyInteger
    city = factory.Faker("address", locale="zh_CN")
    phone = factory.fuzzy.FuzzyText("138", 7, "1", "1234567890")#factory.fuzzy.FuzzyText
    school = factory.SubFactory(SchoolFactory)#factory.SubFactory

    class Params:
        shipped = factory.Trait(
            name=None
        )

3.编写自定义Provider（如果没有Provider需求，可以跳过这个步骤）

from faker import Faker
from faker.providers import BaseProvider
import random

fake = Faker()

class NumProvider(BaseProvider):
    def num(self):
        ri = self.randomInt()
        return str(ri())

    def randomInt(self):
        return lambda a=1,b=30:random.randint(a,b)

    def randomStr18(self,num):
        ri = lambda a=0, b=9: random.randint(a, b)
        s = ''
        for i in range(18):
            s = s + str(ri())
        return s

4.使用

import factory
from demo_factory_boy.fbfactory.userfc import UserFactory

if __name__ == "__main__":
    # 单个生成
    seq = []
    uf = UserFactory()
    print(uf.__dict__)
    seq.append(uf.__dict__)
    seq.append(UserFactory().__dict__)
    print(seq)

    ##批量生成
    list = []
    fss = factory.build_batch(UserFactory, 4)
    for fs in fss:
        list.append(fs.__dict__)
    print(list)
    ##部分字段修改
    uff = UserFactory(shipped=True)
    print(uff.school.__dict__)
    print(uff.__dict__)
    print(uff.__dir__())

结果：