图片扩展 JPG 或 PNG

Question

我写了一个简单的 Python 程序，它工作正常。

基本上我写的输入图像 URL 并将其保存到计算机，但我想再添加一个无法修复的额外内容。 如果图像 URL 没有 Jpg 或 PNG 格式，我想避免错误。 我该怎么做？

代码如下：

import requests

def download():
    url = input("Enter Image URL >> ")
    response = requests.get(url)
    file_name = url.split("/")[-1]

    if file_name == "jpg":

        with open(file_name, "wb") as saveImg:
            if saveImg.write(response.content):
                print("[+] Download Complete")
            else:
                print("[-] Download Failed.")
    else:
        print("Failed")            


download()

Answer 1

你写了：

if file_name == "jpg":

你想要：

import os

_, ext = os.path.splitext(file_name)
if ext == "jpg":

也许：

if ext in ("jpg", "png"):

或者

if ext.lower() in ("jpg", "png"):

如果您不导入os ，您可能仍然可以使用if file_name.endswith(".jpg")之类的东西。

Answer 2

更新代码>>>

import requests
import os.path

def download():
    url = input("Enter Image URL >> ")
    response = requests.get(url)

    file_name = url.split("/")[-1]

    _, ext = os.path.splitext(file_name)

    if ext in (".jpg", ".png"):

        with open(file_name, "wb") as saveImg:
            if saveImg.write(response.content):
                print("[+] Download Complete")
            else:
                print("[-] Download Failed.")
    else:
        print("[-] Please Enter Valid Image URL.")            


download()