X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=aspects%2Fratelimitaspects.py;h=8c28dd599ea4fdcba60dc8cdca0de70f8ed53e90;hb=1c92c333ad668b173ba5abba552464ffaf990fe7;hp=8be22e59343fab212425150f5dc97451e6d01a82;hpb=2241d4ad029eb27176ef792efa2b9faca6653843;p=plcapi.git diff --git a/aspects/ratelimitaspects.py b/aspects/ratelimitaspects.py index 8be22e5..8c28dd5 100644 --- a/aspects/ratelimitaspects.py +++ b/aspects/ratelimitaspects.py @@ -45,8 +45,11 @@ class BaseRateLimit(object): keys_to_check = ["%s_%s_%s" % (self.prefix, api_method_source[0], (now - timedelta(minutes = minute)).strftime("%Y%m%d%H%M")) for minute in range(self.minutes + 1)] try: - mc.incr(current_key) + value = mc.incr(current_key) except ValueError: + value = None + + if value == None: mc.set(current_key, 1, time=self.expire_after) result = mc.get_multi(keys_to_check)