这里分享一下 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}