[英]Image extension JPG or PNG
I wrote a simple Python program and its working fine.我写了一个简单的 Python 程序,它工作正常。
Basically what I wrote that enter image URL and its save to the computer but I want to add one more extra thing that couldn't fix it.基本上我写的输入图像 URL 并将其保存到计算机,但我想再添加一个无法修复的额外内容。 I want to avoid error if image URL don't have Jpg or PNG format.如果图像 URL 没有 Jpg 或 PNG 格式,我想避免错误。 How Should i do that?我该怎么做?
Code is below:代码如下:
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()
You wrote:你写了:
if file_name == "jpg":
import os
_, ext = os.path.splitext(file_name)
if ext == "jpg":
or perhaps:也许:
if ext in ("jpg", "png"):
or或者
if ext.lower() in ("jpg", "png"):
If you don't import os
, you might still be able to get away with something like if file_name.endswith(".jpg")
.如果您不导入os
,您可能仍然可以使用if file_name.endswith(".jpg")
之类的东西。
updated code >>>更新代码>>>
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()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.