如何解决无法使用 Factory Boy 和 Faker
我正在尝试使用 django-factory-boy 和 faker 创建多个 django 模型实例。但是我需要批量创建实例(不是单个)。但我不能使这两个属性都对应(货币的 code
和 name
)。
class Currency(models.Model):
"""Currency model"""
name = models.CharField(max_length=120,null=False,blank=False,unique=True)
code = models.CharField(max_length=3,unique=True)
symbol = models.CharField(max_length=5,default='$')
def __str__(self) -> str:
return self.code
import factory
from apps.Payment.models import Transaction,Currency
from faker import Faker
fake = Faker()
class CurrencyFactory(factory.django.DjangoModelFactory):
class Meta:
model = Currency
# code and name get assigned when the class is called hence if we use
# create_batch(n) we get all n object same
# code,name = fake.currency()
code = factory.LazyAttribute(lambda _: fake.currency()[0])
name = factory.LazyAttribute(lambda _: fake.currency()[1])
symbol = '$'
我面临的问题是 code
和 name
得到不同的值并且不匹配。看看 faker
返回什么。
>>> from faker import Faker
>>> fake = Faker()
>>> fake.currency()
('JPY','Japanese yen')
查看货币名称与货币代码不对应。我还需要使用 CurrencyFactory.create_batch(5)
创建至少 5 到 6 个对象。
# mismatch in code and name
NAME CODE
Netherlands Antillean guilder ZAR
Western Krahn language UGX
Colombian peso KHR
我想要的
NAME CODE
Indian National Rupee INR
Japanese yen JPY
解决方法
最好的方法是通过class Params
:
class CurrencyFactory(factory.model.DjangoModelFactory):
class Meta:
model = Currency
class Params:
currency = factory.Faker("currency") # (code,name)
code = factory.LazyAttribute(lambda o: o.currency[0])
name = factory.LazyAttribute(lambda o: o.currency[1])
想法是:
- 工厂调用 Faker 一次,生成
currency = (code,name)
参数; - 然后将该参数的组件映射到模型的正确字段
- 由于
currency
被声明为参数,因此不会传递给模型(它会自动添加到Meta.exclude
。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。