0

我在 Python 中有一些代码，其中有一堆类，每个类都有一个属性_internal_attribute。我希望能够生成这些属性到原始类的映射。基本上我希望能够做到这一点：

class A(object):
    _internal_attribute = 'A attribute'
class B(object):
    _internal_attribute = 'B attribute'

a_instance = magic_reverse_mapping['A attribute']()
b_instance = magic_reverse_mapping['B attribute']()

我在这里缺少的是如何生成magic_reverse_mapping字典。我有一种直觉，让元类生成 A 和 B 是解决此问题的正确方法；这看起来对吗？

4

2 回答 2

5

您可以使用元类在以下位置自动注册您的类magic_reverse_mapping：

magic_reverse_mapping = {}

class MagicRegister(type):
   def __new__(meta, name, bases, dict):
      cls = type.__new__(meta, name, bases, dict)
      magic_reverse_mapping[dict['_internal_attribute']] = cls
      return cls

class A(object):
    __metaclass__ = MagicRegister
    _internal_attribute = 'A attribute'

afoo = magic_reverse_mapping['A attribute']()

或者，您可以在您的类上使用装饰器来注册它们。我认为这更具可读性和更容易理解：

magic_reverse_mapping = {}

def magic_register(cls):
   magic_reverse_mapping[cls._internal_attribute] = cls
   return cls

@magic_register
class A(object):
    _internal_attribute = 'A attribute'

afoo = magic_reverse_mapping['A attribute']()

或者你甚至可以手工完成。在不使用任何魔法的情况下，这并不是更多的工作：

reverse_mapping = {}

class A(object):
    _internal_attribute = 'A attribute'

reverse_mapping[A._internal_attribute] = A

查看不同的变体，我认为装饰器版本将是最令人愉快的使用。

于 2009-08-12T00:02:52.283 回答
3

您首先需要一些数据结构来存储适用类的列表，但您不必首先生成它。您可以改为从全局变量中读取类。这自然假设您的类扩展了object，就像您在第一篇文章中所做的那样。

def magic_reverse_mapping(attribute_name, attribute_value):
    classobjects = [val for val in globals().values() if isinstance(val, object)]
    attrobjects = [cls for cls in classobjects if hasattr(cls, attribute_name)]
    resultobjects = [cls for cls in attrobjects if object.__getattribute__(cls, attribute_name) == attribute_value]
    return resultobjects

magic_reverse_mapping('_internal_attribute', 'A attribute')
#output:   [<class '__main__.A'>]

请注意，这将返回具有该属性值的类列表，因为可能不止一个。如果你想实例化第一个：

magic_reverse_mapping('_internal_attribute', 'A attribute')[0]()
#output:   <__main__.A object at 0xb7ce486c>

与某事的答案不同，您不必为您的课程添加装饰器（不过，这是一个简洁的解决方案）。但是，无法排除全局命名空间中的任何类。

于 2009-08-11T23:21:40.807 回答