2012年6月30日土曜日

動的に属性を変更可能なクラス (その2)

前回の内容では field を変更したときに値がすべて削除された。
できれば field が a, b, cから b, c, dに変更されたときにはb, c の値を保存しておきたいのでプログラムを修正。

def Dynamic(name, fields):
    """ returns a class object which has fields attribute.
    the class is iterable."""
    dic = dict([x,None] for x in fields)
    o = type(name, (_dynamic,), dic)
    o._fields = fields
    return o
class _dynamic(object):
    _fields = []

    def __init__(self, *args, **kwargs):
        for i,value in enumerate(args[0]):
            setattr(self, self._fields[i], value)

    def __iter__(self):
        for field in self._fields:
            yield getattr(self, field)

    def __setattr__(self, name, value):
        if name == "_fields":
            for field in set(self._fields) - set(value):
                delattr(self, field)
            for field in set(value) - set(self._fields):
                super(_dynamic, self).__setattr__(field, None)
        super(_dynamic, self).__setattr__(name, value)

0 件のコメント: