添加层级深度标志

find_source/crawler/Task.py

@@ -1,3 +1,4 @@
+import json
 import time
 from collections import UserDict
 
@@ -8,6 +9,7 @@ class Task(UserDict):
             self,
             name='',
             host='',
+            depth=None,
             loop_times=0,
             sensitive=False,
             duplication=False,
@@ -18,6 +20,7 @@ class Task(UserDict):
         super(Task, self).__init__(
             name=name,
             host=host,
+            depth=(depth or 1),
             loop_times=loop_times,
             sensitive=sensitive,
             duplication=duplication,
@@ -30,18 +33,7 @@ class Task(UserDict):
 # if __name__ == '__main__':
 #     task = Task(name='剑鱼')
 #     print(task, type(task))
-#     print('name' in task)
 #     b = json.dumps(task, default=lambda obj: obj.__dict__['data'], ensure_ascii=False)
 #     print(b, type(b))
-#     c = json.loads(b)
+#     c = json.loads(b, object_hook=lambda d: Task(**d))
 #     print(c, type(c))
-#     print(c['name'])
-#     print(task.name)
-#     task.name = '857'
-#     print(task.name)
-#     task.loop_times += 1
-#     print(task.loop_times)
-#     print(task)
-
-
-