正文

python如何判断文件类型

小六 /1 浏览量
0924

在Python中判断文件类型可以通过以下几种方法：

使用`filetype`库

首先，确保安装了`filetype`库，可以使用`pip install filetype`进行安装。

然后，在Python脚本中导入`filetype`库，并使用`filetype.guess（）`方法来猜测文件类型。

```python

import filetype

def guess_file_type（file_path）:

kind = filetype.guess（file_path）

if kind is None:

return "Cannot guess file type!"

return kind.extension, kind.mime

file_path = 'test.txt'

file_extension, file_mime = guess_file_type（file_path）

print（f"File extension: {file_extension}"）

print（f"File MIME type: {file_mime}"）

通过文件扩展名判断
使用`os.path.splitext`函数获取文件扩展名，然后根据扩展名判断文件类型。```python
import os
def guess_file_type_by_extension（file_path）:
 _, ext = os.path.splitext（file_path）
 if ext == '.txt':
 return "This is a text file."
 elif ext in ['.jpg', '.jpeg']:
 return "This is a JPEG image file."
 其他文件类型判断...
file_path = 'example.txt'
print（guess_file_type_by_extension（file_path））

通过文件头判断

读取文件的前几个字节，根据文件头中的特定字节序列来判断文件类型。

```python

def guess_file_type_by_header（file_path）:

with open（file_path, 'rb'） as file:

header = file.read（8） 读取文件头的前8个字节

根据文件头判断文件类型

if header.startswith（b'\xFF\xD8\xFF'）:

return "JPEG"

elif header.startswith（b'\x89\x50\x4E\x47'）:

return "PNG"

其他文件头判断...

file_path = 'example.jpg'

print（guess_file_type_by_header（file_path））

使用`mimetypes`库
`mimetypes`库可以根据文件扩展名猜测文件的MIME类型。```python
import mimetypes
def guess_file_type_by_mimetypes（file_path）:
 mime_type, _ = mimetypes.guess_type（file_path）
 return mime_type
file_path = 'example.txt'
print（guess_file_type_by_mimetypes（file_path））

以上方法各有优缺点，`filetype`库和`mimetypes`库较为简单，但可能不如通过文件头判断准确。通过文件扩展名判断的方法简单直观，但存在被恶意更改扩展名的风险。通过文件头判断的方法较为准确，但需要读取文件内容，可能不适用于大文件。

请根据具体需求选择合适的方法

-- 展开阅读全文 --