读取CSV文件,获取文件中的url地址,并对url地址进行解析,将图片以原名称下载保存到本地文件夹
import csv
import os
import requests
import tqdm
# 该脚本主要实现,对算法平台的调用记录生成的csv文件进行读取,获取文件中的url地址,并对url地址进行解析,将图片以原名称下载保存到本地文件夹中
def download_url(source,save_path):
with open(source) as f:
f_csv = csv.reader(f)
headers = next(f_csv)
for row in f_csv:
dict1 = row[-2]
# 排除返回值为空的情况
if len(dict1)==0:
continue
# 为str类型,先转为dict类型
dict1 = eval(dict1)
# 读取key对应的value值
url_path = dict1['url']
print(url_path)
response = requests.get(url_path)
file_name = url_path.split('/')[-1]
# 如果图片名称中带":",命名方式要加以修改
if ":" in file_name:
file_name0 = url_path.split('/')[-1].split(':')[-3]
file_name1 = url_path.split('/')[-1].split(':')[-2]
file_name2 = url_path.split('/')[-1].split(':')[-1]
file_name = file_name0 + '-' + file_name1 + '-' + file_name2
file_path = save_path + '/' + file_name
with open(file_path, 'wb') as f: # �Զ�����д���ļ�����
f.write(response.content)
print("图片保存完成")
首次更新,希望能对你有所帮助!