run_youtube_follow.py 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. # -*- coding: utf-8 -*-
  2. # @Author: wangkun
  3. # @Time: 2023/2/3
  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 youtube.youtube_follow.youtube_follow_api import YoutubeFollow
  11. def main(log_type, crawler, strategy, oss_endpoint, env, machine):
  12. while True:
  13. Common.logger(log_type, crawler).info('开始抓取youtube定向榜\n')
  14. YoutubeFollow.get_follow_videos(log_type, crawler, strategy, oss_endpoint, env, machine)
  15. Common.del_logs(log_type, crawler)
  16. Common.logger(log_type, crawler).info('抓取完一轮,休眠 1 分钟\n')
  17. time.sleep(60)
  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('--strategy') ## 添加参数
  23. parser.add_argument('--our_uid') ## 添加参数
  24. parser.add_argument('--oss_endpoint') ## 添加参数
  25. parser.add_argument('--env') ## 添加参数
  26. parser.add_argument('--machine') ## 添加参数
  27. args = parser.parse_args() ### 参数赋值,也可以通过终端赋值
  28. # print(args)
  29. main(log_type=args.log_type,
  30. crawler=args.crawler,
  31. strategy=args.strategy,
  32. oss_endpoint=args.oss_endpoint,
  33. env=args.env,
  34. machine=args.machine)