Python 装饰器补充

定义了一些函数，这些函数都要被外部所调用，但是这些函数在被调用之前，都有些相同的功能需要被实现，在这种情况下，装饰器是最好的解决方案：

def f1():
    print ("F1")

def f2():
    print ("F2")

在另一个模块中调用的时候

import s1

s1.f1()
s1.f2()
执行结果：
F1
F2

这个时候，我们需要在f1,f2中分别添加打印日志的功能，于是，函数的定义变成了

def f1():
    print ("log")
    print ("F1")


def f2():
    print ("log")
    print ("F2")

在模块中调用执行的结果是：

log
F1
log
F2

这个时候整个函数都要修改，工作量非常之大，于是想到了用一个函数来定义日志的打印功能

def log():
    print ("log")


def f1():
    log()
    print ("F1")


def f2():
    log()
    print ("F2")

通过这样改变代码，我们也可以实现功能，而且每次都只需要修改函数的内容即可。但是这样还是改变了f1,f2函数体的内容，违反了开放封闭原则（函数内的不要修改，函数外的可以修改）

def outer(func):
    def inner():
        print ("log")
        func()
    return inner

@outer     #f1 =  outer(f1)
def f1():
    print ("F1")

@outer
def f2():
    print ("F2")

装饰器的内层原理：@outer有两个功能：1，自动执行outer函数，并且将其紧挨的下面的函数名f1传递给outer函数；2，将outer函数的返回值重新赋值给了f1函数

带返回值的装饰器

def outer(func):
    def inner():
        print ("before")
        func()
        print ("after")
    return inner
# @outer
def f2():
    print ("F2")
    return "SB"

在另一个模块调用，没有使用装饰器的前提下（有返回值）：

r = s1.f2()
print (r)

结果：
F2
SB

添加了装饰器之后，代码改为下面的格式（没有记录返回值）：

def outer(func):
    def inner():
        print ("before")
        func()
        print ("after")
    return inner
@outer
def f2():
    print ("F2")
    return "SB"

执行结果：
before
F2
after
None

打印的结果跟我们需要的不一致（我们需要记录到返回值），于是重新修改了装饰器代码：

def outer(func):
    def inner():
        print ("before")
        r = func()
        print ("after")
        return r
    return inner
@outer
def f2():
    print ("F2")
    return "SB"

执行结果：
before
F2
after
SB

定义带参数的装饰器

def outer(func):
    def inner(a):
        print ("before")
        r = func(a)
        print ("after")
        return r
    return inner
@outer
def f2(a):
    print (a)
    return "SB"

在模块中调用
r = s1.f2("ffffff")
print (r)

执行结果：
before
ffffff
after
SB

双层装饰器

__author__ = 'Alex'
#coding:utf-8

LOING_USER={"Is_login":False}

def outer(func):
    def inner(*args,**kwargs):
        if LOING_USER['Is_login']:
            func()
        else:
            print ("Please login first!")
    return inner

# def outer2(func):
#     def inner(*args,**kwargs):
#         if LOING_USER['Is_login']:
#             if LOING_USER.get("user_level") == 2:
#                 func()
#             else:
#                 print ("You do not have priviledge!")
#         else:
#             print ("Please login first!")
#     return inner

def outer2(func):
    def inner(*args,**kwargs):
        if LOING_USER.get("user_level") == 2:
            func()
        else:
            print ("You do not have priviledge!")
    return inner



@outer
def check():
        print ("welcome %s login" %LOING_USER['current_user'])
    # else:
    #     print ("if you want to check,Please login first!")

@outer2
@outer
def manager():
    # if LOING_USER['Is_login']:
        print ("welcome %s login" %LOING_USER['current_user'])
        print ("You level is %s" %LOING_USER['user_level'])
    # else:
    #     print ("Please login first!")


def login():
    username = input("Please input the username: ")
    if username == "alex":
        LOING_USER["Is_login"] = True
        LOING_USER["current_user"] = username
        print ("You login success")
    elif username == "admin":
        LOING_USER["Is_login"] = True
        LOING_USER["current_user"] = username
        LOING_USER["user_level"] = 2
    else:
        print ("You are invalid")



def main():
    while True:
        print('''
        1.登录
        2.后台管理
        3.检查
        ''')
        inp = input ("Please input the choise:")
        if inp == '1':
            login()
        elif inp == '2':
            manager()
        elif inp == '3':
            check()
        print (LOING_USER)
main()