python 第一篇博客媒体链接4

Posted 2021-05-09

import requests
url = 'https://www.reddit.com/r/wallpapers.json'
response = requests.get(url, headers={'User-agent': 'your-bot-name 0.1'})
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']
image = requests.get(image_url)
# checks url and sets appropriate extension for the image file
if '.png' in image_url:            
    extension = '.png'        
elif '.jpg' in image_url or '.jpeg' in image_url:            
    extension = '.jpeg'
if(image.status_code == 200):
    try:
        output_filehandle = open(first_post['title'] + extension, mode='bx')
        output_filehandle.write(image.content)
    except:
        pass