run_xigua_follow.py 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. # -*- coding: utf-8 -*-
  2. # @Author: wangkun
  3. # @Time: 2023/2/17
  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 xigua.xigua_follow.xigua_follow import Follow
  11. from common.feishu import Feishu
  12. def main(log_type, crawler, strategy, oss_endpoint, env, machine):
  13. while True:
  14. try:
  15. Common.logger(log_type, crawler).info('开始抓取 西瓜视频 定向榜\n')
  16. Follow.get_follow_videos(log_type, crawler, strategy, oss_endpoint, env, machine)
  17. Common.del_logs(log_type, crawler)
  18. Common.logger(log_type, crawler).info('抓取完一轮,休眠 1 分钟\n')
  19. time.sleep(60)
  20. except Exception as e:
  21. Common.logger(log_type, crawler).info(f"西瓜视频异常,触发报警:{e}\n")
  22. Feishu.bot(log_type, crawler, e)
  23. if __name__ == "__main__":
  24. parser = argparse.ArgumentParser() ## 新建参数解释器对象
  25. parser.add_argument('--log_type', type=str) ## 添加参数,注明参数类型
  26. parser.add_argument('--crawler') ## 添加参数
  27. parser.add_argument('--strategy') ## 添加参数
  28. parser.add_argument('--our_uid') ## 添加参数
  29. parser.add_argument('--oss_endpoint') ## 添加参数
  30. parser.add_argument('--env') ## 添加参数
  31. parser.add_argument('--machine') ## 添加参数
  32. args = parser.parse_args() ### 参数赋值,也可以通过终端赋值
  33. # print(args)
  34. main(log_type=args.log_type,
  35. crawler=args.crawler,
  36. strategy=args.strategy,
  37. oss_endpoint=args.oss_endpoint,
  38. env=args.env,
  39. machine=args.machine)