1 # Author: Lockegogo
 2 
 3 user, passwd = 'LK', '130914'
 4 def auth(auth_type):
 5     print('auth func:', auth_type)
 6     def outher_wrapper(func):
 7         def wrapper(*args, **kwargs):
 8             print('wrapper func:', *args, **kwargs)
 9             if auth_type == 'local':
10                 username = input('username:').strip()
11                 password = input('password:').strip()
12                 if user == username and password == passwd:
13                     print('\033[32;1mUser has passed authentication\033[0m')
14                     res = func(*args, **kwargs)
15                     return res
16                 else:
17                     exit('\033[32;1mInvalid Username or password\033[0m')
18             elif auth_type == 'ldap':
19                 print('ldap，不会')
20         return wrapper
21     return outher_wrapper
22 
23 def index():
24     print('welcome to index page')
25 @auth(auth_type='local') # home = outher_wrapper(home)
26 def home():
27     print('welcome to home page')
28     return 'from home'
29 @auth(auth_type='ldap')
30 def bbs():
31     print('welcome to bbs page')
32 
33 index()
34 print(home())
35 bbs()