python装饰器-限制函数调用次数的方法(10s调用一次)


Posted in Python onApril 21, 2018

这是博主最近一家大公司的面试题,写一个装饰器,限制函数每10s调用一次。当时是笔试的,只写了大概的代码,回来后温习了python装饰器的基础知识,把代码写完了。决定写篇博客记录下。

装饰器分为带参数得装饰器以及不带参数得装饰器。

#不带参数的装饰器
@dec1
@dec2
def func():
  ...
#这个函数声明等价于
func = dec1(dec2(func))
#带参数的装饰器
@dec(some_args)
def func():
  ...
#这个函数声明等价于
func = dec(some_args)(func)

不带参数的装饰器需要注意的一些细节

1. 关于装饰器函数(decorator)本身

因此一个装饰器一般对应两个函数,一个是decorator函数,用来进行一些初始化操作处理,一个是decorated_func用来实现对被装饰的函数func的额外处理。并且为了保持对func的引用,decorated_func一般作为decorator的内部函数

def decorator(func):
  def decorator_func()
    func()
  return decorated_func

decorator函数只在函数声明的时候被调用一次

装饰器实际上是语法糖,在声明函数之后就会被调用,产生decorated_func,并把func符号的引用替换为decorated_func。之后每次调用func函数,实际调用的是decorated_func(这个很重要,装饰之后,其实每次调用的是decorated_func)。

>>> def decorator(func):
...   def decorated_func():
...     func(1)
...   return decorated_func
... 
#声明时就被调用
>>> @decorator
... def func(x):
...   print x
... 
decorator being called 
#使用func()函数实际上使用的是decorated_func函数
>>> func()
1
>>> func.__name__
'decorated_func'

如果要保证返回的decorated_func的函数名与func的函数名相同,应当在decorator函数返回decorated_func之前,加入decorated_func.name = func.name, 另外functools模块提供了wraps装饰器,可以完成这一动作。

#@wraps(func)的操作相当于
#在return decorated_func之前,执行
#decorated_func.__name__ = func.__name__
#func作为装饰器参数传入, 
#decorated_func则作为wraps返回的函数的参数传入
>>> def decorator(func):
...   @wraps(func)
...   def decorated_func():
...     func(1)
...   return decorated_func
... 
#声明时就被调用
>>> @decorator
... def func(x):
...   print x
... 
decorator being called 
#使用func()函数实际上使用的是decorated_func函数
>>> func()
1
>>> func.__name__
'func'

decorator函数局部变量的妙用

因为closure的特性(详见(1)部分闭包部分的详解),decorator声明的变量会被decorated_func.func_closure引用,所以调用了decorator方法结束之后,decorator方法的局部变量也不会被回收,因此可以用decorator方法的局部变量作为计数器,缓存等等。

值得注意的是,如果要改变变量的值,该变量一定要是可变对象,因此就算是计数器,也应当用列表来实现。并且声明一次函数调用一次decorator函数,所以不同函数的计数器之间互不冲突,例如:

#!/usr/bin/env python
#filename decorator.py
def decorator(func):
  #注意这里使用可变对象
  a = [0]
  def decorated_func(*args,**keyargs):
    func(*args, **keyargs)
    #因为闭包是浅拷贝,如果是不可变对象,每次调用完成后符号都会被清空,导致错误
    a[0] += 1
    print "%s have bing called %d times" % (func.__name__, a[0])
  return decorated_func
@decorator
def func(x):
  print x
@decorator
def theOtherFunc(x):
  print x

下面我们开始写代码:

#coding=UTF-8
#!/usr/bin/env python
#filename decorator.py
import time
from functools import wraps
def decorator(func):
  "cache for function result, which is immutable with fixed arguments"
  print "initial cache for %s" % func.__name__
  cache = {}
  @wraps(func)
  def decorated_func(*args,**kwargs):
    # 函数的名称作为key
    key = func.__name__
    result = None
    #判断是否存在缓存
    if key in cache.keys():
      (result, updateTime) = cache[key]
      #过期时间固定为10秒
      if time.time() -updateTime < 10:
        print "limit call 10s", key
        result = updateTime
      else :
        print "cache expired !!! can call "
        result = None
    else:
      print "no cache for ", key
    #如果过期,或则没有缓存调用方法
    if result is None:
      result = func(*args, **kwargs)
      cache[key] = (result, time.time())
    return result
  return decorated_func
@decorator
def func(x):
  print 'call func'

随便测试了下,基本没有问题。

>>> from decorator import func
initial cache for func
>>> func(1)
no cache for func
call func
>>> func(1)
limit call 10s func
1488082913.239092
>>> func(1)
cache expired !!! can call
call func
>>> func(1)
limit call 10s func
1488082923.298204
>>> func(1)
cache expired !!! can call
call func
>>> func(1)
limit call 10s func
1488082935.165979
>>> func(1)
limit call 10s func
1488082935.165979

以上这篇python装饰器-限制函数调用次数的方法(10s调用一次)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python BeautifulSoup中文乱码问题的2种解决方法
Apr 22 Python
用python实现的线程池实例代码
Jan 06 Python
浅谈Django的缓存机制
Aug 23 Python
在Python中实现替换字符串中的子串的示例
Oct 31 Python
Python实现的爬取百度文库功能示例
Feb 16 Python
Python opencv实现人眼/人脸识别以及实时打码处理
Apr 29 Python
python实现复制大量文件功能
Aug 31 Python
Pytorch实现神经网络的分类方式
Jan 08 Python
django rest framework serializer返回时间自动格式化方法
Mar 31 Python
Python求凸包及多边形面积教程
Apr 12 Python
Python的历史与优缺点整理
May 26 Python
Python爬虫之Selenium库的使用方法
Jan 03 Python
对Python中的@classmethod用法详解
Apr 21 #Python
python3+dlib实现人脸识别和情绪分析
Apr 21 #Python
Python通过属性手段实现只允许调用一次的示例讲解
Apr 21 #Python
使用Python和xlwt向Excel文件中写入中文的实例
Apr 21 #Python
使用pandas读取csv文件的指定列方法
Apr 21 #Python
Python 3.7新功能之dataclass装饰器详解
Apr 21 #Python
pandas or sql计算前后两行数据间的增值方法
Apr 20 #Python
You might like
比较详细PHP生成静态页面教程
2012/01/10 PHP
简单谈谈PHP中的Reload操作
2016/12/12 PHP
php7函数,声明,返回值等新特性介绍
2018/05/25 PHP
js 与或运算符 || &amp;&amp; 妙用
2009/12/09 Javascript
使用GruntJS链接与压缩多个JavaScript文件过程详解
2013/08/02 Javascript
jQuery中:reset选择器用法实例
2015/01/04 Javascript
jquery实现炫酷的叠加层自动切换特效
2015/02/01 Javascript
js重写方法的简单实现
2016/07/10 Javascript
Javascript基础学习笔记(菜鸟必看篇)
2016/07/22 Javascript
详解JavaScript的内置对象
2016/12/07 Javascript
浅谈js-FCC算法Friendly Date Ranges(详解)
2017/04/10 Javascript
Angular HMR(热模块替换)功能实现方法
2018/04/04 Javascript
JS实现的JSON数组去重算法示例
2018/04/11 Javascript
JQuery插件tablesorter表格排序实现过程解析
2020/05/28 jQuery
python获得一个月有多少天的方法
2015/06/04 Python
通过mod_python配置运行在Apache上的Django框架
2015/07/22 Python
Django1.7+python 2.78+pycharm配置mysql数据库
2016/10/09 Python
python爬虫基本知识
2018/03/05 Python
python清除字符串前后空格函数的方法
2018/10/21 Python
Python OpenCV读取png图像转成jpg图像存储的方法
2018/10/28 Python
python列表生成器迭代器实例解析
2019/12/19 Python
pandas 中对特征进行硬编码和onehot编码的实现
2019/12/20 Python
django表单中的按钮获取数据的实例分析
2020/07/31 Python
Python pexpect模块及shell脚本except原理解析
2020/08/03 Python
python 实现Harris角点检测算法
2020/12/11 Python
python中doctest库实例用法
2020/12/31 Python
CSS3实现酷炫的3D旋转透视效果
2019/11/21 HTML / CSS
通过canvas转换颜色为RGBA格式及性能问题的解决
2019/11/22 HTML / CSS
Carter’s OshKosh加拿大:购买婴幼儿服装和童装
2018/11/27 全球购物
会计毕业自我鉴定
2014/02/05 职场文书
《赵州桥》教学反思
2014/02/17 职场文书
yy婚礼司仪主持词
2014/03/14 职场文书
学校欢迎标语
2014/06/18 职场文书
机电一体化专业毕业生自荐信
2014/06/19 职场文书
委托证明模板
2014/09/16 职场文书
搭建zabbix监控以及邮件报警的超级详细教学
2022/07/15 Servers