Python之装饰器

装饰器是由函数去生成的，用于装饰某个函数或者方法或者类，他可以让这个函数在执行之前或者执行之后做一些操作。

实例

先定义一个函数func

1. #!/usr/bin/env python

2. # _*_ coding: utf-8 _*_

4. def func(arg):  # 接受一个参数arg

5.    print(arg)  # 输出这个参数

7. func("Hello World!")  # 调用脚本并且传入参数

执行脚本，输出的结果为：

1. C:\Python35\python.exe F:/Python_code/Note/装饰器.py

2. func

4. Process finished with exit code 0

现要在执行func这个函数前后执行一些操作，就可以创建一个装饰器来实现：

1. #!/usr/bin/env python

2. # _*_ coding: utf-8 _*_

4. def decorator(func):  # 创建一个装饰器函数，接受的参数arg参数就是func函数名

6.    def inner(*args, **kwargs):

7.        print("执行函数之前")

8.        ret = func(*args, **kwargs)

9.        print("执行函数之后")

10.        return ret

12.    return inner

14. @decorator  # 如果要让某个函数使用装饰器，只需要在这个函数上面加上@+装饰器名

15. def func(arg):

16.    print(arg)

18. func("Hello World!")

输出结果为：

1. /usr/bin/python3.5 /home/ansheng/Documents/PycharmProjects/blogcodes/装饰器.py

2. 执行函数之前

3. Hello World!

4. 执行函数之后

6. Process finished with exit code 0

多个装饰器装饰同一个函数

1. #!/usr/bin/env python

2. # _*_ coding: utf-8 _*_

4. def decorator1(func):

5.    def inner():

6.        print("开始之前执行装饰器01")

7.        ret = func()

8.        print("结束之后执行装饰器01")

9.        return ret

11.    return inner

14. def decorator2(func):

15.    def inner():

16.        print("decorator2>>>Start...")

17.        ret = func()

18.        print("decorator2>>>End...")

19.        return ret

21.    return inner

24. @decorator1

25. @decorator2

26. def index():

27.    print("执行函数...")

29. index()

输出结果：

1. /usr/bin/python3.5 /home/ansheng/Documents/PycharmProjects/blogcodes/装饰器.py

2. 开始之前执行装饰器01

3. decorator2>>>Start...

4. 执行函数...

5. decorator2>>>End...

6. 结束之后执行装饰器01

8. Process finished with exit code 0

更多实例

1. #!/usr/bin/env python

2. # _*_ coding:utf-8 _*_

4. # Created by 安生 on 2017/2/9

6. """

7. 函数装饰器

8. """

11. def decorator(func):

12.    def wrapped(*args, **kwargs):

13.        return func(*args, **kwargs)

15.    return wrapped

18. @decorator

19. def func(a, b):

20.    return a + b

23. print(func(1, 2))

24. """

25. 类装饰器

26. """

29. class decorator:

30.    def __init__(self, func):

31.        self.func = func

33.    def __call__(self, *args, **kwargs):

34.        return self.func(*args, **kwargs)

37. @decorator

38. def func(a, b):

39.    return a + b

42. print(func(1, 2))

43. """

44. 带参数的函数装饰器

45. """

48. def parameter(a, b):

49.    print(a, b)

51.    def decorator(func):

52.        def wrapped(*args, **kwargs):

53.            return func(*args, **kwargs)

55.        return wrapped

57.    return decorator

60. @parameter(1, 2)

61. def func(a, b):

62.    return a + b

65. print(func(10, 20))

66. """

67. 带参数的类装饰器

68. """

71. def parameter(a, b):

72.    print(a + b)

74.    class decorator:

75.        def __init__(self, func):

76.            self.func = func

78.        def __call__(self, *args, **kwargs):

79.            return self.func(*args, **kwargs)

81.    return decorator

84. @parameter(1, 2)

85. def func(a, b):

86.    return a + b

89. print(func(10, 20))

91. """

92. 带参数的类装饰器

93. """

96. def parameter(a, b):

97.    print(a, b)

99.    def decorator(cls):

100.        class wrapped:

101.            def __init__(self, *args, **kwargs):

102.                self.cls = cls(*args, **kwargs)

104.            def __getattr__(self, item):

105.                return getattr(self.cls, item)

107.        return wrapped

109.    return decorator

112. @parameter(1, 2)

113. class CLS:

114.    def __init__(self):

115.        self.a = 'a'

117.    def P(self, v):

118.        print(v)

121. obj = CLS()

122. print(obj.a)

123. obj.P('Hello,')

125. """

126. 为函数中和类中的方法添加装饰器

127. """

130. def Call(aClass):

131.    calls = 0

133.    def onCall(*args, **kwargs):

134.        nonlocal calls

135.        calls += 1

136.        print('call %s to %s' % (calls, func.__name__))

137.        return aClass(*args, **kwargs)

139.    return onCall

142. @Call

143. def func(a, b):

144.    return a + b

147. print(func(1, 2))

150. class CLS:

151.    def __init__(self):

152.        self.a = 'a'

154.    @Call

155.    def b(self):

156.        return self.a

159. obj = CLS()

160. print(obj.b())