使用python 判断请求url是否是图片链接
时间: 2024-06-09 18:10:50 浏览: 280
你可以使用Python中的urllib库来获取请求的URL,并使用os库中的path库来获取文件扩展名。然后,可以通过检查文件扩展名来确定URL是否是图像链接。下面是代码示例:
```python
import urllib.request
import os.path
def is_image_url(url):
# 获取URL的文件扩展名
ext = os.path.splitext(url)[1]
# 如果扩展名是图片类型,则返回True
if ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp']:
return True
# 否则返回False
return False
# 测试代码
url = 'https://www.example.com/image.jpg'
response = urllib.request.urlopen(url)
if response.getheader('Content-Type').startswith('image/'):
print('This is an image URL')
else:
print('This is not an image URL')
```
在上面的代码中,我们首先使用urllib库中的urlopen()函数打开URL。接下来,我们检查响应的Content-Type头是否以'image/'开头。如果是,则可以确定URL是一个图像链接。如果不是,我们可以使用os.path.splitext()函数获取URL的文件扩展名,并检查它是否是图像类型。如果是,我们返回True,否则返回False。
阅读全文