首页 > 编程语言 > 详细

python装饰器高级用法

时间:2017-07-23 15:00:25      阅读:311      评论:0      收藏:0      [点我收藏+]

1.装饰类

  下面,直接来看代码是如何实现装饰类的:

def decorator(aClass):
    class newClass:
        def __init__(self, age):
            self.total_display   = 0
            self.wrapped         = aClass(age)
        def display(self):
            self.total_display += 1
            print("total display", self.total_display)
            self.wrapped.display()
    return newClass

@decorator # 接受一个类参数
class Bird:
    def __init__(self, age):
        self.age = age
    def display(self):
        print("My age is",self.age)

eagleLord = Bird(5)
for i in range(3):
    eagleLord.display()

  在decorator中,我们返回了一个新类newClass。在新类中,我们记录了原来类生成的对象(self.wrapped),并附加了新的属性total_display,用于记录调用display的次数。我们也同时更改了display方法。

  通过修改,我们的Bird类可以显示调用display的次数了

2.装饰器捕捉异常

  有一个check类,其中有方法read_value()。由于某些原因,方法read_value有可能抛出异常而使程序崩溃,所以需要对整个方法做try....except处理,如:

  没有捕捉异常之前:

class check(object):
    def __init__(self):
        pass

    def receive(self):
        print(receive from exception.)

    def read_value(self):
        print(here i will do something)
        a + 1  # 这里会报异常

c = check()
c.read_value()

  加了try....except捕捉异常后:

class check(object):
    def __init__(self):
        pass

    def receive(self):
        print(receive from exception.)

    def read_value(self):
        try:
            print(here i will do something)
            a + 1
        except Exception as e:
            print(e)

c = check()
c.read_value()

  虽然程序不会报出异常,但这样处理,有些丑陋。来看看装饰器是如何处理的:

def catch_exception(func):
    def wrapper(*args,**kwargs):
        try:
            u = func(*args,**kwargs)
            return u
        except Exception:
            return an exception raised
    return wrapper


class check(object):
    def __init__(self):
        pass

    def receive(self):
        print(receive from exception.)

    @catch_exception
    def read_value(self):
        print(here i will do something)
        a + 1

c = check()
c.read_value()

  很好,使用装饰器,来装饰函数,这样使我们的代码更加pythonic。但是,如果程序报出异常后,调用类中的另外一个方法,那怎么办?很简单,给装饰器加上一个参数self:

def catch_exception(func):
    def wrapper(self,*args,**kwargs):
        try:
            u = func(self,*args,**kwargs)
            return u
        except Exception:
            self.receive()
            return an exception raised
    return wrapper
class check(object):
    def __init__(self):
        pass
    
    def receive(self):
        print(receive from exception.)
    
    @catch_exception
    def read_value(self):
        print(here i will do something)
        a + 1
        
c = check()
c.read_value()

  有木有感觉,很神奇。

 

python装饰器高级用法

原文:http://www.cnblogs.com/vipchenwei/p/7224491.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!