装饰器
装饰器是一个返回函数的高阶函数。
装饰器常见用法:
打印日志
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
(2, 'a', 'b', 'c')
关键字参数(**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'}
将函数作为参数传递给另一个函数
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