# Author：Allister.Liu
# !/usr/bin/env python
import time
"""
装饰器：本质是函数（装饰其它函数），就是为其它函数添加附加功能。
原则：
1、不能修改被装饰的函数的源代码；
2、不能修改被装饰的函数的调用方式
实现装饰器只是储备：
1、函数即“变量”；
2、高阶函数；
a、把一个函数名当做实参传递给另一个函数（再不修改源代码的同时，为其它函数附加新功能）
b、返回值中包含函数名（不修改函数的调用方式）
3、嵌套函数。
高阶函数 + 嵌套函数 ==》 装饰器
"""
# def start():
# print("in the start...")
#
# def test(func):
# start_time = time.time()
# func()
# stop_time = time.time()
# print("the func run time is %5f s" % (stop_time - start_time))
# start()
"""
装饰器举例：test函数中，func（可以是一个函数名《函数名对应的是内存地址》）参数被传入函数start，即 test = func = start，其中func()也就是调用被传入的参数func即start()
所以，test()调用了start()的同时，也附加了打印执行时间的功能。
"""
# test(start)
"""
def timmer(func):
def warpper():
start_time = time.time()
func()
stop_time = time.time()
print("the func run time is %4f 秒" % (stop_time - start_time))
return warpper
def test01():
time.sleep(2)
print("in the test01")
test01 = timmer(test01)
test01()
第一种：
"""
"""
第二种：
@timmer : 等于 test01 = timmer(test01) test01() = timmer()
def timmer(func):
def warpper():
start_time = time.time()
func()
stop_time = time.time()
print("the func run time is %4f 秒" % (stop_time - start_time))
return warpper
@timmer
def test01():
time.sleep(2)
print("in the test01")
test01()
"""
"""
最终版：适用于所有方法，有参数无参数都可以
"""
def timmer(func):
def warpper(*args, **kwargs):
start_time = time.time()
func(*args, **kwargs)
stop_time = time.time()
print("the func run time is %4f 秒" % (stop_time - start_time))
return warpper
@timmer
def test01():
time.sleep(2)
print("in the test01")
test01()
# Author：Allister.Liu
# !/usr/bin/env python
# -*- coding:utf-8-*-
"""
需求：使用装饰器实现登录页面的验证
公司网站：每一个页面当做一个函数，其中主页（index）不需要登录，其余页面都需要登录后才能访问
user, pwd = "Allister", "123456"
# 装饰器实现登录验证
def auth(func):
def wrapper(*args, **kwargs):
username = input("UserName:").strip()
password = input("Password:").strip()
if username == user and pwd == password:
print("user logined sucess!")
return func(*args, **kwargs)
else:
exit("Invalid username or password ")
return wrapper
# 无需登录
def index():
print("welcome to index page")
# 需登录
@auth
def manage():
print("welcome to manage page")
# 需登录
@auth
def bbs():
print("welcome to bbs page")
index()
manage()
bbs()
"""
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
"""
需求优化：manage使用local验证，bbs使用ldap验证
"""
user, pwd = "Allister", "123456"
# 装饰器实现登录验证
def auth(auth_type):
def auth_status(func): #manage at 0x000000000413DD08
def wrapper(*args, **kwargs):
if auth_type == "local":
username = input("UserName:").strip()
password = input("Password:").strip()
if username == user and pwd == password:
print("user logined sucess!")
return func(*args, **kwargs)
else:
exit("Invalid username or password ")
elif auth_type == "ldap":
print("ldap验证登录...")
return wrapper
return auth_status
# 无需登录
def index():
print("welcome to index page")
# 需登录
@auth(auth_type="local") # manage = auth(auth_type="local") ：auth_status门牌,也就是manage=auth_status(manage), wrapper门牌
def manage():
print("welcome to manage page")
# 需登录
@auth(auth_type="ldap")
def bbs():
print("welcome to bbs page")
# index()
manage()
# bbs()
# Author：Allister.Liu
def decorator(func):
print("decorator")
def wrapper(*args, **kwargs):
print("wrapper")
func()
return wrapper
@decorator
def test():
print("the test function")
test()