python装饰器和函数传参

装饰器

装饰器是一个返回函数的高阶函数。

• 装饰器常见用法：
  打印日志

def logger(func):
	def wrapper(*args, **kw):
		print 'do {}'.format(func.__name__)

		func(*args, **kw)

		print 'finish'
	return wrapper

@logger
def add(x,y):
	print '{} + {} = {}'.format(x,y,x+y)

add(3,5)

在函数执行前，打印一行日志do...；函数执行结束，打印一行日志finish。执行结果如下：

do add
3 + 5 = 8
finish

计算时间

import time

def timer(func):
	def wrapper(*args, **kw):
		t1 = time.time()
		func(*args,**kw)
		t2 = time.time()

		cost_time = t2 - t1
		print 'cost time: {} s'.format(cost_time)

	return wrapper

@timer
def cost_time(sleep_time):
	time.sleep(sleep_time)

cost_time(10)

• 带参数的函数装饰器

def say_hello(country):
	def wrapper(func):
		def decorate(*args,**kw):
			if country == 'en':
				print 'hello'
			elif country == 'usa':
				print 'hi'
			else:
				return
			func(*args,**kw)
		return decorate
	return wrapper

@say_hello("usa")
def usa():
	print 'i am from usa'

@say_hello("en")
def en():
	print 'i am from england'

usa()
print '----------------------'
en()

装饰器本身是一个函数，使用两层嵌套传参，执行结果如下：

hi
i am from usa
----------------------
hello
i am from england

• 不带参数的类装饰器
  基于类装饰器的实现，必须实现__call__和__init__两个内置函数。
  __init__ ：接收被装饰函数
  __call__ ：实现装饰逻辑

class logger(object):
	def __init__(self,func):
		self.func = func

	def __call__(self,*args,**kwargs):
		print 'the function {}() is running...'
			.format(self.func.__name__)
		return self.func(*args,**kwargs)

@logger
def say(something):
	print 'say {}!'.format(something)

say('hello')

运行结果如下：

the function say() is running...
say hello!

• 带参数的类装饰器
  带参数和不带参数的类装饰器有很大的不同。
  __init__ ：不再接收被装饰函数，而是接收传入参数
  __call__ ：接收被装饰函数，实现装饰逻辑

class logger(object):
	def __init__(self,level='INFO'):
		self.level = level

	def __call__(self,func):
		def wrapper(*args,**kwargs):
			print '{level}: the function {func} () is running...'
				.format(level=self.level, func=func.__name__)

			func(*args,**kwargs)
		return wrapper

@logger(level='WARNING')
def say(something):
	print 'say {}!'.format(something)

say('hello')

运行结果如下：

WARNING: the function say () is running...
say hello!

函数的参数

• 位置参数

def power(x, n):
    s = 1
    while n > 0:
        n = n - 1
        s = s * x
    return s

power(x, n)函数有两个参数：x和n，这两个参数都是位置参数，调用函数时，传入的两个值按照位置顺序依次赋值给参数x和n。

• 默认参数

def power(x, n=2):
    s = 1
    while n > 0:
        n = n - 1
        s = s * x
    return s

power(x, n)函数有两个参数：x和n，如果想在不传入n值时，默认计算x的平方，此时可以将n设为默认值2。

• 可变参数(*args)

def function(f_arg, *args):
	print f_arg, type(f_arg)
	print args, type(args)
nums = ['a','b','c']
function(1,2,*nums)

定义可变参数时，需要在参数前面加一个*号，可变参数的个数是可变的。在函数内部，参数*args接收到的是一个tuple。输出结果如下：

1 <type 'int'>
(2, 'a', 'b', 'c') <type 'tuple'>

• 关键字参数(**kwargs)

def person(name,age,**kwargs):
	print 'name:',name,'age:',age,'other:',kwargs,type(kwargs)

person('mark',30,city='shanghai')

**kwargs允许将不定长度的键值对，作为参数传递给一个函数，关键字参数在函数内部自动组装为一个dict。输出结果如下：

name: mark age: 30 other: {'city': 'shanghai'} <type 'dict'>

• 将函数作为参数传递给另一个函数

def hi():
	return 'hi friends'

def function(func):
	print 'just test'
	print func()

function(hi)

function()函数将hi函数作为参数接收，输出结果如下：

just test
hi friends

time模块

• 获取当前时间

>>> time.localtime()
time.struct_time(tm_year=2019, tm_mon=8, tm_mday=21, tm_hour=14, tm_min=31, tm_sec=18, tm_wday=2, tm_yday=233, tm_isdst=0)

• 获取格式化的时间

>>> time.ctime()
'Wed Aug 21 14:51:28 2019'
>>> time.asctime()
'Wed Aug 21 14:51:34 2019'

• 格式化日期

>>> time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
'2019-08-21 14:35:02'
>>> time.strftime('%a %b %d %H:%M:%S %Y',time.localtime())
'Wed Aug 21 14:36:09 2019'

• 计算运行时间

import time
start = time.time()
time.sleep(2)
end = time.time()
print end-start