import requests
import csv
import random
import os
首先创建文件存储路径
path = 'ski_img'
if not os.path.exists(path):
os.mkdir(path)
定义存储图片函数,传入两个参数,雪板型号以及图片url。这里UA可以替换自己的可用ua。
def save_img(model, img_url):
try:
response = requests.get(url=img_url, headers={'User-Agent':random.choice(USER_AGENTS)})
img_data = response.content
with open(f'{path}/{model}.jpg','wb') as f:
f.write(img_data)
except Exception as e:
print('Error',e)
打开上篇文章保存好的csv文件, save!
if __name__ == '__main__':
with open('ski_info.csv','r') as f:
li_1 = csv.reader(f)
for row in li_1:
save_img(row[1],row[4])
成果展示 👍