提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  Javascript

当前栏目

用 Classmethod 自定义创建对象的方式

2023-02-19 12:28:30 时间

这里分享一下 classmethod 的实用小技巧。

classmethod 是一个非常实用的装饰器，可以用它来改变类创建对象的形为，比如说这样一段代码：

class Pet:
    def __init__(self, name):
        self.name = name

传入一个 name，初始化一个 Pet 对象，如果你觉得这太单调了，想实现非常个性的初始化方法，那么可以修改 __init__ 函数的代码。

但是，有些情况你不能修改 __init__​ 函数的代码，比如代码已经运行很久了，有很多地方调用了 Pet 的 __init__ 方法，如果直接修改，可能会产生很多问题。

这时候就可以借助于装饰器 classmethod，自定义对象的创建方式：

import random

class Pet:
    def __init__(self, name):
        self.name = name

    @classmethod
    def create(cls):
        number = random.randint(100_000,999_999)
        return cls(f"Pet_{number}")

然后，可以这样调用，输出结果在注释里。

dog = Pet("Dog")
print(dog.name)
# Dog
pet = Pet.create()
print(pet.name)
# Pet_444383
print(isinstance(dog, Pet))
# True
print(isinstance(pet, Pet))
# True

此外，标准库里也有很多使用 classmethod 的例子，比如 dict.fromkeys 就是一个例子：

>>> dict.fromkeys("ABC", 0)
{'A': 0, 'B': 0, 'C': 0}
>>> dict.fromkeys(["hello","world"])
{'hello': None, 'world': None}

相关文章