如何在Django中为外键字段添加Faker模拟数据

问题原因 #

你在创建Product时直接给user字段传整数1会报错，因为Django的ForeignKey字段user对应的是User模型的实例对象，而非数据库存储的ID值。如果要直接传ID，需使用外键字段对应的_id后缀字段（即user_id）。

解决方法 #

方法1：使用user_id直接传入ID值 #

修改创建代码，将user=1改为user_id=1：

Product.objects.create(
    Product_name=fake.name(),
    Product_desc=fake.text(),
    Price=fake.random_int(3000, 10000),
    user_id=1
)

方法2：传入User实例对象 #

先获取已存在的User实例，再传递给user字段：

# 确保id为1的User已存在，否则会抛出DoesNotExist异常
user = User.objects.get(id=1)
Product.objects.create(
    Product_name=fake.name(),
    Product_desc=fake.text(),
    Price=fake.random_int(3000, 10000),
    user=user
)

方法3：随机关联已有的User（适合批量生成测试数据） #

如果要批量生成Product并随机关联现有User，可使用以下方式：

# 随机选取一个存在的User实例
user = User.objects.order_by('?').first()
if user:  # 避免User表为空时触发异常
    Product.objects.create(
        Product_name=fake.name(),
        Product_desc=fake.text(),
        Price=fake.random_int(3000, 10000),
        user=user
    )

注意事项 #

• 无论采用哪种方法，都要确保目标User在数据库中已存在，否则会引发异常。
• 批量生成测试数据时，建议先批量创建足够的User记录，再关联到Product。

内容的提问来源于stack exchange，提问作者Yousaf Zahid