我怎么知道哪个基类为子类对象添加了特定的属性

时间:2018-11-22 09:51:45

标签: python python-3.x multiple-inheritance

我正在研究一个项目,该项目具有来自不同文件中几个模块的长类层次结构。

我想知道何时在继承链中类C获得属性A (然后我可以得到定义了C的模块M并检查代码)

考虑以下代码示例,假设除GrandChildOf1ChildOf2以外的所有类都在其他模块中,是否有命令,例如:attribute_base(o4,'a'),输出为:Base1

class SweetDir:
    def __sweet_dir__(self):
        """
        Same as dir, but will omit special attributes
        :return: string
        """
        full_dir = self.__dir__()
        sweet_dir  = []
        for attribute_name in full_dir:

            if not (    attribute_name.startswith('__')
                    and  attribute_name.endswith('__')):
                #not a special attribute
                sweet_dir.append(attribute_name)
        return sweet_dir

class Base1(SweetDir):
    def __init__(self):
        super(Base1,self).__init__()
        self.a = 'a'

class Base2(SweetDir):
    def __init__(self):
        super(Base2,self).__init__()
        self.b = 'b'

class ChildOf1 (Base1):
    def __init__(self):
        super(ChildOf1,self).__init__()
        self.c = 'c'

class GrandChildOf1ChildOf2 (Base2,ChildOf1):
    def __init__(self):
        super(GrandChildOf1ChildOf2,self).__init__()
        self.d = 'd'   


o1 = Base1()
o2 = Base2()
o3 = ChildOf1()
o4 = GrandChildOf1ChildOf2()
print(o1.__sweet_dir__())
print(o2.__sweet_dir__())
print(o3.__sweet_dir__())
print(o4.__sweet_dir__())

输出:

['a']
['b']
['a', 'c']
['a', 'b', 'c', 'd']

1 个答案:

答案 0 :(得分:1)

我认为没有内置函数,但是类似的东西会起作用(需要改进):

def attribute_base(your_class, your_attr):

    for base_class in your_class.__mro__:
        if base_class != your_class:
            tmp_inst = base_class()
            if hasattr(tmp_inst, your_attr):
                return base_class

这将返回类的第一个基类,该基类具有您要查找的属性。这显然不是完美的。如果您的两个或多个基类具有相同的属性(具有相同的名称),则它可能不会返回您获得该属性的实际类,但是在您的示例中它将起作用。 [更新为AKX注释:使用__mro__实际上应该可以解决此问题]

[更新:有一种方法可以在没有实例的情况下执行此操作,它遵循以下详细记录的答案:list-the-attributes-of-a-class-without-instantiating-an-object]

from inspect import getmembers

def attribute_base(your_class, your_attr):
    for base_class in your_class.__mro__:
        if base_class != your_class:
            members = [member[1].__code__.co_names for member in getmembers(base_class) if '__init__' in member and hasattr(member[1], "__code__")]
            for member in members:
                if your_attr in members:
                    return base_class

getmembers为您提供了类的每个成员,包括我们想要的init方法。我们需要检查它是否真的是一个函数(hasattr(member[1], "__code__")),因为如果没有为一个类定义__init__函数(例如您的示例中的SweetDir),这将返回wrapper_descriptor。在极少数情况下(可能吗?),我们会循环访问成员。有几种__init__方法。