Python 学习之路 - 装饰器

装饰器

装饰器也是函数，用于被装饰函数执行时，增加被装饰函数的功能。

def decor(func):
    def warpper(*args,**kwargs):
        print("Welcome!")
        func(*args,**kwargs)
        print("*-------------*")
    return warpper


@decor
def simple(name):
    print(name)

simple("ChangDu")

import time

def timmer(func):
    def warpper(*args,**kwargs):
        start_time=time.time()
        func(*args,**kwargs)
        stop_time=time.time()
        print("The running time:%s"%(stop_time-start_time))
    return warpper

# @ + 函数名
# 1. 自动执行timmer函数并将下面的函数名test1当做参数传递
# 2. 将timmer函数的返回值重新赋值给test1
@timmer #test1=timmer(test1)
def test1(name,ss):#一旦一个函数被装饰，这个函数就会被赋值成装饰函数的内层函数
    time.sleep(3)
    print("test")
    print(name,ss)

test1("amy",ss="uu")

db_user_name,db_password= "amy" , "123"

def auth(func):
    print("----",func)
    def warpper(*args,**kwargs):
        user_name=input("UserName:").strip()
        password=input("Password:").strip()
        if db_user_name==user_name and db_password==password:
            r = func()
            print("hhhh")
            return r
        else:
            print("Error")
    return warpper



def index():
    print("welcome to index page")

@auth
def home():
    print("welcome to home page")
    return "welcome"

@auth
def bbs():
    print("welcome to bbs page")


index()
print(home())
bbs()

name,password="amy","123"
def auth(way):
    def outer(func):
        def wrapper(*args,**kwargs):
            if way == "local":
                user_name=input("UserName:").strip()
                user_password=input("Password:").strip()
                if user_name==name and user_password==password:
                    print("Welcome------local")
                    r=func()
                    print("End")
                    return r
                else:
                    print("Error")
            elif way == "other":
                print("forget it-----other")
                r = func()
                return r
        return wrapper
    return outer


def index():
    print("Welcome to index")


@auth(way="local")
def home():
    print("Welcome to home")
    return True


@auth(way="other")
def bbs():
    print("Welcome to bbs")

index()
print(home())
bbs()

def outer(func):
    def warpper(*args,**kwargs):
        print("1")
        r = func(*args,**kwargs)
        return r
    return warpper


def outer2(func):
    def warpper(*args,**kwargs):
        print("2")
        r = func(*args,**kwargs)
        return r
    return warpper

@outer
@outer2
def f1():
    print("test")

f1()