Python之裝飾器

裝飾器是由函數(shù)去生成的，用于裝飾某個(gè)函數(shù)或者方法或者類，他可以讓這個(gè)函數(shù)在執(zhí)行之前或者執(zhí)行之后做一些操作。

實(shí)例

先定義一個(gè)函數(shù)func

1. #!/usr/bin/env python

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

4. def func(arg):  # 接受一個(gè)參數(shù)arg

5.    print(arg)  # 輸出這個(gè)參數(shù)

7. func("Hello World!")  # 調(diào)用腳本并且傳入?yún)?shù)

執(zhí)行腳本，輸出的結(jié)果為：

1. C:\Python35\python.exe F:/Python_code/Note/裝飾器.py

2. Hello World!
   

4. Process finished with exit code 0

現(xiàn)要在執(zhí)行func這個(gè)函數(shù)前后執(zhí)行一些操作，就可以創(chuàng)建一個(gè)裝飾器來(lái)實(shí)現(xiàn)：

1. #!/usr/bin/env python

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

4. def decorator(func):  # 創(chuàng)建一個(gè)裝飾器函數(shù)，接受的參數(shù)arg參數(shù)就是func函數(shù)名

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

7.        print("執(zhí)行函數(shù)之前")

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

9.        print("執(zhí)行函數(shù)之后")

10.        return ret

12.    return inner

14. @decorator  # 如果要讓某個(gè)函數(shù)使用裝飾器，只需要在這個(gè)函數(shù)上面加上@+裝飾器名

15. def func(arg):

16.    print(arg)

18. func("Hello World!")

輸出結(jié)果為：

1. /usr/bin/python3.5 /home/ansheng/Documents/PycharmProjects/blogcodes/裝飾器.py

2. 執(zhí)行函數(shù)之前

3. Hello World!

4. 執(zhí)行函數(shù)之后

6. Process finished with exit code 0

多個(gè)裝飾器裝飾同一個(gè)函數(shù)

1. #!/usr/bin/env python

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

4. def decorator1(func):

5.    def inner():

6.        print("開始之前執(zhí)行裝飾器01")

7.        ret = func()

8.        print("結(jié)束之后執(zhí)行裝飾器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("執(zhí)行函數(shù)...")

29. index()

輸出結(jié)果：

1. /usr/bin/python3.5 /home/ansheng/Documents/PycharmProjects/blogcodes/裝飾器.py

2. 開始之前執(zhí)行裝飾器01

3. decorator2>>>Start...

4. 執(zhí)行函數(shù)...

5. decorator2>>>End...

6. 結(jié)束之后執(zhí)行裝飾器01

8. Process finished with exit code 0

更多實(shí)例

1. #!/usr/bin/env python

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

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

6. """

7. 函數(shù)裝飾器

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. 帶參數(shù)的函數(shù)裝飾器

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. 帶參數(shù)的類裝飾器

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. 帶參數(shù)的類裝飾器

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. 為函數(shù)中和類中的方法添加裝飾器

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())

本文出自 “一盞燭光” 博客，謝絕轉(zhuǎn)載！