Python中dataclass的方法继承问题求助

你的代码核心问题在于：get_columns作为类方法直接访问cls.col_converters，但col_converters是dataclass的实例字段，并非类属性——类本身不会持有该字段的值，只有创建类的实例时才会初始化；再加上子类用field(default_factory)定义默认值，这个值是在实例化阶段才生成的，类层面无法直接读取。

以下是几种可行的解决方法：

方案一：将默认转换器定义为类属性（推荐） #

把子类的默认col_converters提取为类属性，dataclass字段的默认值指向这个类属性，类方法直接引用类属性即可：

from dataclasses import dataclass, field
import polars as pl

@dataclass(frozen=True)
class IngestionMetadata:
    file: str
    sheet_name: str
    col_converters: dict
    col_names: list = field(init=False)
    
    def __post_init__(self):
        object.__setattr__(self, 'col_names', list(self.col_converters.keys()))
    
    @classmethod
    def get_columns(cls):
        if hasattr(cls, '_default_col_converters'):
            return list(cls._default_col_converters.keys())
        raise NotImplementedError("当前类未定义默认col_converters")


@dataclass(frozen=True)
class ProductGroupLookup(IngestionMetadata):
    # 类属性存储默认转换器
    _default_col_converters = {
        'Product 1': pl.String,
        'Product 2': pl.String,
        'Product 3': pl.String,
        'Product Category Code 4': pl.String
    }
    
    file: str = 'product_group_table.xlsx'
    sheet_name: str = 'INGESTION'
    col_converters: dict = field(default_factory=lambda: cls._default_col_converters.copy())

这样调用im.ProductGroupLookup.get_columns()就能直接返回列名列表，同时用copy()避免多个实例共享同一个字典的问题。

方案二：类方法中创建临时实例获取值 #

由于子类已为所有字段设置默认值，可以在类方法里创建临时实例，从实例中读取col_converters：

from dataclasses import dataclass, field
import polars as pl

@dataclass(frozen=True)
class IngestionMetadata:
    file: str
    sheet_name: str
    col_converters: dict
    col_names: list = field(init=False)
    
    def __post_init__(self):
        object.__setattr__(self, 'col_names', list(self.col_converters.keys()))
    
    @classmethod
    def get_columns(cls):
        # 创建临时实例（子类已设全默认值，无需传参）
        temp_instance = cls()
        return temp_instance.col_names


@dataclass(frozen=True)
class ProductGroupLookup(IngestionMetadata):
    file: str = 'product_group_table.xlsx'
    sheet_name: str = 'INGESTION'
    col_converters: dict = field(default_factory= lambda: {
        'Product 1': pl.String,
        'Product 2': pl.String,
        'Product 3': pl.String,
        'Product Category Code 4': pl.String
    })

该方案无需修改原有字段定义，但每次调用都会创建实例，频繁调用会有微小性能损耗。

方案三：通过dataclasses API读取默认工厂 #

利用dataclasses.fields获取字段的默认工厂，调用工厂函数生成默认值：

from dataclasses import dataclass, field, fields
import polars as pl

@dataclass(frozen=True)
class IngestionMetadata:
    file: str
    sheet_name: str
    col_converters: dict
    col_names: list = field(init=False)
    
    def __post_init__(self):
        object.__setattr__(self, 'col_names', list(self.col_converters.keys()))
    
    @classmethod
    def get_columns(cls):
        for f in fields(cls):
            if f.name == 'col_converters':
                default_converters = f.default_factory()
                return list(default_converters.keys())
        raise ValueError("类中未定义col_converters字段")


@dataclass(frozen=True)
class ProductGroupLookup(IngestionMetadata):
    file: str = 'product_group_table.xlsx'
    sheet_name: str = 'INGESTION'
    col_converters: dict = field(default_factory= lambda: {
        'Product 1': pl.String,
        'Product 2': pl.String,
        'Product 3': pl.String,
        'Product Category Code 4': pl.String
    })

此方案直接通过dataclass元数据获取默认值，无需创建实例，但需确保子类的col_converters确实使用了default_factory。

内容的提问来源于stack exchange，提问作者matt.aurelio