图像扩展JPG或PNG
提问于 2020-05-16 14:41:25
我编写了一个简单的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()
回答 2
Stack Overflow用户
回答已采纳
发布于 2020-05-16 14:56:49
你写道：
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")的东西。
Stack Overflow用户
发布于 2020-05-16 22:20:33
更新代码>>>
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()页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/61844563
复制
腾讯云开发者
