From 0f5bb2e8cbfa99be7b106f36ef15768b8536a043 Mon Sep 17 00:00:00 2001 From: laike9m Date: Sat, 19 Mar 2016 23:15:25 +0800 Subject: [PATCH] add remove_question_before.py --- query/remove_question_before.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 query/remove_question_before.py diff --git a/query/remove_question_before.py b/query/remove_question_before.py new file mode 100644 index 0000000..4e64517 --- /dev/null +++ b/query/remove_question_before.py @@ -0,0 +1,23 @@ +""" +移除某个时间点之后提出的问题, 以及它们的答案. 在程序出问题时使用 +""" + +from datetime import datetime + +import pymongo + +t = datetime(2016,3,11,18,0,0) + +db = pymongo.MongoClient('127.0.0.1', 27017).zhihu_data +q_colls = ["19550517_q", "19551147_q", "19561087_q", "19553298_q"] + +for q_coll in q_colls: + coll = db.get_collection(q_coll) + acoll = db.get_collection(q_coll[:-1] + 'a') + for qdoc in coll.find(): + if qdoc['time'] < t: + result = coll.delete_one({'qid': qdoc['qid']}) + assert result.deleted_count == 1 + print("question %s removed" % qdoc['qid']) + result = acoll.delete_many({'qid': qdoc['qid']}) + print("%d answers removed" % result.deleted_count)