Python读取本地txt文件中图片的url并下载

Python读取本地txt文件中图片的url并下载

from urllib.request import urlretrieve
import os
import ssl

ssl._create_default_https_context = ssl._create_unverified_context

'''
通过txt网址文件，下载图片到本地
'''
def download():

    # 读取 test_data.txt 文本内的网址信息
    categories = ['test_data']

    for category in categories:

        # 新建存储picture文件夹存储图片
        os.makedirs('picture/%s' % category, exist_ok=True)
        
        # 读取txt文件
        with open('%s.txt' % category, 'r') as file:
            urls = file.readlines()
            
            # 计算链接地址条数
            n_urls = len(urls)

            # 遍历链接地址下载图片
            for i, url in enumerate(urls):
                try:
                     # 请求下载图片，并截取最后链接第一最后一节字段命名图片
                     urlretrieve(url.strip(), 'picture/%s/%s' % (category, url.strip().split('/')[-1]))
                     print('%s %i/%i' % (category, i, n_urls))
                except:
                     print('%s %i/%i' % (category, i, n_urls), 'no image')

if __name__ == '__main__':
    download();