被数据库延迟加载坑了一把

背景

目前做Flask开发用的数据库框架是peewee。peewee查询时默认是延迟加载，也就是说一个查询语句不会立刻将所有结果返回，而是等到使用这个查询结果时才真正从数据库中查询出来。在这样一个背景下，就踩了这个坑。

过程

按照我需要处理的业务逻辑，首先要记录用户使用记录，然后给用户重置数据，最后恢复用户的使用记录。

1.查询记录保存到变量
在这业务逻辑下我想到的就是首先将用户的使用记录查询出来

learn_status = sm.UserTask.select().where(sm.UserTask.learning_status.in_([1,2]), sm.UserTask.plan_id==plan_id, sm.UserTask.user_id==user_id)

2.然后给用户重置

此处省略过程

3.从变量生成列表，恢复记录

learn_finished_days = [item.date for item in learn_status]
sm.UserTask.update(learning_status=0, checkin_status=0).where(sm.UserTask.user_id==34113904, sm.UserTask.plan_id==2).execute()
sm.UserTaskTrain.update(is_finish=0).where(sm.UserTaskTrain.user_id==34113904, sm.UserTaskTrain.plan_id==2).execute()

当我恢复记录时就傻眼了，发现一个记录都没有了，仔细一想问题就出现在peewee的延迟加载上，在第一步中查询出记录，此时并没有真正将记录赋值给变量，延迟加载机制生效了，而在第二步中又删除了用户所有的记录，所以最后在第三步中使用变量写列表生成式时，就找不到相应的记录。

教训是惨痛的，最后是我手动修复了那些有问题的数据。如果给我一次重来的机会，避免这个问题： 在数据库查询之后立刻生成列表推导式。当后面删除真正的数据时也不会有影响。