DeprecationWarning: count is deprecated. Use Collection.count_documents instead

环境

python 3.7
mongodb v4.2.1
pymongo 3.9.0

问题

DeprecationWarning: count is deprecated. Use Collection.count_documents instead.
print(cursor.count())

场景

当我使用pymongo查询出对应的cursor（find出的document的迭代器），然后查看查询出数据的数量时使用如下代码：

db = MongoClient(host='192.168.1.3', port=27017)
# dbname为操作的数据库名称，collectionname为操作的集合名称
coll = db.dbname.collectionname
cursor = coll.find({'status': '1'}).limit(10)
print(cursor.count())

然后会弹出警报：`DeprecationWarning: count is deprecated. Use Collection.count_documents instead.
print(cursor.count())`

原因

这是由于mongodb现在迭代器数据的搜集Collection.count()方式已经被弃用，更高的版本不适用。已经使用Collection.count_documents来统计了。

尝试

当我使用如下代码时

cursor = coll.find({'status': '1'})
print(cursor.count_documents())

会出现错误：AttributeError: 'Cursor' object has no attribute 'count_documents'，原来find后的对象并没有count_documents这个方法或者属性，正确的使用方法是这样的：

num = coll.count_documents({'status': '1'})
print(num)

思考

新的用法固然好用，查询与统计只需一个就解决了，但是想想我的需求我需要先查找出，判断是否有数据，来执行下一步操作。这样新的方法就不满足需求了，如果查找一次再用count_documents可能俩次查找统计的并不是同一批数据。这个怎么办，我又不想用count（看着警报不爽），曲线救国吧，我发现我条件查询的数据并不是很多（为什么我要强调不是很多呢，因为一旦将迭代器转化成list，就需要占据内存，不能过于大，我建议最好再1w条数据以内），那就将迭代器转化成list，这样不就可以用len来统计数量了。

实现

cursor = coll.find({'status': '1'}).limit(10)
# 迭代器只能遍历使用一次，最好将list定义到一个新的变量，以便重复使用
cursor_list = list(cursor)
# 查看是否有status为0的数据
if len(cursor_list) != 0:
    # 这里循环不能使用cursor，因为cursor迭代器在转化成list时已经使用，再次使用就没有东西迭代了
    for document in cursor_list:
        print(document)