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()