python 修改的函数装饰器

把好的代码记录下来 方便以后学习

修改的函数参数装饰器

from functools import wraps
import time
import logging
def warn(timeout):
#timeout=[timeout] python 2.x
def decorator(func):
def wrapper(*args,**kargs):
start = time.time()
res = func(*args,**kargs)
used = time.time()-start
#if(used > timeout[0]): python 2.x
if(used > timeout):
msg = '"%s":"%s" > %s' % (func.__name__,used,timeout)
logging.warn(msg)
return res
def setTimeOut(k):
nonlocal timeout
timeout = k
#timeout[0] = k python 2.x
wrapper.setTimeOut = setTimeOut
return wrapper
return decorator from random import randint
@warn(1.5)
def test():
print('in test')
while randint(0,1):
time.sleep(0.5) for _ in range(30):
test()
test.setTimeOut(1)
for _ in range(30):
test()

检测带参数类型的函数装饰器

from inspect import signature
def typeassert(*ty_args,**ty_kargs):
def decorator(func):
sig = signature(func)
btypes = sig.bind_partial(*ty_args,**ty_kargs).arguments
def wrapper(*args,**kargs):
for name,obj in sig.bind_partial(*args,**kargs).arguments.items():
if name in btypes:
if not isinstance(obj,btypes[name]):
raise TypeError("%s must be %s" % (name,btypes))
return func(*args,**kargs)
return wrapper
return decorator @typeassert(int,str,list)
def f(a,b,c):
print(a,b,c) f(1,'abc',[1,2,3])
f(1,2,[1,2,3])
上一篇:Java连载145-对象概论总结


下一篇:作为iOS开发者不得不follow的52人