接口自动化测试小技巧-上传文件自动判断文件类型（详细）

---

前言

如何判断一个文件的类型呢，判断这个文件是png还jpg，还是mp3文件?
filetype 包是 python 用来判断文件类型的依赖包。

filetype 安装

pip install filetype

filetype是一个小巧自由开放Python开发包，主要用来获得文件类型。包要求Python 3.+

功能特色：
简单友好的API
支持宽范围文件类型
提供文件扩展名和MIME类型判断
文件的MIME类型扩展新增
通过文件（图像、视频、音频…）简单分析
可插拔：添加新的自定义类型的匹配
快，即使处理大文件
只需要前261个字节表示的最大文件头，这样你就可以通过一个单字节
依赖自由（只是Python代码，没有C的扩展，没有libmagic绑定）
跨平台文件识别

使用示例

import filetype


def main():
    kind = filetype.guess('tests/fixtures/sample.jpg')
    if kind is None:
        print('Cannot guess file type!')
        return

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

if __name__ == '__main__':
    main