python装饰器

今天在Leetcode刷题的时候,碰到了python里一个装饰器,这是python后来版本里加的一个名词,在此之前有一些概念

 

举个例子,一个基本的实例方法就向下面这个:

 
class Kls(object):
    def __init__(self, data):
        self.data = data
    def printd(self):
        print(self.data)
ik1 = Kls('arun')
ik2 = Kls('seema')
ik1.printd()
ik2.printd()

 

这会给出如下的输出:
arun
seema

然后看一下代码和示例图片:

  • 1,2参数传递给方法.
  • 3 self参数指向当前实例自身.
  • 4 我们不需要传递实例自身给方法,Python解释器自己会做这些操作的.

如果现在我们想写一些仅仅与类交互而不是和实例交互的方法会怎么样呢? 我们可以在类外面写一个简单的方法来做这些,但是这样做就扩散了类代码的关系到类定义的外面. 如果像下面这样写就会导致以后代码维护的困难:

 
def get_no_of_instances(cls_obj):
    return cls_obj.no_inst
class Kls(object):
    no_inst = 0
    def __init__(self):
        Kls.no_inst = Kls.no_inst + 1
ik1 = Kls()
ik2 = Kls()
print(get_no_of_instances(Kls))

 

输出:
2
@classmethod
我们要写一个只在类中运行而不在实例中运行的方法. 如果我们想让方法不在实例中运行,可以这么做:

 
def iget_no_of_instance(ins_obj):
    return ins_obj.__class__.no_inst
class Kls(object):
    no_inst = 0
    def __init__(self):
    Kls.no_inst = Kls.no_inst + 1
ik1 = Kls()
ik2 = Kls()
print iget_no_of_instance(ik1)

 

输出
2
在Python2.2以后可以使用@classmethod装饰器来创建类方法.

 
class Kls(object):
    no_inst = 0
    def __init__(self):
        Kls.no_inst = Kls.no_inst + 1
    @classmethod
    def get_no_of_instance(cls_obj):
        return cls_obj.no_inst
ik1 = Kls()
ik2 = Kls()
print ik1.get_no_of_instance()
print Kls.get_no_of_instance()

 

输出:
2
2
这样的好处是: 不管这个方式是从实例调用还是从类调用,它都用第一个参数把类传递过来.
@staticmethod
经常有一些跟类有关系的功能但在运行时又不需要实例和类参与的情况下需要用到静态方法. 比如更改环境变量或者修改其他类的属性等能用到静态方法. 这种情况可以直接用函数解决, 但这样同样会扩散类内部的代码,造成维护困难.
比如这样:

 
IND = 'ON'
def checkind():
    return (IND == 'ON')
class Kls(object):
     def __init__(self,data):
        self.data = data
def do_reset(self):
    if checkind():
        print('Reset done for:', self.data)
def set_db(self):
    if checkind():
        self.db = 'new db connection'
        print('DB connection made for:',self.data)
ik1 = Kls(12)
ik1.do_reset()
ik1.set_db()

输出:
Reset done for: 12
DB connection made for: 12
如果使用@staticmethod就能把相关的代码放到对应的位置了.

 
IND = 'ON'
class Kls(object):
    def __init__(self, data):
        self.data = data
    @staticmethod
    def checkind():
        return (IND == 'ON')
    def do_reset(self):
        if self.checkind():
            print('Reset done for:', self.data)
    def set_db(self):
        if self.checkind():
            self.db = 'New db connection'
        print('DB connection made for: ', self.data)
ik1 = Kls(12)
ik1.do_reset()
ik1.set_db()

 

输出:
Reset done for: 12
DB connection made for: 12
下面这个更全面的代码和图示来展示这两种方法的不同
@staticmethod 和 @classmethod的不同

 
class Kls(object):
    def __init__(self, data):
        self.data = data
    def printd(self):
        print(self.data)
    @staticmethod
    def smethod(*arg):
        print('Static:', arg)
    @classmethod
    def cmethod(*arg):
        print('Class:', arg)
 >>> ik = Kls(23)
>>> ik.printd()
23
>>> ik.smethod()
Static: ()
>>> ik.cmethod()
Class: (<class '__main__.Kls'>,)
>>> Kls.printd()
TypeError: unbound method printd() must be called with Kls instance as first argument (got nothing instead)
>>> Kls.smethod()
Static: ()
>>> Kls.cmethod()
Class: (<class '__main__.Kls'>,)

下面这个图解释了以上代码是怎么运行的:

 

类方法

类方法的第一个参数是定义此方法的类对象,通过类或者创建的实例都可以调用,可以访问类属性

静态方法

静态方法基本是一个对立的方法,跟类貌似没啥关系,仅仅是放在了一个类里的函数,所以无法访问类属性

#!/usr/bin/env python
# coding = utf-8

class MyClass(object):
    a = 2
    def __init__(self):
        self.b = 1
    def foo1(self):
        self.b += 1
    def foo2():
        print 'foo2, nothing to do'
    staticfoo2 = staticmethod(foo2)
    def foo3(cls):
        print 'foo3, ' * cls.a
    classfoo3 = classmethod(foo3)

if __name__ == '__main__':
    lihui = MyClass()
    lihui.foo1()
    print lihui.b
    lihui.staticfoo2()
    lihui.classfoo3()

这里foo2是静态方法,无法访问a和b;foo3是类方法,可以访问a,但是无法访问b

后来,staticmethod和classmethod写法被简化了,代码也美观了不少

#!/usr/bin/env python
# coding = utf-8

class MyClass(object):
    a = 2
    def __init__(self):
        self.b = 1
    def foo1(self):
        self.b += 1
    @staticmethod
    def foo2():
        print 'foo2, nothing to do'
    @classmethod
    def foo3(cls):
        print 'foo3, ' * cls.a

if __name__ == '__main__':
    lihui = MyClass()
    lihui.foo1()
    print lihui.b
    lihui.foo2()
    lihui.foo3()

输出结果

E:\python\python.exe E:/py/hello.py
2
foo2, nothing to do
foo3, foo3, 

Process finished with exit code 0

 

 

发表回复