当前位置: 首页 > 文档资料 > Python 之旅 >

类 - 陌生的 metaclass

优质
小牛编辑
126浏览
2023-12-01

Python 中的元类(metaclass)是一个深度魔法,平时我们可能比较少接触到元类,本文将通过一些简单的例子来理解这个魔法。

类也是对象

在 Python 中,一切皆对象。字符串,列表,字典,函数是对象,类也是一个对象,因此你可以:

  • 把类赋值给一个变量
  • 把类作为函数参数进行传递
  • 把类作为函数的返回值
  • 在运行时动态地创建类

看一个简单的例子:

  1. class Foo(object):
  2. foo = True
  3. class Bar(object):
  4. bar = True
  5. def echo(cls):
  6. print cls
  7. def select(name):
  8. if name == 'foo':
  9. return Foo # 返回值是一个类
  10. if name == 'bar':
  11. return Bar
  12. >>> echo(Foo) # 把类作为参数传递给函数 echo
  13. <class '__main__.Foo'>
  14. >>> cls = select('foo') # 函数 select 的返回值是一个类,把它赋给变量 cls
  15. >>> cls
  16. __main__.Foo

熟悉又陌生的 type

在日常使用中,我们经常使用 object 来派生一个类,事实上,在这种情况下,Python 解释器会调用 type 来创建类。

这里,出现了 type,没错,是你知道的 type,我们经常使用它来判断一个对象的类型,比如:

  1. class Foo(object):
  2. Foo = True
  3. >>> type(10)
  4. <type 'int'>
  5. >>> type('hello')
  6. <type 'str'>
  7. >>> type(Foo())
  8. <class '__main__.Foo'>
  9. >>> type(Foo)
  10. <type 'type'>

事实上,type 除了可以返回对象的类型,它还可以被用来动态地创建类(对象)。下面,我们看几个例子,来消化一下这句话。

使用 type 来创建类(对象)的方式如下:

type(类名, 父类的元组(针对继承的情况,可以为空),包含属性和方法的字典(名称和值))

最简单的情况

假设有下面的类:

  1. class Foo(object):
  2. pass

现在,我们不使用 class 关键字来定义,而使用 type,如下:

  1. Foo = type('Foo', (object, ), {}) # 使用 type 创建了一个类对象

上面两种方式是等价的。我们看到,type 接收三个参数:

  • 第 1 个参数是字符串 ‘Foo’,表示类名
  • 第 2 个参数是元组 (object, ),表示所有的父类
  • 第 3 个参数是字典,这里是一个空字典,表示没有定义属性和方法

在上面,我们使用 type() 创建了一个名为 Foo 的类,然后把它赋给了变量 Foo,我们当然可以把它赋给其他变量,但是,此刻没必要给自己找麻烦。

接着,我们看看使用:

  1. >>> print Foo
  2. <class '__main__.Foo'>
  3. >>> print Foo()
  4. <__main__.Foo object at 0x10c34f250>

有属性和方法的情况

假设有下面的类:

  1. class Foo(object):
  2. foo = True
  3. def greet(self):
  4. print 'hello world'
  5. print self.foo

type 来创建这个类,如下:

  1. def greet(self):
  2. print 'hello world'
  3. print self.foo
  4. Foo = type('Foo', (object, ), {'foo': True, 'greet': greet})

上面两种方式的效果是一样的,看下使用:

  1. >>> f = Foo()
  2. >>> f.foo
  3. True
  4. >>> f.greet
  5. <bound method Foo.greet of <__main__.Foo object at 0x10c34f890>>
  6. >>> f.greet()
  7. hello world
  8. True

继承的情况

再来看看继承的情况,假设有如下的父类:

  1. class Base(object):
  2. pass

我们用 Base 派生一个 Foo 类,如下:

  1. class Foo(Base):
  2. foo = True

改用 type 来创建,如下:

  1. Foo = type('Foo', (Base, ), {'foo': True})

什么是元类(metaclass)

元类(metaclass)是用来创建类(对象)的可调用对象。这里的可调用对象可以是函数或者类等。但一般情况下,我们使用类作为元类。对于实例对象、类和元类,我们可以用下面的图来描述:

  1. 类是实例对象的模板,元类是类的模板
  2. +----------+ +----------+ +----------+
  3. | | | | | |
  4. | | instance of | | instance of | |
  5. | instance +------------>+ class +------------>+ metaclass|
  6. | | | | | |
  7. | | | | | |
  8. +----------+ +----------+ +----------+

我们在前面使用了 type 来创建类(对象),事实上,type 就是一个元类

那么,元类到底有什么用呢?要你何用…

元类的主要目的是为了控制类的创建行为。我们还是先来看看一些例子,以消化这句话。

元类的使用

先从一个简单的例子开始,假设有下面的类:

  1. class Foo(object):
  2. name = 'foo'
  3. def bar(self):
  4. print 'bar'

现在我们想给这个类的方法和属性名称前面加上 my_ 前缀,即 name 变成 my_name,bar 变成 my_bar,另外,我们还想加一个 echo 方法。当然,有很多种做法,这里展示用元类的做法。

1.首先,定义一个元类,按照默认习惯,类名以 Metaclass 结尾,代码如下:

  1. class PrefixMetaclass(type):
  2. def __new__(cls, name, bases, attrs):
  3. # 给所有属性和方法前面加上前缀 my_
  4. _attrs = (('my_' + name, value) for name, value in attrs.items())
  5. _attrs = dict((name, value) for name, value in _attrs) # 转化为字典
  6. _attrs['echo'] = lambda self, phrase: phrase # 增加了一个 echo 方法
  7. return type.__new__(cls, name, bases, _attrs) # 返回创建后的类

上面的代码有几个需要注意的点:

  • PrefixMetaClass 从 type 继承,这是因为 PrefixMetaclass 是用来创建类的
  • __new__ 是在 __init__ 之前被调用的特殊方法,它用来创建对象并返回创建后的对象,对它的参数解释如下:
    • cls:当前准备创建的类
    • name:类的名字
    • bases:类的父类集合
    • attrs:类的属性和方法,是一个字典

2.接着,我们需要指示 Foo 使用 PrefixMetaclass 来定制类。

在 Python2 中,我们只需在 Foo 中加一个 __metaclass__ 的属性,如下:

  1. class Foo(object):
  2. __metaclass__ = PrefixMetaclass
  3. name = 'foo'
  4. def bar(self):
  5. print 'bar'

在 Python3 中,这样做:

  1. class Foo(metaclass=PrefixMetaclass):
  2. name = 'foo'
  3. def bar(self):
  4. print 'bar'

现在,让我们看看使用:

  1. >>> f = Foo()
  2. >>> f.name # name 属性已经被改变
  3. ---------------------------------------------------------------------------
  4. AttributeError Traceback (most recent call last)
  5. <ipython-input-774-4511c8475833> in <module>()
  6. ----> 1 f.name
  7. AttributeError: 'Foo' object has no attribute 'name'
  8. >>>
  9. >>> f.my_name
  10. 'foo'
  11. >>> f.my_bar()
  12. bar
  13. >>> f.echo('hello')
  14. 'hello'

可以看到,Foo 原来的属性 name 已经变成了 my_name,而方法 bar 也变成了 my_bar,这就是元类的魔法。

再来看一个继承的例子,下面是完整的代码:

  1. class PrefixMetaclass(type):
  2. def __new__(cls, name, bases, attrs):
  3. # 给所有属性和方法前面加上前缀 my_
  4. _attrs = (('my_' + name, value) for name, value in attrs.items())
  5. _attrs = dict((name, value) for name, value in _attrs) # 转化为字典
  6. _attrs['echo'] = lambda self, phrase: phrase # 增加了一个 echo 方法
  7. return type.__new__(cls, name, bases, _attrs)
  8. class Foo(object):
  9. __metaclass__ = PrefixMetaclass # 注意跟 Python3 的写法有所区别
  10. name = 'foo'
  11. def bar(self):
  12. print 'bar'
  13. class Bar(Foo):
  14. prop = 'bar'

其中,PrefixMetaclass 和 Foo 跟前面的定义是一样的,只是新增了 Bar,它继承自 Foo。先让我们看看使用:

  1. >>> b = Bar()
  2. >>> b.prop # 发现没这个属性
  3. ---------------------------------------------------------------------------
  4. AttributeError Traceback (most recent call last)
  5. <ipython-input-778-825e0b6563ea> in <module>()
  6. ----> 1 b.prop
  7. AttributeError: 'Bar' object has no attribute 'prop'
  8. >>> b.my_prop
  9. 'bar'
  10. >>> b.my_name
  11. 'foo'
  12. >>> b.my_bar()
  13. bar
  14. >>> b.echo('hello')
  15. 'hello'

我们发现,Bar 没有 prop 这个属性,但是有 my_prop 这个属性,这是为什么呢?

原来,当我们定义 class Bar(Foo) 时,Python 会首先在当前类,即 Bar 中寻找 __metaclass__,如果没有找到,就会在父类 Foo 中寻找 __metaclass__,如果找不到,就继续在 Foo 的父类寻找,如此继续下去,如果在任何父类都找不到 __metaclass__,就会到模块层次中寻找,如果还是找不到,就会用 type 来创建这个类。

这里,我们在 Foo 找到了 __metaclass__,Python 会使用 PrefixMetaclass 来创建 Bar,也就是说,元类会隐式地继承到子类,虽然没有显示地在子类使用 __metaclass__,这也解释了为什么 Bar 的 prop 属性被动态修改成了 my_prop。

写到这里,不知道你理解元类了没?希望理解了,如果没理解,就多看几遍吧~

小结

  • 在 Python 中,类也是一个对象。
  • 类创建实例,元类创建类。
  • 元类主要做了三件事:
    • 拦截类的创建
    • 修改类的定义
    • 返回修改后的类
  • 当你创建类时,解释器会调用元类来生成它,定义一个继承自 object 的普通类意味着调用 type 来创建它。

参考资料