更新Django模型对象后关联外键模型值未同步问题排查

问题本质 #

r和r2是内存中两个独立的Python对象，它们虽然对应数据库里同一条Room记录，但修改r2并保存后，r的内存数据不会自动同步数据库的最新状态，所以出现ID一致但颜色不匹配的情况。

详细过程拆解 #

• 执行h.rooms.create()时，Django完成两个操作：在数据库插入Room记录，同时在内存生成一个Room对象并赋值给r，此时r.color为初始值'blue'。
• 调用h.rooms.get()时，Django会重新查询数据库，基于查询结果生成全新的Room对象赋值给r2——它和r是完全分离的内存实例，二者共享数据库记录，但内存中的属性值各自独立。
• 修改r2.color为'green'并保存，只会更新数据库里的记录，不会影响r这个内存对象的属性值，r.color依然是最初的'blue'，所以断言self.assertEqual(r2.color, r.color)必然失败。

解决办法 #

方法1：同步内存对象与数据库最新状态

在断言前调用refresh_from_db()方法，让r从数据库拉取最新数据：

def test_color_change(self):
    h = House(name='new house')
    h.save()

    r = h.rooms.create(
        name='living room',
        color='blue'
    )
    r2 = h.rooms.get(name='living room')
    r2.color = 'green'
    r2.save()
    r.refresh_from_db()  # 同步数据库最新值到r
    self.assertEqual(r.id, r2.id)
    self.assertEqual(r2.color, r.color)

方法2：复用同一个内存对象修改

既然已经持有r对象，无需重新查询r2，直接修改r的属性并保存即可避免实例不一致问题：

def test_color_change(self):
    h = House(name='new house')
    h.save()

    r = h.rooms.create(
        name='living room',
        color='blue'
    )
    r.color = 'green'
    r.save()
    self.assertEqual(r.color, 'green')
    # 如果需要验证数据库值，可重新查询后断言
    r2 = h.rooms.get(name='living room')
    self.assertEqual(r.color, r2.color)

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