装饰器(decoreter)

# -*-coding:utf-8 -*-
# Author:GOY

#事先预留用于登录的用户名密码
username = "goy"
password = "123"

#用于登录的装饰器
def login(func):
    def wrapperd(*args,**kwargs):
        username = input("input username:").strip()
        password = input("input password:").strip()
        if username == "goy" and password == "123":
            print("\033[32;1m login successful \033[0m")
            res = func(*args,**kwargs)
            print("-----------------------------")
            return res
        else:
            exit("login failed")
    return wrapperd

def index():
    print("index")

@login
def home():
    print("home")
    return "from home"

@login
def bbs():
    print("bbs")

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

 

posted @ 2019-08-11 13:26  小白的忧郁  阅读(179)  评论(0)    收藏  举报