run_scheduling_task_v3.py 1.2 KB

123456789101112131415161718192021222324252627282930313233
  1. # -*- coding: utf-8 -*-
  2. # @Author: wangkun
  3. # @Time: 2023/3/7
  4. import argparse
  5. import os
  6. import sys
  7. sys.path.append(os.getcwd())
  8. from common.common import Common
  9. from scheduling.scheduling_v3.crawler_scheduling_v3 import SchedulingV3
  10. class SchedulingTask:
  11. @classmethod
  12. def scheduling_task(cls, log_type, crawler, env):
  13. # Common.logger(log_type, crawler).info("开始调度爬虫任务")
  14. SchedulingV3.scheduling_task(log_type, crawler, env)
  15. # Common.logger(log_type, crawler).info("爬虫任务调度完成")
  16. Common.del_logs(log_type, crawler)
  17. if __name__ == "__main__":
  18. parser = argparse.ArgumentParser() ## 新建参数解释器对象
  19. parser.add_argument('--log_type', default='follow', type=str) ## 添加参数,注明参数类型
  20. parser.add_argument('--crawler', default='youtube') ## 添加参数
  21. parser.add_argument('--env', default='dev') ## 添加参数
  22. # parser.add_argument('--machine') ## 添加参数
  23. args = parser.parse_args() ### 参数赋值,也可以通过终端赋值
  24. # print(args)
  25. SchedulingTask.scheduling_task(
  26. log_type=args.log_type,
  27. crawler=args.crawler,
  28. env=args.env,
  29. )