python装饰器整理笔记

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# def set_fun(fun):
#     def call_fun(*args, **kwargs):
#         print("调ping()前")
#         fun(*args, **kwargs)
#         print("调完ping()")
#     return call_fun
#
# @set_fun
# def ping():
#     print("原函数")
#
# ping()


# def set_fun(fun):
#     def call_fun(*args, **kwargs):
#         return fun(*args, **kwargs)
#     return call_fun
#
# @set_fun
# def ping(a,b):
#     return a + b
#
# print(ping(190,111))

def set_level(level_num):
    def set_fun(fun):
        def call_fun(*args, **kwargs):
            if level_num == 1:
                print("这里用了test1")
            elif level_num == 2:
                print("这里用了test2")
            return fun(*args, **kwargs)
        return call_fun
    return set_fun

@set_level(1)
def test1():
    print("test1")

@set_level(2)
def test2():
    print("test2")

test1()
print("-" * 30)
test2()