読者です 読者をやめる 読者になる 読者になる

何かを書き留める何か

数学や読んだ本について書く何かです。最近は社会人として生き残りの術を学ぶ日々です。

Python 3.6における『Effective Python』 項目34はこう変わる

Python Effective Python

項目34はかなりシンプルになる

Python 3.6がリリースされた。 Python 3.6で導入された新機能の一つに__init_subclass__がある、というのは前回の流れと同じ。

xaro.hatenablog.jp

クラスが定義された際に自動で何らかの辞書に登録したいという場面でメタクラスを活用すればできるよというのが項目34の主題である。

import json

class BetterSerializable(object):
    def __init__(self, *args):
        self.args = args

    def serialize(self):
        return json.dumps({
            'class': self.__class__.__name__,
            'args': self.args,
        })

    def __repr__(self):
        return '%s(%s)' % (
            self.__class__.__name__,
            ', '.join(str(x) for x in self.args))

registry = {}

def register_class(target_class):
    registry[target_class.__name__] = target_class

def deserialize(data):
    params = json.loads(data)
    name = params['class']
    target_class = registry[name]
    return target_class(*params['args'])


class Meta(type):
    def __new__(meta, name, bases, class_dict):
        cls = type.__new__(meta, name, bases, class_dict)
        register_class(cls)
        return cls


class RegisteredSerializable(BetterSerializable, metaclass=Meta):
    pass


class Meta(type):
    def __new__(meta, name, bases, class_dict):
        cls = type.__new__(meta, name, bases, class_dict)
        register_class(cls)
        return cls


class RegisteredSerializable(BetterSerializable, metaclass=Meta):
    pass


class Vector3D(RegisteredSerializable):
    def __init__(self, x, y, z):
        super().__init__(x, y, z)
        self.x, self.y, self.z = x, y, z

if __name__ == '__main__':
    
    v3 = Vector3D(10, -7, 3)
    print('Before:    ', v3)
    data = v3.serialize()
    print('Serialized:', data)
    print('After:     ', deserialize(data))

これを__init_subclass__を使うとかなりシンプルに書くことができる。 具体的には、メタクラスが不要となり、直接BetterSerializableに記述することができる。

import json


class BetterSerializable(object):
    def __init__(self, *args):
        self.args = args

    def __init_subclass__(cls, **kwargs):
        register_class(cls)

    def serialize(self):
        return json.dumps({
            'class': self.__class__.__name__,
            'args': self.args,
        })

    def __repr__(self):
        return '%s(%s)' % (
            self.__class__.__name__,
            ', '.join(str(x) for x in self.args))

registry = {}

def register_class(target_class):
    registry[target_class.__name__] = target_class


def deserialize(data):
    params = json.loads(data)
    name = params['class']
    target_class = registry[name]
    return target_class(*params['args'])


class Vector3D(BetterSerializable):
    def __init__(self, x, y, z):
        super().__init__(x, y, z)
        self.x, self.y, self.z = x, y, z


if __name__ == '__main__':
    
    v3 = Vector3D(10, -7, 3)
    print('Before:    ', v3)
    data = v3.serialize()
    print('Serialized:', data)
    print('After:     ', deserialize(data))