(feat) custom ratelimit for configuration provider

Author: chuan137

rate_limit/provider.py

@@ -101,7 +101,10 @@ def get_local_rate_limits(self, scope, action, target_type_uri, **kwargs):
         :param kwargs: optional, additional parameters
         :return: the local rate limit or -1 if not set
         """
-        ttu_ratelimits = self.local_ratelimits.get(target_type_uri, [])
+        _ratelimits = self.custom_ratelimits.get(scope, None)
+        if not _ratelimits:
+            _ratelimits = self.local_ratelimits
+        ttu_ratelimits = _ratelimits.get(target_type_uri, [])
         if not ttu_ratelimits:
             ttu_ratelimits = self._get_wildcard_ratelimits(
                 self.local_ratelimits,
@@ -157,6 +160,9 @@ def read_rate_limits_from_config(self, config_path):
         rates = config.get('rates', {})
         self.global_ratelimits = rates.get('global', {})
         self.local_ratelimits = rates.get('default', {})
+        self.custom_ratelimits = {}
+        for scope, rl in rates.get('custom', {}).items():
+            self.custom_ratelimits[scope] = rl
 
 
 class LimesRateLimitProvider(RateLimitProvider):