如何强制所有BudgetCategories派生类在__init__中执行指定校验函数？

要确保所有继承BudgetCategories的派生类在对象创建阶段执行唯一性校验，最可靠的方式是把校验逻辑整合到基类的初始化流程中，同时通过抽象方法约束派生类必须实现分类数据的初始化逻辑。这样派生类无需手动调用校验方法，基类会自动完成校验，从根源上避免遗漏。

修改后的代码实现 #

from abc import ABCMeta, abstractmethod
import csv
from collections import defaultdict

class BudgetCategories(metaclass=ABCMeta):
    def __init__(self) -> None:
        self._budget_categories: list[str] = []
        # 触发派生类的分类初始化逻辑
        self._setup_budget_categories()
        # 自动执行唯一性校验
        self._check_budget_categories_are_unique()

    @abstractmethod
    def _setup_budget_categories(self) -> None:
        """派生类必须实现此方法，用于填充预算分类列表"""
        pass

    def _check_budget_categories_are_unique(self) -> None:
        # 优化重复检测逻辑，时间复杂度从O(n²)降至O(n)
        category_counts = defaultdict(int)
        duplicates = set()
        for category in self._budget_categories:
            lower_cat = category.lower()
            category_counts[lower_cat] += 1
            if category_counts[lower_cat] > 1:
                duplicates.add(lower_cat)
        
        if duplicates:
            raise ValueError(
                f"预算分类必须唯一。重复项：{duplicates}"
            )

    @property
    def budget_categories(self) -> list[str]:
        # 返回列表副本，防止外部直接修改内部数据
        return self._budget_categories.copy()

class BudgetCategoriesFromFile(BudgetCategories):
    def __init__(self, path_to_budget_categories_csv_file: str) -> None:
        self._csv_path = path_to_budget_categories_csv_file
        # 调用基类初始化，自动触发后续流程
        super().__init__()

    def _setup_budget_categories(self) -> None:
        with open(self._csv_path, "r") as f:
            reader = csv.reader(f)
            for row in reader:
                # 过滤空行和空字符串，避免无效分类
                if row and row[0].strip():
                    self._budget_categories.append(row[0].strip())

方案核心优势 #

• 强制性约束：抽象方法_setup_budget_categories强制所有派生类必须实现分类初始化逻辑，否则无法实例化对象。
• 流程自动化：基类__init__统一调度初始化和校验步骤，派生类无需手动调用校验方法，确保校验在对象创建阶段执行。
• 职责清晰：基类负责核心流程和通用校验逻辑，派生类只需专注于自身的数据源处理（如从文件读取），符合单一职责原则。

额外代码优化建议 #

1. 提升重复检测效率：原代码用list.count()遍历检测重复，时间复杂度为O(n²)，改用字典统计次数后效率大幅提升，尤其适合分类数量较多的场景。
2. 增强封装性：budget_categories属性返回列表副本，避免外部代码直接修改内部数据，破坏对象状态的一致性。
3. 过滤无效数据：读取CSV时增加空行和空字符串过滤，避免将无效的空分类加入列表，减少后续校验的无效逻辑。
4. 类型提示完善：所有方法和属性都添加清晰的类型提示，提升代码可读性和IDE的自动补全支持。

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