装饰器测试

zhanghe06 · zhanghe06 · commit 240f6f96157a · 2017-07-22T16:32:48.000+08:00
diff --git a/test/test_wrapper.py b/test/test_wrapper.py
@@ -0,0 +1,78 @@
+#!/usr/bin/env python
+# encoding: utf-8
+
+"""
+@author: zhanghe
+@software: PyCharm
+@file: test_wrapper.py
+@time: 2017/8/5 &#19979;&#21320;7:57
+"""
+
+
+def w1(func):
+    def wrapper(*args, **kw):
+        back_func = func(*args, **kw)
+        print func.__name__, 'w1'
+        return back_func
+    return wrapper
+
+
+def w2(func):
+    def wrapper(*args, **kw):
+        back_func = func(*args, **kw)
+        print func.__name__, 'w2'
+        return back_func
+    return wrapper
+
+
+def w3(func):
+    def wrapper(*args, **kw):
+        back_func = func(*args, **kw)
+        print func.__name__, 'w3'
+        return back_func
+    return wrapper
+
+
+@w1
+@w2
+@w3
+def func1():
+    return 1
+
+
+@w1
+@w2
+@w3
+def func2():
+    return 2
+
+
+@w1
+@w2
+@w3
+def func3():
+    return 3
+
+
+@w1
+@w2
+@w3
+def fuck(func):
+    def wrapper(*args, **kw):
+        back_func = func(*args, **kw)
+        print func.__name__, 'fuck'
+        return back_func
+    return wrapper
+
+
+@fuck
+def test():
+    return 'test'
+
+
+if __name__ == '__main__':
+    # print func1()
+    # print func2()
+    # print func3()
+    # print test()
+    pass
