From 8ac821fcc46020a51fd45949f026de4a8d6f466e Mon Sep 17 00:00:00 2001 From: Martyn Inglis Date: Fri, 12 May 2017 16:10:00 +0100 Subject: [PATCH] Fixed import paths --- app/notifications/validators.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/app/notifications/validators.py b/app/notifications/validators.py index e601ec6be..a680f446e 100644 --- a/app/notifications/validators.py +++ b/app/notifications/validators.py @@ -10,11 +10,12 @@ from app.models import KEY_TYPE_TEST, KEY_TYPE_TEAM, SMS_TYPE from app.service.utils import service_allowed_to_send_to from app.v2.errors import TooManyRequestsError, BadRequestError, RateLimitError from app import redis_store +from notifications_utils.clients.redis import rate_limit_cache_key, daily_limit_cache_key def check_service_over_api_rate_limit(service, api_key): if current_app.config['API_RATE_LIMIT_ENABLED']: - cache_key = redis_store.rate_limit_cache_key(service.id, api_key.key_type) + cache_key = rate_limit_cache_key(service.id, api_key.key_type) rate_limit = current_app.config['API_KEY_LIMITS'][api_key.key_type]['limit'] interval = current_app.config['API_KEY_LIMITS'][api_key.key_type]['interval'] if redis_store.exceeded_rate_limit(cache_key, rate_limit, interval): @@ -24,7 +25,7 @@ def check_service_over_api_rate_limit(service, api_key): def check_service_over_daily_message_limit(key_type, service): if key_type != KEY_TYPE_TEST: - cache_key = redis_store.daily_limit_cache_key(service.id) + cache_key = daily_limit_cache_key(service.id) service_stats = redis_store.get(cache_key) if not service_stats: service_stats = services_dao.fetch_todays_total_message_count(service.id)