对于重要的系统我们仅仅控制登录是不够的,对于固定人员使用到的系统我们还是要进行权限的细分。下面是bollte框架下的一个简单的例子。
def www_auth(func):
'''
装饰器限制ip访问功能
'''
def wrapper(*args,**kargs):
ips = ['192.168.1.1']
remoteip = request.environ.get('REMOTE_ADDR')
#print remoteip
if remoteip in ips:
return func(**kargs)
else:
redirect('/error')
return wrapper
#我们再函数的前面加@www_auth即可,这个是对index请求限制的例子。
@route('/index', method='GET')
@www_auth
def index():
#其他逻辑代码
return template('html/test.html')
@route('/error',method='GET')
def error():
return template('html/error.html')
下面的例子是类装饰器对类中的方法进行装饰。
class Foo(object):
def __init__(self):
pass
def __call__(self,func):
def _call(*args,**kw):
print 'class decorator runing'
return func(*args,**kw)
return _call
class Bar(object):
@Foor()
def bar(self,test,ids):
print 'bar'
Bar().bar('aa','ids')
我们经常会用到的比如说给多线程函数加锁来保证共享数据操作的完整性。
线程锁装饰器如下:
class StandardOut(object):
"""
线程锁装饰器
"""
def __init__(self):
self.thread_lock = threading.Lock()
def __call__(self,func):
def _call(*args,**kw):
self.thread_lock.acquire()
func(*args,**kw)
self.thread_lock.release()
return _call