X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=aspects%2Fratelimitaspects.py;fp=aspects%2Fratelimitaspects.py;h=8be22e59343fab212425150f5dc97451e6d01a82;hb=2241d4ad029eb27176ef792efa2b9faca6653843;hp=78548c2365c925f018ec2acbee5bf22604aa6606;hpb=2a525d30691f94fa66dcfcb350bd9e979260de6b;p=plcapi.git diff --git a/aspects/ratelimitaspects.py b/aspects/ratelimitaspects.py index 78548c2..8be22e5 100644 --- a/aspects/ratelimitaspects.py +++ b/aspects/ratelimitaspects.py @@ -23,6 +23,8 @@ class BaseRateLimit(object): self.requests = 50 # Number of allowed requests in that time period self.expire_after = (self.minutes + 1) * 60 + self.whitelist = [] + def before(self, wobj, data, *args, **kwargs): # ratelimit_128.112.139.115_201011091532 = 1 # ratelimit_128.112.139.115_201011091533 = 14 @@ -33,7 +35,7 @@ class BaseRateLimit(object): api_method_name = wobj.name api_method_source = wobj.source - if api_method_source == None or api_method_source[0] == self.config.PLC_API_IP: + if api_method_source == None or api_method_source[0] == self.config.PLC_API_IP or api_method_source[0] in self.whitelist: return mc = memcache.Client(["%s:11211" % self.config.PLC_API_HOST])