問(wèn)題

你想自動(dòng)記錄一個(gè)類中屬性和方法定義的順序，然后可以利用它來(lái)做很多操作（比如序列化、映射到數(shù)據(jù)庫(kù)等等）。

解決方案

利用元類可以很容易的捕獲類的定義信息。下面是一個(gè)例子，使用了一個(gè)OrderedDict來(lái)記錄描述器的定義順序：

from collections import OrderedDict

# A set of descriptors for various types
class Typed:
    _expected_type = type(None)
    def __init__(self, name=None):
        self._name = name

    def __set__(self, instance, value):
        if not isinstance(value, self._expected_type):
            raise TypeError('Expected ' + str(self._expected_type))
        instance.__dict__[self._name] = value

class Integer(Typed):
    _expected_type = int

class Float(Typed):
    _expected_type = float

class String(Typed):
    _expected_type = str

# Metaclass that uses an OrderedDict for class body
class OrderedMeta(type):
    def __new__(cls, clsname, bases, clsdict):
        d = dict(clsdict)
        order = []
        for name, value in clsdict.items():
            if isinstance(value, Typed):
                value._name = name
                order.append(name)
        d['_order'] = order
        return type.__new__(cls, clsname, bases, d)

    @classmethod
    def __prepare__(cls, clsname, bases):
        return OrderedDict()

在這個(gè)元類中，執(zhí)行類主體時(shí)描述器的定義順序會(huì)被一個(gè) OrderedDict``捕獲到，生成的有序名稱從字典中提取出來(lái)并放入類屬性 ``_order 中。這樣的話類中的方法可以通過(guò)多種方式來(lái)使用它。例如，下面是一個(gè)簡(jiǎn)單的類，使用這個(gè)排序字典來(lái)實(shí)現(xiàn)將一個(gè)類實(shí)例的數(shù)據(jù)序列化為一行CSV數(shù)據(jù)：

class Structure(metaclass=OrderedMeta):
    def as_csv(self):
        return ','.join(str(getattr(self,name)) for name in self._order)

# Example use
class Stock(Structure):
    name = String()
    shares = Integer()
    price = Float()

    def __init__(self, name, shares, price):
        self.name = name
        self.shares = shares
        self.price = price

我們?cè)诮换ナ江h(huán)境中測(cè)試一下這個(gè)Stock類：

>>> s = Stock('GOOG',100,490.1)
>>> s.name
'GOOG'
>>> s.as_csv()
'GOOG,100,490.1'
>>> t = Stock('AAPL','a lot', 610.23)
Traceback (most recent call last):
    File "<stdin>", line 1, in <module>
    File "dupmethod.py", line 34, in __init__
TypeError: shares expects <class 'int'>
>>>

討論

本節(jié)一個(gè)關(guān)鍵點(diǎn)就是OrderedMeta元類中定義的 __prepare__() 方法。這個(gè)方法會(huì)在開(kāi)始定義類和它的父類的時(shí)候被執(zhí)行。它必須返回一個(gè)映射對(duì)象以便在類定義體中被使用到。我們這里通過(guò)返回了一個(gè)OrderedDict而不是一個(gè)普通的字典，可以很容易的捕獲定義的順序。

如果你想構(gòu)造自己的類字典對(duì)象，可以很容易的擴(kuò)展這個(gè)功能。比如，下面的這個(gè)修改方案可以防止重復(fù)的定義：

from collections import OrderedDict

class NoDupOrderedDict(OrderedDict):
    def __init__(self, clsname):
        self.clsname = clsname
        super().__init__()
    def __setitem__(self, name, value):
        if name in self:
            raise TypeError('{} already defined in {}'.format(name, self.clsname))
        super().__setitem__(name, value)

class OrderedMeta(type):
    def __new__(cls, clsname, bases, clsdict):
        d = dict(clsdict)
        d['_order'] = [name for name in clsdict if name[0] != '_']
        return type.__new__(cls, clsname, bases, d)

    @classmethod
    def __prepare__(cls, clsname, bases):
        return NoDupOrderedDict(clsname)

下面我們測(cè)試重復(fù)的定義會(huì)出現(xiàn)什么情況：

>>> class A(metaclass=OrderedMeta):
... def spam(self):
... pass
... def spam(self):
... pass
...
Traceback (most recent call last):
    File "<stdin>", line 1, in <module>
    File "<stdin>", line 4, in A
    File "dupmethod2.py", line 25, in __setitem__
        (name, self.clsname))
TypeError: spam already defined in A
>>>

最后還有一點(diǎn)很重要，就是在 __new__() 方法中對(duì)于元類中被修改字典的處理。盡管類使用了另外一個(gè)字典來(lái)定義，在構(gòu)造最終的 class 對(duì)象的時(shí)候，我們?nèi)匀恍枰獙⑦@個(gè)字典轉(zhuǎn)換為一個(gè)正確的 dict 實(shí)例。通過(guò)語(yǔ)句 d = dict(clsdict) 來(lái)完成這個(gè)效果。

對(duì)于很多應(yīng)用程序而已，能夠捕獲類定義的順序是一個(gè)看似不起眼卻又非常重要的特性。例如，在對(duì)象關(guān)系映射中，我們通常會(huì)看到下面這種方式定義的類：

class Stock(Model):
    name = String()
    shares = Integer()
    price = Float()

在框架底層，我們必須捕獲定義的順序來(lái)將對(duì)象映射到元組或數(shù)據(jù)庫(kù)表中的行（就類似于上面例子中的 as_csv() 的功能）。這節(jié)演示的技術(shù)非常簡(jiǎn)單，并且通常會(huì)比其他類似方法（通常都要在描述器類中維護(hù)一個(gè)隱藏的計(jì)數(shù)器）要簡(jiǎn)單的多。