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双向链表实现实例代码
Nov 21 Python
Django接受前端数据的几种方法总结
Nov 04 Python
Python+MongoDB自增键值的简单实现
Nov 04 Python
python自带的http模块详解
Nov 06 Python
用tensorflow搭建CNN的方法
Mar 05 Python
Python实现使用request模块下载图片demo示例
May 24 Python
python同时替换多个字符串方法示例
Sep 17 Python
Python List列表对象内置方法实例详解
Oct 22 Python
Python类继承和多态原理解析
Feb 05 Python
Jupyter加载文件的实现方法
Apr 14 Python
Pytorch通过保存为ONNX模型转TensorRT5的实现
May 25 Python
使用bandit对目标python代码进行安全函数扫描的案例分析
Jan 27 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实现用户认证及管理完全源码
2007/03/11 PHP
PHP5中使用PDO连接数据库的方法
2010/08/01 PHP
PHP代码审核的详细介绍
2013/06/13 PHP
PHP中大括号'{}'用法实例总结
2017/02/08 PHP
php封装的page分页类完整实例代码
2020/02/01 PHP
ASP中用Join和Array,可以加快字符连接速度的代码
2007/08/22 Javascript
CCPry JS类库 代码
2009/10/30 Javascript
跨浏览器的 mouseenter mouseleave 以及 compareDocumentPosition的使用说明
2010/05/04 Javascript
JS限制文本框只能输入数字和字母方法
2015/02/28 Javascript
JavaScript动态插入CSS的方法
2015/12/10 Javascript
基于jQuery实现仿搜狐辩论投票动画代码(附源码下载)
2016/02/18 Javascript
使用Vue.js创建一个时间跟踪的单页应用
2016/11/28 Javascript
原生Javascript插件开发实践
2017/01/09 Javascript
JavaScript拖动层Div代码
2017/03/01 Javascript
vue2.0+koa2+mongodb实现注册登录
2018/04/10 Javascript
Layui数据表格 前后端json数据接收的方法
2019/09/19 Javascript
Vant picker 多级联动操作
2020/11/02 Javascript
JavaScript实现点击切换验证码及校验
2021/01/10 Javascript
[01:24:34]2014 DOTA2华西杯精英邀请赛5 24 DK VS LGD
2014/05/25 DOTA
Python的Tornado框架实现异步非阻塞访问数据库的示例
2016/06/30 Python
python版DDOS攻击脚本
2019/06/12 Python
python使用 cx_Oracle 模块进行查询操作示例
2019/11/28 Python
Python 日期的转换及计算的具体使用详解
2020/01/16 Python
python时间日期操作方法实例小结
2020/02/06 Python
PyCharm最新激活码(2020/10/27全网最新)
2020/10/27 Python
CSS中垂直居中的简单实现方法
2015/07/06 HTML / CSS
video.js支持m3u8格式直播的实现示例
2020/05/20 HTML / CSS
澳大利亚在线时尚精品店:Hello Molly
2018/02/26 全球购物
Chain Reaction Cycles俄罗斯:世界上最大的在线自行车商店
2019/08/27 全球购物
情人节活动策划方案
2014/02/27 职场文书
探亲假请假条
2014/04/11 职场文书
公司晚会策划方案
2014/05/17 职场文书
2015年度企业工作总结
2015/05/21 职场文书
《世界多美呀》教学反思
2016/02/22 职场文书
表扬信范文
2019/04/22 职场文书
MongoDB连接数据库并创建数据等使用方法
2021/11/27 MongoDB