Python:从一个网站拉取.png，输出到另一个网站

import requests

def download_image(url, save_path):
    response = requests.get(url)
    if response.status_code == 200:
        with open(save_path, 'wb') as file:
            file.write(response.content)
        return True
    return False

def upload_image(url, file_path):
    with open(file_path, 'rb') as file:
        files = {'file': file}
        response = requests.post(url, files=files)
        if response.status_code == 200:
            return True
        return False

def main():
    # 从网站A拉取.png文件
    url_a = 'http://example.com/image.png'
    save_path = 'image.png'
    if download_image(url_a, save_path):
        print('成功从网站A拉取.png文件')

        # 输出到网站B
        url_b = 'http://example.com/upload'
        if upload_image(url_b, save_path):
            print('成功将.png文件输出到网站B')
        else:
            print('将.png文件输出到网站B失败')
    else:
        print('从网站A拉取.png文件失败')

if __name__ == '__main__':
    main()

pip install requests