[pymongo] 자주 쓰는 명령어 find_one_and_update (응용)

find_one_and_update는 말 그대로, 특정조건에 부합하는 문서를 찾아서, update 해주는 명령어다.

사용하다보면 update와 동시에 update된 문서를 받아서 사용하고 싶을때가 있다.

그냥 아래처럼 저장하면,

vData = user_col.find_one_and_update({'uid':1}, {'$inc':{ 'version': 1}})

uid:1을 가진 문서를 찾아서 'version' field의 값을 1 증가($inc)시키고, 문서를 return 하게되는데, 이때 return하는 문서는 update되기 이전의 문서, 즉 original문서이다.

이것을 업데이트된 문서로 반환받고 싶다면,

from pymongo import ReturnDocument

vData = user_col.find_one_and_update({'uid':1}, {'$inc':{ 'version': 1}}, return_document=ReturnDocument.AFTER)

이렇게 pymongo에서 ReturnDocument를 import하고,

return_document=ReturnDocument.AFTER를 추가로 전달해주면 된다. 

 

mongodb shell script는 아래와 같다.

getCollection('col').findOneAndUpdate({'uid':1}, {$inc:{'version': 1}}, {returnNewDocument:true})