使用python-magic判断文件是否是文档文件格式
要使用python-magic库判断文件是否是文档文件格式,你需要先安装python-magic库。可以使用以下命令安装:
pip install python-magic
安装完成后,可以使用以下代码进行判断:
import magic
def is_document_file(filename):
mime = magic.from_file(filename, mime=True)
if 'text' in mime or 'pdf' in mime or 'word' in mime or 'excel' in mime:
return True
else:
return False
# 测试示例
print(is_document_file('example.txt')) # True
print(is_document_file('example.png')) # False
在上面的代码中,is_document_file函数接受一个文件名作为参数,并使用magic.from_file方法获取文件的MIME类型。然后,通过判断MIME类型中是否包含文本、PDF、Word或Excel等关键词来确定文件是否是文档文件格式。如果是文档文件格式,则返回True,否则返回False。
请注意,magic.from_file方法可以根据文件的内容进行判断,而不仅仅是根据文件扩展名。因此,它可以更准确地判断文件类型
原文地址: http://www.cveoy.top/t/topic/iXqv 著作权归作者所有。请勿转载和采集!