1.判断用户权限
['root','admin','redhat']
1.user in [] (先判断用户是否有权限登陆)
2.user+vip (再判断用户是否是有特殊权限)
import functools
import inspect
login_session = ['root','admin','redhat']
def is_login(fun):
@functools.wraps(fun)
def warapper(*args,**kwargs):
if args[0] in login_session:
temp = fun(*args,**kwargs)
return temp
else:
print('Error:用户%s 没有登陆成功' %(args[0]))
return warapper
def is_admin(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
inspect_res = inspect.getcallargs(fun,*args,**kwargs)
print('inspect的返回值是:%s' %(inspect_res))
if inspect_res.get('name') == 'root':
temp = fun(*args,**kwargs)
return temp
else:
print('not root user,no permisson add user')
return wrapper
@is_login
@is_admin
def add_user(name): # is_login((is_admin(adduser)) is_login(wrapper())
print('add_user')
add_user('haha')
当输入非法用户时:
当输入一般用户时:
当输入拥有特殊权限的用户时:
2. 编写装饰器required_ints, 条件如下:
1). 确保函数接收到的每一个参数都是整数;
2). 如果参数不是整形数, 打印 TypeError:参数必须为整形
import functools
def required_ints(func):
@functools.wraps(func)
def wrapper(*args,**kwargs):
for i in args:
if not isinstance(i,int):
print('函数所有的参数并非都是int型')
break
else:
res = func(*args,**kwargs)
return res
return wrapper
@required_ints
def add(a,b):
return a+b
@required_ints
def mymax(a,b,c,d):
return max(a,b,c,d)
print(add(1,5))