run_shipinhao_search.py 1.1 KB

123456789101112131415161718192021222324252627
  1. # -*- coding: utf-8 -*-
  2. # @Author: wangkun
  3. # @Time: 2023/5/8
  4. import argparse
  5. import os
  6. import sys
  7. sys.path.append(os.getcwd())
  8. from common.common import Common
  9. from shipinhao.shipinhao_search.shipinhao_search import ShipinhaoSearch
  10. def main(log_type, crawler, env):
  11. Common.logger(log_type, crawler).info('开始抓取:视频号搜索\n')
  12. Common.logging(log_type, crawler, env, '开始抓取:视频号搜索\n')
  13. ShipinhaoSearch.get_search_videos(log_type, crawler, env)
  14. Common.del_logs(log_type, crawler)
  15. Common.logger(log_type, crawler).info('抓取完一轮\n')
  16. Common.logging(log_type, crawler, env, '抓取完一轮\n')
  17. if __name__ == "__main__":
  18. parser = argparse.ArgumentParser() ## 新建参数解释器对象
  19. parser.add_argument('--log_type', type=str) ## 添加参数,注明参数类型
  20. parser.add_argument('--crawler') ## 添加参数
  21. parser.add_argument('--env') ## 添加参数
  22. args = parser.parse_args() ### 参数赋值,也可以通过终端赋值
  23. main(log_type=args.log_type, crawler=args.crawler, env=args.env)