解析json中的url并下载到本地

最新推荐文章于 2024-09-26 17:23:32 发布

studyeboy

最新推荐文章于 2024-09-26 17:23:32 发布

阅读量1.4k

点赞数

文章标签： json url urllib python

本文链接：https://blog.csdn.net/studyeboy/article/details/103125853

版权

json文件
提取json中的url并将图像下载到本地

import sys
import os
import argparse

import json
from urllib.request import urlretrieve


def parse_arguments(argv):
    parser = argparse.ArgumentParser()
    parser.add_argument('--input_json',type=str,default='./test1.json',\
                        help='input json file')
    parser.add_argument('--output_dir',type=str,default='./output',\
                        help='output dir')
    return parser.parse_args(argv)

def main(args):
    if not os.path.exists(args.output_dir):
        os.makedirs(args.output_dir)
    with open(args.input_json, 'r', encoding='utf-8') as f:
        data = json.load(f)
        print('data length:', len(data))
        for i in range(len(data)):
            #print(data[i]['src'])
            try:
                print('trying to get %s' % data[i]['src'])
                filename = os.path.basename(data[i]['src'])
                urlretrieve(data[i]['src'], os.path.join(args.output_dir,
                    filename ))
            except:
                print('Warning: Could not download image %s' % data[i]['src'])


if __name__ == '__main__':
    main(parse_arguments(sys.argv[1:]))