编写一个decorator,能在函数调用的前后打印出'begin call'和'end call'的日志

注意:我们要借助Python的@语法,把decorator置于函数的定义处

初步尝试:

def log(func):
	def wrapper(*args, **kw):
		print 'begin call: %s' %func.__name__
		return func(*args, **kw)
	print 'end call: %s' %func.__name__
	return wrapper

@log
def now():
	print '2018-06-21'

now()

结果不如预期~:


打印的日志都在函数运行之前

这是因为函数只在返回时调用,如果能调用后,加一句日志,再返回,就可以了。

更改如下:

def log(func):
	def wrapper(*args, **kw):
		print 'end call: %s' %func.__name__
		c = func(*args, **kw)
		print 'begin call: %s' %func.__name__
		return c
	return wrapper

@log
def now():
	print '2018-06-21'

now()

调用的装饰器log中,函数wrapper里将now()函数赋值给C,相当于调用了一次now(),再加上最后的日志,而最后返回的是c,一样是now函数

测试:

成功

-------------------------------------------------------------------------

再思考一下能否写出一个@log的decorator,使它既支持:

@log
def f():
    pass

又支持:

@log('execute')
def f():
    pass

import functools

def log(text = ''):
	def decorator(func):
		def wrapper(*args, **kw):
			print '%s end call: %s' %(text, func.__name__)
			c = func(*args, **kw)
			print '%s begin call: %s' %(text, func.__name__)
			return c
		return wrapper
	return decorator

@log()
def now():
	pass
	
now()

测试:

只需将代码中的@log()后参数换成‘execute’即可测试,结果如下图右,成功。



评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值