Python中使用装饰器和元编程实现结构体类实例

2019-11-25 18:01:17

字体：大中小

来源：转载

供稿：网友

Ruby中有一个很方便的Struct类，用来实现结构体。这样就不用费力的去定义一个完整的类来仅仅用作访问属性。

复制代码代码如下:

class Dog < Struct.new(:name, :age)
end

fred = Dog.new("fred", 5)
printf "name:%s age:%d", fred.name, fred.age
##name:fred age:5

Python3.4中也可以这么干，但写法很累赘。其中包含self.name = name 这种很烦人的写法。

复制代码代码如下:

class Dog(object):
    def __init__(self, name, age):
        self.name = name
        self.age = age

fred = Dog("fred", 5)
print('name:{name} age:{age}'.format(name=fred.name, age=fred.age))
##name:fred age:5

想到我大Python是无所不能的，有没有一种简化结构体类属性定义的方法呢?答案肯定是有的。在补习了一些Python黑魔法技术后，我想到利用装饰器函数和元编程技术来实现。

复制代码代码如下:

def struct(*name):
    """ 装饰器函数
        用途：用于在类定义中，自动设置self.value = value
    """
    def decorator(func):
        def wrapper(*args, **kw):
            for i in range(len(name)):
                setattr(args[0], name[i], args[i+1])
            return func(*args, **kw)
        return wrapper
    return decorator

class Dog(object):
    @struct('name','age')   #黑魔法所在!
    def __init__(self, *all_value):
        pass

fred = Dog("fred", 5)
print('name:{name} age:{age}'.format(name=fred.name, age=fred.age))
##name:fred age:5

要注意的是，这种写法会造成代码结构的不清晰。

上一篇：Python中集合类型（set）学习小结

下一篇：Python实现Const详解