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

2022-10-25 21:07:36 浏览数 (1)

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

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

代码语言:javascript复制
class Pet:
    def __init__(self, name):
        self.name = name

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

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

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

代码语言:javascript复制
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}")

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

代码语言:javascript复制
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 就是一个例子:

代码语言:javascript复制
>>> dict.fromkeys("ABC", 0)
{'A': 0, 'B': 0, 'C': 0}
>>> dict.fromkeys(["hello","world"])
{'hello': None, 'world': None}

0 人点赞