浏览代码

删除直接访问数据库任务时强制更新任务状态为1的处理机制

dongzhaorui 1 年之前
父节点
当前提交
726d55f6dc
共有 1 个文件被更改,包括 2 次插入13 次删除
  1. 2 13
      FworkSpider/feapder/pipelines/mongo_pipeline.py

+ 2 - 13
FworkSpider/feapder/pipelines/mongo_pipeline.py

@@ -12,6 +12,7 @@ from typing import Dict, List, Tuple
 from feapder.db.mongodb import MongoDB
 from feapder.pipelines import BasePipeline
 from feapder.utils.log import log
+import feapder.utils.tools as tools
 
 
 class MongoPipeline(BasePipeline):
@@ -92,17 +93,5 @@ class TaskPipeline(MongoPipeline):
         @param str table: 表名
         @param dict condition: 查询条件
         @param limit: 查询数量
-        @return:
         """
-        items = self.to_db.find(table, condition, limit)
-        datas = [{'_id': item['_id'], 'state': 1} for item in items]
-        update_keys = ['state']
-        try:
-            self.to_db.add_batch(
-                coll_name=table,
-                datas=datas,
-                update_columns=update_keys,
-            )
-        except Exception as e:
-            log.exception(e)
-        return items
+        return self.to_db.find(table, condition, limit)