# -*- coding: utf-8 -*- # @Author: wangkun # @Time: 2023/9/13 import argparse import os import sys sys.path.append(os.getcwd()) from common.common import Common from haitunzhufu.haitunzhufu_recommend.haitunzhufu_recommend2 import HTZFRecommend class HTZFMain: @classmethod def main(cls, log_type, crawler, env): videos_cnt = 100 Common.logger(log_type, crawler).info('开始抓取"海豚祝福"') HTZFRecommend.start_wechat(log_type=log_type, crawler=crawler, videos_cnt=videos_cnt, env=env) Common.logger(log_type, crawler).info("抓取一轮结束") Common.del_logs(log_type, crawler) if __name__ == "__main__": parser = argparse.ArgumentParser() ## 新建参数解释器对象 parser.add_argument('--log_type', type=str) ## 添加参数,注明参数类型 parser.add_argument('--crawler') ## 添加参数 parser.add_argument('--env') ## 添加参数 args = parser.parse_args() ### 参数赋值,也可以通过终端赋值 HTZFMain.main(log_type=args.log_type, crawler=args.crawler, env=args.env)