Python开发【第一篇】Python基础之装饰器

写代码要遵循开发封闭原则，虽然在这个原则是用的面向对象开发，但是也适用于函数式编程，简单来说，它规定已经实现的功能代码不允许被修改，但可以被扩展，即：

封闭：已实现的功能代码块
开放：对扩展开发

#s2.py

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

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

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

@outer
def f100():
    print("F100")

#b1.py
import s2

s2.f1()
s2.f2()

一、装饰器流程剖析储备知识：

1、
def f1():
    print('123')

def f1():
    print('456')

f1()

2、
def f1():
    print('123')

def f2(xxx):
    xxx()

f2(f1)

　二、装饰器流程剖析：

def outer(func):
    def inner():
        print('before')
        func()
        print('after')
    return inner

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

# @ + 函数名
# 功能
# 1. 自动执行outer函数并且将其下面的函数名f1当作参数传递
# 2. 将outer函数的返回值，重新复制给 f1

# 定义函数，为调用，函数内部不执行
# 函数名 》 代指函数

三、装饰器流程剖析之返回值：

# s1.py

def outer(func):
    def inner():
        print('before')
        r = func()
        print('after')
        return r
    return inner

@outer
def f1():
    print("F1")
    return "nini"

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

# b1.py
import s2

ret = s2.f1()
print("返回值",ret)

注：return 不能先执行，如果先执行后面的操作将不会操作，所以return放在最后。

四、装饰器流程剖析之参数：

# s1.py

def outer(func):
    def inner(a):
        print('before')
        r = func(a)
        print('after')
        return r
    return inner

@outer
def f1(arg):
    print(arg)
    return "nini"

# b1.py

import s2

ret = s2.f1('faffa')
print("返回值",ret)

多参数传递，万能参数：

#s1.py

def outer(func):
    def inner(*args,**kwargs):
        print('before')
        r = func(*args,**kwargs)
        print('after')
        return r
    return inner

@outer
def f1(arg):
    print(arg)
    return "nini"

@outer
def f2(a1,a2):
    print("F2")

#b1.py
import s2

ret = s2.f1('faffa')
print("返回值",ret)

s2.f2(11,22)

五、实例：用户管理程序：

装饰器最常用的功能之一是可以做权限的验证

LOGIN_USER = {"is_login":False}

def outer(func):
    def inner(*args,**kwargs):
        if LOGIN_USER['is_login']:
            r = func()
            return r
        else:
            print("请登录")
    return inner

@outer
def order():
    print("欢迎%s登录" % LOGIN_USER['current_user'])

@outer
def changepwd():
    print("欢迎%s登录" % LOGIN_USER['current_user'])

@outer
def manager():
    print("欢迎%s登录" % LOGIN_USER['current_user'])

def login(user,pwd):
    if user == "alex" and pwd == "123":
        LOGIN_USER['is_login'] = True
        LOGIN_USER['current_user']=user
        manager()

def main():
    while True:
        inp = input("1,后台管理; 2,登录")
        if inp == '1':
            manager()
        elif inp == '2':
            username = input("请输入用户名")
            pwd = input("请输入密码")
            login(username,pwd)

main()

六、双层装饰器实现用户登录和权限验证：

逻辑：多层装饰器，是从下往上一步步执行得。

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

USER_INFO = {}

def check_login(func):
    def inner(*args,**kwargs):
        if USER_INFO.get('is_login', None):
            ret = func(*args,**kwargs)
            return ret
        else:
            print('请登录')
    return inner

def check_admin(func):
    def inner(*args,**kwargs):
        if USER_INFO.get('is_login', None):
            if USER_INFO.get('user_type', None) == 2:
                ret = func(*args,**kwargs)
                return ret
            else:
                print('无权限查看')
        else:
            print('请登录')
    return inner

@check_login
@check_admin
def index():
    """
    管理员的功能
    :return:
    """
    print('Index')

@check_login
def home():
    """
    普通用户的功能
    :return:
    """
    print('home')

def login():
    user = input('请输入用户名:')
    if user == 'admin':
        USER_INFO['is_login'] = True
        USER_INFO['user_type'] = 2
    else:
        USER_INFO['is_login'] = True
        USER_INFO['user_type'] = 1

def main():
    while True:
        inp = input('1、登录；2、查看信息；3、超级管理员管理 \n >>> ')
        if inp == "1":
            login()
        elif inp =='2':
            home()
        elif inp == '3':
            index()

main()

参考自：http://www.cnblogs.com/wupeiqi/articles/4980620.html