python 第一篇博客媒体链接6

Posted 2021-05-09

import requests
url = 'https://www.reddit.com/r/wallpapers.json?limit=100'
response = requests.get(url)
if not response.ok:
    print("Error", response.status_code)
    exit()
data = response.json()['data']['children']
for i in range(len(data)):
    current_post = data[i]['data']
    image_url = current_post['url']
    if '.png' in image_url:            
        extension = '.png'        
    elif '.jpg' in image_url or '.jpeg' in image_url:            
        extension = '.jpeg'
    elif 'imgur' in image_url:            
        image_url += '.jpeg'           
        extension = '.jpeg'        
    else:            
        continue
    image = requests.get(image_url, allow_redirects=False)
    if(image.status_code == 200):
        try:
            output_filehandle = open(current_post['title'] + extension, mode='bx')
            output_filehandle.write(image.content)
        except:
            pass