如何减少重复创建CSV DictWriter的代码并确保文件正确关闭？

你之前尝试直接把DictWriter绑定为类属性的方式确实会导致文件无法自动关闭，因为open()返回的文件对象没有被上下文管理器接管，容易造成资源泄漏。下面是几种既能减少重复代码又能保证文件自动关闭的可行方案：

方法1：类内封装上下文管理器，复用DictWriter配置 #

在类中定义一个私有辅助方法，用contextmanager装饰器把它变成上下文管理器，统一处理文件打开和DictWriter的配置。这样每次写入时只需要传入打开模式，不用重复写DictWriter的参数：

import csv
from contextlib import contextmanager

class Example:
    # 提前定义好字段名和文件路径（可以是类属性或实例属性）
    fieldnames = ['col1', 'col2', 'col3']
    _file_path = 'filepath'

    @contextmanager
    def _get_csv_writer(self, mode):
        # 内层with保证文件自动关闭
        with open(self._file_path, mode, newline='') as file:
            writer = csv.DictWriter(file, fieldnames=self.fieldnames, lineterminator='\n')
            yield writer

    def func1(self):
        with self._get_csv_writer('w') as csv_writer:
            csv_writer.writeheader()
            csv_writer.writerow({'col1': 'val1', 'col2': 'val2', 'col3': 'val3'})
            # 其他写入逻辑

    def func2(self):
        with self._get_csv_writer('a') as csv_writer:
            csv_writer.writerow({'col1': 'val4', 'col2': 'val5', 'col3': 'val6'})
            # 其他追加逻辑

方法2：抽离独立工具类，跨场景复用 #

如果这个CSV写入逻辑需要在多个类中使用，可以把它抽成一个独立的工具类，专门处理文件管理和DictWriter配置：

import csv
from contextlib import contextmanager

class CSVWriterHelper:
    def __init__(self, file_path, fieldnames):
        self.file_path = file_path
        self.fieldnames = fieldnames

    @contextmanager
    def get_writer(self, mode):
        with open(self.file_path, mode, newline='') as file:
            writer = csv.DictWriter(file, fieldnames=self.fieldnames, lineterminator='\n')
            yield writer

# 原类使用工具类
class Example:
    def __init__(self):
        self.csv_helper = CSVWriterHelper('filepath', ['col1', 'col2', 'col3'])

    def func1(self):
        with self.csv_helper.get_writer('w') as writer:
            writer.writeheader()
            writer.writerow({'col1': 'val1'})

    def func2(self):
        with self.csv_helper.get_writer('a') as writer:
            writer.writerow({'col1': 'val4'})

方法3：存储配置参数，按需创建Writer #

如果不想用上下文管理器装饰器，也可以把DictWriter的配置参数存在实例属性中，每次写入时直接展开参数创建Writer，依然用with保证文件关闭：

import csv

class Example:
    def __init__(self, file_path, fieldnames):
        self.file_path = file_path
        # 把重复的配置存在字典里
        self._writer_config = {
            'fieldnames': fieldnames,
            'lineterminator': '\n'
        }

    def func1(self):
        with open(self.file_path, 'w', newline='') as file:
            csv_writer = csv.DictWriter(file, **self._writer_config)
            csv_writer.writeheader()
            # 写入逻辑

    def func2(self):
        with open(self.file_path, 'a', newline='') as file:
            csv_writer = csv.DictWriter(file, **self._writer_config)
            # 追加逻辑

内容的提问来源于stack exchange，提问作者yFoxy _