|
|
import os |
|
|
|
|
|
video_extensions = { |
|
|
'.mp4', '.m4a', '.m4v', '.f4v', '.f4a', '.m4b', '.m4r', '.f4b', '.mov', |
|
|
'.3gp', '.3gp2', '.3g2', '.3gpp', '.3gpp2', '.ogg', '.oga', '.ogv', '.ogx', |
|
|
'.wmv', '.wma', '.asf', '.webm', '.flv', '.avi', '.gifv', '.mkv', '.rm', |
|
|
'.rmvb', '.vob', '.dvd', '.mpg', '.mpeg', '.mp2', '.mpe', '.mpv', '.mpg', |
|
|
'.mpeg', '.m2v', '.svi', '.3gp', '.mxf', '.roq', '.nsv', '.flv', '.f4v', |
|
|
'.f4p', '.f4a', '.f4b' |
|
|
} |
|
|
|
|
|
image_extensions = { |
|
|
'.jpg', '.jpeg', '.jpe', '.jif', '.jfif', '.jfi', '.png', '.gif', |
|
|
'.webp', '.tiff', '.tif', '.psd', '.raw', '.arw', '.cr2', '.nrw', |
|
|
'.k25', '.bmp', '.dib', '.heif', '.heic', '.ind', '.indd', '.indt', |
|
|
'.jp2', '.j2k', '.jpf', '.jpx', '.jpm', '.mj2', '.svg', '.svgz', |
|
|
'.ai', '.eps', '.ico' |
|
|
} |
|
|
|
|
|
|
|
|
def is_video_file(filename): |
|
|
return os.path.splitext(filename)[-1].lower() in video_extensions |
|
|
|
|
|
|
|
|
def is_image_file(filename): |
|
|
return os.path.splitext(filename)[-1].lower() in image_extensions |
|
|
|
|
|
|
|
|
def is_video_or_image(filename): |
|
|
file_extension = os.path.splitext(filename)[-1].lower() |
|
|
|
|
|
return file_extension in video_extensions or file_extension in image_extensions |
|
|
|
