用于ORM的python枚举类

基于更新的新答案

我认为这满足了你所有的特定要求。如果没有,我们可能可以添加您需要的任何内容。

def enum(classname, values):
    class EnumMeta(type):
        def __iter__(cls):
            return cls._instances.itervalues()

    class EnumType(object):
        __metaclass__ = EnumMeta
        _instances = {}
        _next_id = 0
        def __init__(self, value):
            self.value = value
            self.id = type(self)._next_id
            type(self)._next_id += 1

        def instance(self, value):
            return type(self)._instances[value]

    cls = type(classname, (EnumType, ), {})
    instances = dict((value, cls(value)) for value in values)
    cls._instances = instances

    def __new__(cls, value):
        raise TypeError('No more instances allowed')

    cls.__new__ = staticmethod(__new__)
    return cls


Genre = enum('Genre', ['scifi', 'comic', 'science'])


for item in Genre:
    print item, item.value, item.id
    assert(item is Genre(item.value))
    assert(item is item.instance(item.value))

Genre('romance')

旧答案

为了回应你对诺蒂斯天空塔的回答,你说你想 Genre.comic = Genre('comic') (未经测试):

class Genre(GenreBase):
    genres = ['comic', 'scifi', ... ]
    def __getattr__(self, attr):
        if attr in type(self).genres:
            self.__dict__[attr] = type(self)(attr)
        return self.__dict__[attr]

这将创建一个流派实例以响应对其进行访问的尝试,并将其附加到请求它的实例。如果要将它附加到整个类,请替换该行

self.__dict__[attr] == type(self)(attr) 

具有

type(self).__dict__[attr] = type(self)(attr)

它还具有所有子类创建子类的实例以响应请求。如果希望子类创建 Genre 替换 type(self)(attr) 具有 Genre(attr)

您可以使用 type 建筑:

类型(名称、基、dict)

返回新类型对象。这是 本质上是 类语句。名称字符串是 类名称并成为 __name__ 属性;基元组 将基类逐项化并成为 这个 __bases__ 属性;以及dict 字典是包含 类体的定义并成为 这个 __dict__ 属性。例如, 以下两个语句创建 相同类型的对象:

>>> class X(object):
...     a = 1
...
>>> X = type('X', (object,), dict(a=1))

New in version 2.2.

在这种情况下:

genre_mapping = { }
for genre in { 'scifi', 'romance', 'comic', 'science' }:
    genre_mapping[ 'genre' ] = type( genre, ( Genre, ), { } )

或者在python 2.7+中:

genre_mapping = { genre: type( genre, ( Genre, ), { } ) for genre in genres }

如果你经常这样做,你可以把模式抽象掉。

>>> def enum( cls, subs ):
...     return { sub: type( sub, ( cls, ), { } ) for sub in subs }
...
>>> enum( Genre, [ 'scifi', 'romance', 'comic', 'science' ] )
{'romance': <class '__main__.romance'>, 'science': <class '__main__.science'>,
'comic': <class '__main__.comic'>, 'scifi': <class '__main__.scifi'>}

编辑:我错过要点了吗?(我以前没有用过sqlacalchemy。)你在问如何创建新的 子类 属于 Genre 或如何创建新的 实例 ?前者在直觉上似乎是正确的,但后者正是你所要求的。这很容易:

list( map( Genre, [ 'scifi', ... ] ) )

将为您列出:

[ Genre( 'scifi' ), ... ]

也许这个来自Verse Quiz程序的枚举函数对您有一定的用处: Verse Quiz