生成任务第一次更新

DAPlatform/Dao/data_set.py

@@ -10,14 +10,13 @@ collection = db.dataset
 
 
 # 添加用户的数据单
-def insert_data_set(user_id, data_list, database):
+def insert_data_set(user_id, data_list):
     try:
         data_set = {
             'user_id': user_id,
             'date': datetime.datetime.now(),
             'data_list': data_list,
-            'complete': 0,
-            'database': database,  # 用来标识该数据是位于爬虫数据库中哪个collection中的
+            'complete': 0
         }
         re = collection.insert_one(data_set)
         return re
@@ -66,11 +65,25 @@ def set_complete(_id):
 def createtask(id):
     try:
         result = collection.find_one({'_id': ObjectId(id)})
-        print(result['data_list'])
+        for i in result['data_list']:
+            print(i['id'])
+        return True
     except Exception as e:
         print(e)
         return False
 
 
 if __name__ == '__main__':
-    createtask("5f787bfb18cf2c88516f48c8")
+    # user_id = 1
+    # data_list = [
+    #     {
+    #         'id': ObjectId("5f6ab8573180f90885a62014"),
+    #         'collection': 'Traffic'
+    #     },
+    #     {
+    #         'id': ObjectId("5f6ab84f900bf9eddbe1233c"),
+    #         'collection': 'Weather'
+    #     }
+    # ]
+    # insert_data_set(user_id, data_list)
+    createtask(ObjectId("5f8282d8a94094ae27938e76"))