run_scheduling_task.py 1.3 KB

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