run_cpu_memory.py 1.1 KB

12345678910111213141516171819202122232425262728
  1. # -*- coding: utf-8 -*-
  2. # @Author: wangkun
  3. # @Time: 2023/7/20
  4. import argparse
  5. import os
  6. import sys
  7. sys.path.append(os.getcwd())
  8. from common.common import Common
  9. from monitor.cpu_memory.cpu_memory import MonitorCpuMemory
  10. class MonitorMain:
  11. @classmethod
  12. def monitor_main(cls, log_type, crawler, env):
  13. Common.logger(log_type, crawler).info(f"开始监控脚本 cpu、memory 状态\n")
  14. Common.logging(log_type, crawler, env, f"开始监控脚本 cpu、memory 状态\n")
  15. MonitorCpuMemory.get_all_script_cpu_memory(log_type, crawler, env)
  16. Common.logger(log_type, crawler).info("监控一轮结束\n")
  17. Common.logging(log_type, crawler, env, "监控一轮结束\n")
  18. if __name__ == "__main__":
  19. parser = argparse.ArgumentParser() ## 新建参数解释器对象
  20. parser.add_argument('--log_type', type=str) ## 添加参数,注明参数类型
  21. parser.add_argument('--crawler') ## 添加参数
  22. parser.add_argument('--env') ## 添加参数
  23. args = parser.parse_args() ### 参数赋值,也可以通过终端赋值
  24. MonitorMain.monitor_main(log_type=args.log_type, crawler=args.crawler, env=args.env)