python 第一篇博客媒体链接5

Posted 2021-05-09

import requests
url = 'https://www.reddit.com/r/wallpapers.json'
response = requests.get(url)
if not response.ok:
    print("Error", response.status_code)
    exit()
data = response.json()['data']['children']
first_post = data[0]['data']
image_url = first_post['url']
if '.png' in image_url:            
    extension = '.png'        
elif '.jpg' in image_url or '.jpeg' in image_url:            
    extension = '.jpeg'
else:            
    image_url += '.jpeg'           
    extension = '.jpeg'
# prevents thumbnails denoting removed images from being downloaded
image = requests.get(image_url, allow_redirects=False)
if(image.status_code == 200):
    try:
        output_filehandle = open(first_post['title'] + extension, mode='bx')
        output_filehandle.write(image.content)
    except:
        pass