dongzhaorui 2 年之前
父節點
當前提交
1e31f095a1
共有 1 個文件被更改,包括 5 次插入1 次删除
  1. 5 1
      find_source/build_excavate.py

+ 5 - 1
find_source/build_excavate.py

@@ -1,5 +1,9 @@
+import sys
+
 from crawler import BreadthCrawler
 
+EXCAVATE_WORKERS = sys.argv[0] or 1
+
 
 def main():
     BreadthCrawler(
@@ -7,7 +11,7 @@ def main():
         enable_search=False,
         enable_query=False,
         enable_excavate=True,
-        excavate_workers=1,
+        excavate_workers=EXCAVATE_WORKERS,
     ).start()