字段的python装饰器

Question

我知道没有这样的事情。这就是为什么我正在寻找一些不错的等价物。有这个类:

class MyClass:
   a = 5
   b  = "foo"
   c = False

我想将字段 a 和 b 组合在一起，以便能够以某种方式仅迭代来自该组的成员。所以最好有某种字段装饰器，比如:

class MyClass:
   @bar
   a = 5
   @bar
   b  = "foo"
   c = False

还有一些像 myDir(MyClass, 'bar') 这样的函数会返回 ('a', 'b')。

到目前为止我们有哪些选择？ 1. 按照特殊约定命名这些字段，例如“a_bar”、“b_bar”——但不幸的是我不能这样做。 2. 将姓名列表作为另一个类(class)成员——我想让这个分组属性接近属性，所以我不喜欢这种方法。 3. 我可以创建继承自整数和字符串的类，并添加另一个基类(如“Group”)，然后检查类型，而不是将 5 分配给“a”和“foo”——这样我将不得不生成此类每次我都有新的类型，所以我也不喜欢这个解决方案。

还有其他想法吗？

Answer 1

装饰器只是函数调用的语法糖，所以要将装饰器应用于属性，您只需在初始化器上调用它:

a = bar(5)

在您的情况下，您可以创建实现描述符协议(protocol)的对象:

class GroupedAttribute(object):
    def __init__(self, group, obj):
        self.group = group
        self.obj = obj
    def __get__(self, obj, owner):
        return self.obj
    def __set__(self, obj, value):
        self.obj = value

为了优雅，您可以为属性组编写一个类:

class GroupedAttribute(object):
    def __init__(self, group, obj):
        self.group = group
        self.obj = obj
    def __get__(self, obj, owner):
        return self.obj
    def __set__(self, obj, value):
        self.obj = value

class AttributeGroup(object):
    def __call__(self, obj):
        return GroupedAttribute(self, obj)
    def __get__(self, obj, owner):
        return BoundAttributeGroup(self, obj, owner)

class BoundAttributeGroup(object):
    def __init__(self, group, obj, owner):
        self.group = group
        self.obj = obj
        self.owner = owner
    def __dir__(self):
        items = dir(self.owner if self.obj is None else self.obj)
        return [item for item in items if
                getattr(self.owner.__dict__.get(item, None),
                        'group', None) is self.group]

用法:

class MyClass(object):
    bar = AttributeGroup()
    a = bar(5)
    b = bar("foo")
    c = False
dir(MyClass.bar)    # ['a', 'b']