python 限制函數(shù)調(diào)用次數(shù)的實(shí)例講解

發(fā)布時(shí)間：2020-10-08 02:04:02 來(lái)源：腳本之家閱讀：357 作者：隨便起個(gè)名字啊欄目：開(kāi)發(fā)技術(shù)

如下代碼，限制某個(gè)函數(shù)在某個(gè)時(shí)間段的調(diào)用次數(shù)，

靈感來(lái)源：python裝飾器-限制函數(shù)調(diào)用次數(shù)的方法(10s調(diào)用一次) 歡迎訪問(wèn)

原博客中指定的是緩存，我這里換成限制訪問(wèn)次數(shù)，異曲同工

#newtest.py
#!/usr/bin/env python
#-*- coding:utf-8 -*-
import time
def stat_called_time(func):
 cache={}
 limit_times=[10]
 def _called_time(*args,**kwargs):
  key=func.__name__
  if key in cache.keys():
   [call_times,updatetime]=cache[key]
   if time.time()-updatetime <60:
    cache[key][0]+=1
   else:
    cache[key]=[1,time.time()]
  else:
   call_times=1
   cache[key]=[call_times,time.time()]
  print('調(diào)用次數(shù): %s' % cache[key][0])
  print('限制次數(shù): %s' % limit_times[0])
  if cache[key][0] <= limit_times[0]:
   res=func(*args,**kwargs)
   cache[key][1] = time.time()
   return res
  else:
   print("超過(guò)調(diào)用次數(shù)了")
   return None
 return _called_time
@stat_called_time
def foo():
 print("I'm foo")
if __name__=='__main__':
 for i in range(10):
  foo()

#test.py
from newtest import foo
import time
for i in range(30):
 foo()
print('*'*20)
foo()
foo()
print('*'*20)
for i in range(60):
 print(i)
 time.sleep(1)
for i in range(11):
 foo()

以上這篇python 限制函數(shù)調(diào)用次數(shù)的實(shí)例講解就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問(wèn)一下細(xì)節(jié)

python 限制函數(shù)調(diào)用次數(shù)的實(shí)例講解

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽