2 回答

TA貢獻(xiàn)1898條經(jīng)驗(yàn) 獲得超8個(gè)贊
如果可能的話,我總是更喜歡使用 os 內(nèi)置的路徑并捕獲不受支持的類型
import os
video_types = ('.mp4', '.avi', '.jpeg')
image_types = ('.png', '.jpg')
filenames = ["/test/1.jpg","/test/2.avi","/test/unknown.xml","/test/noextention"]
for filename in filenames:
print(filename)
if os.path.splitext(filename)[1] in video_types:
print("Its a Video")
elif os.path.splitext(filename)[1] in image_types:
print("Its an Image")
else:
print("No Idea")
/test/1.jpg
Its an Image
/test/2.avi
Its a Video
/test/unknown.xml
No Idea
/test/noextention
No Idea

TA貢獻(xiàn)1829條經(jīng)驗(yàn) 獲得超7個(gè)贊
from urlparse import urlparse
from os.path import splitext
url = "sample/test/image.png"
image_formats = [".png", ".jpeg"]
video_formats = [".mp4", ".mp3", ".avi"]
def get_ext(url):
"""Return the filename extension from url, or ''."""
parsed = urlparse(url)
root, ext = splitext(parsed.path)
return ext
if get_ext(url) in image_formats:
print("it's an image")
elif get_ext(url) in video_formats:
print("it's a video")
else:
print("some differnet format")
它適用于任何類型的網(wǎng)址
www.example.com/image.jpg
https://www.example.com/page.html?foo=1&bar=2#fragment
https://www.example.com/resource
添加回答
舉報(bào)