Python 3.x中处理.txt与.gz文件的Pythonic风格文件打开方案咨询

这问题问得好！处理这种多类型文件但后续逻辑一致的场景，Python里确实有更优雅的方式，完全不用写重复函数或者堆一堆if判断。

核心思路：封装文件打开逻辑，统一入口 #

把不同后缀的文件打开逻辑封装成一个单独的函数，让主处理逻辑只需要和这个统一入口打交道，完全不用关心底层是用open()还是gzip.open()。

方法一：简洁的条件判断封装

这种方式简单直接，适合类型不多的场景：

import gzip

def open_file(filename):
    if filename.endswith('.gz'):
        return gzip.open(filename, 'rt')  # 'rt'以文本模式读取gz文件
    elif filename.endswith('.txt'):
        return open(filename, 'r')
    else:
        raise ValueError(f"不支持的文件类型：{filename}")

# 统一的处理函数
def process_file(filename):
    with open_file(filename) as f:
        # 这里写所有文件都通用的处理逻辑，比如逐行读取处理
        for line in f:
            processed_line = line.strip().upper()
            print(processed_line)

方法二：字典映射（更易扩展）

如果以后可能要支持更多文件类型（比如.bz2），用字典映射后缀和对应的打开函数会更灵活，符合开闭原则：

import gzip

# 后缀到打开函数的映射
FILE_OPENERS = {
    '.txt': lambda fname: open(fname, 'r'),
    '.gz': lambda fname: gzip.open(fname, 'rt')
}

def open_file(filename):
    for ext, opener in FILE_OPENERS.items():
        if filename.endswith(ext):
            return opener(filename)
    raise ValueError(f"不支持的文件类型：{filename}")

# 同样的统一处理逻辑
def process_file(filename):
    with open_file(filename) as f:
        # 通用处理代码
        pass

为什么这是Pythonic的？ #

• 遵循**DRY（Don't Repeat Yourself）**原则：完全避免了重复写处理逻辑
• 关注点分离：文件打开的逻辑和业务处理逻辑分开，代码更清晰
• 利用了Python的上下文管理器（with语句）：自动处理文件关闭，不用手动管close()
• 扩展性强：要加新的文件类型，只需要修改open_file函数或者字典，主处理逻辑完全不用动

内容的提问来源于stack exchange，提问作者Denis Rasulev