python如何检测文件或图片类型_python 获取是否是图片-CSDN博客

filetype · PyPI # 可以查看支持的文件类型

安装

pip install filetype

基本使用

import os
import filetype


def main():
    d = r'D:\Game\.cache'
    video = 'test.mp4'
    p = os.path.join(d, video)
    kind = filetype.guess(p)
    if kind is None:
        print('Cannot guess file type!')
        return

    print('File extension: %s' % kind.extension)  # 输出 File extension: mp4
    print('File MIME type: %s' % kind.mime)  # 输出 File MIME type: video/mp4


if __name__ == '__main__':
    main()

特殊用法

如果把一个png类型的图片文件手动改了后缀？

import filetype

def main(filename):
    kind = filetype.guess(filename)
    if kind is None:
        print('Cannot guess file type!')

    print('File extension: %s' % kind.extension)
    print('File MIME type: %s' % kind.mime)

if __name__ == '__main__':
    main('111.png')

输出为：

File extension: jpg
File MIME type: image/jpeg

这是因为本模块有check功能，只需要处理文件头的前261个字节即可，所以就算是大文件速度也很快。

[文章导入自 http://qzq-go.notion.site/54b56a087af4457fab2517fc60d559d3 访问原文获取高清图片]