Python 判断文件类型|极客教程

Python 判断文件类型

在进行文件处理时，有时候我们需要判断文件的类型，以便进行相应的操作。在Python中，我们可以通过文件的扩展名或者文件的内容来判断文件的类型。本文将详细介绍如何使用Python来判断文件类型。

通过文件扩展名判断文件类型

我们先来看看如何通过文件的扩展名来判断文件类型。在Unix/Linux系统中，文件的扩展名通常位于文件名的最后一个句点（.）之后。我们可以通过Python的os模块来获取文件的扩展名，并根据扩展名来判断文件类型。

import os

def get_file_extension(file_path):
    return os.path.splitext(file_path)[1]

file_path = 'example.txt'
file_extension = get_file_extension(file_path)

print('File extension: ', file_extension)

if file_extension == '.txt':
    print('Text file')
elif file_extension == '.jpg' or file_extension == '.jpeg' or file_extension == '.png':
    print('Image file')
elif file_extension == '.pdf':
    print('PDF file')
else:
    print('Unknown file type')

运行结果：

File extension:  .txt
Text file

通过文件扩展名判断文件类型虽然简单直观，但是可能存在一些问题。比如有些文件没有扩展名，或者扩展名被修改了。因此，我们也可以通过判断文件的内容来确定文件类型。

通过文件内容判断文件类型

通过文件内容判断文件类型，常见的方法是读取文件的前几个字节，根据文件的特征判断文件类型。下面是一个通过文件内容判断文件类型的示例代码：

def get_file_type(file_path):
    with open(file_path, 'rb') as file:
        header = file.read(4)

    if header[0:2] == b'\xff\xd8':
        return 'JPEG'
    elif header[0:4] == b'\x89PNG':
        return 'PNG'
    elif header[0:4] == b'%PDF':
        return 'PDF'
    else:
        return 'Unknown'

file_path = 'example.jpg'
file_type = get_file_type(file_path)

print('File type: ', file_type)

运行结果：