python学习-装饰器问答

#-*- coding:utf-8 -*-
import functools
import types
'''
请编写一个decorator，能在函数调用的前后打印出
'begin call'和'end call'的日志。
'''
#这个不难
def logt(text):
def log(func):
@functools.wraps(func)
def wrapper(*args,**kw):
print '%s %s():'% (text,func.__name__)
func(*args,**kw)
print 'finish %s():'% func.__name__
return wrapper
return log
@logt('execute')
def current3():
print 'nihaoa'

current3()

'''
再思考一下能否写出一个@log的decorator，使它
既支持：
@log
def f():
pass

又支持：
@log('execute')
def f():
pass
'''
#这个就有点难为我了,我就在此沦陷啦
#首先要使用默认值,这一点还是比较清楚的
#以下代码为模仿别人的
def log(text='execute'):
def logdecorate(func):
@functools.wraps(func)
def wrapper(*args,**kw):
'''
如果text有值,那就要打印出该值
如果没值,那就打印默认值,所以都是
text
'''
print '%s %s()' % (text,func.__name__)
return func(*args,**kw)
return wrapper
'''
如果text参数有传值过来的话,
说明用的是log('text'),即log('text')(now)
这个时候要将log('text')转成logdecorate才成
即函数原型,然后才能变为log('text')(now)==logdecorate(now)
所以有参数时直接返回logdecorate函数
'''
#判断text是否属于函数类型
if not isinstance(text,types.FunctionType):
#为log('text')
return logdecorate
'''
如果没有传值过来的话,说明用的是log,即log(now).
text==now,其实就是要调用logdecorate(now)
所以无参数时,直接返回调用logdecorate(now)的结果
'''
if isinstance(text,types.FunctionType):
#为log形式
return logdecorate(text)

@log
def func_one():
print 'log has no args'

@log('diff')
def func_two():
print 'log has args'

func_one()

#然后有人用一个程序包含了上面2道题,
#我只能在地上挖矿啦.贴出这位前辈的代码
#才疏学浅,解释不清
def log(text='execute'):

def decorator(func):

@functools.wraps(func)

def wrapper(*args, **kw):

print "begin call %s():" %(func.__name__)

result = func(*args, **kw)

if isinstance(text,types.StringType) and len(text) != 0:
print 'text 的值为字符串类型'
print "%s %s()" %(text, func.__name__)
else:
print "end call %s()" %(func.__name__)
return result

return wrapper

if isinstance(text, types.FunctionType):
print text
print 'text 的值为函数类型'
return decorator(text)

else:
print text
print 'text的值不为函数类型，无参调用函数'
return decorator

@log
def now():
print 'mult'
print 'log 无参数:'
now()
#log(now)

@log('test')
def again():
print 'you args'

print 'log 带参数:'
again()
#log('test')(now)
#我还有救么