interval from settings

UnitapApp · May 20, 2024 · ad3bdb7 · ad3bdb7
1 parent 27d8eca
commit ad3bdb7
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/brightIDfaucet/settings.py b/brightIDfaucet/settings.py
@@ -74,6 +74,7 @@ def str2bool(v):
 TELEGRAM_CHANNEL_ID = os.environ.get("TELEGRAM_CHANNEL_ID")
 TELEGRAM_API_TOKEN = os.environ.get("TELEGRAM_API_TOKEN")
 
+
 assert DEPLOYMENT_ENV in ["dev", "main"]
 
 
@@ -264,3 +265,5 @@ def before_send(event, hint):
     ),
 }
 CELERY_BROKER_URL = REDIS_URL
+
+TELEGRAM_MIN_LOG_INTERVAL = 10 # seconds
diff --git a/core/telegram.py b/core/telegram.py
@@ -6,6 +6,7 @@
 
 API_TOKEN = settings.TELEGRAM_API_TOKEN
 CHANNEL_ID = settings.TELEGRAM_CHANNEL_ID
+MIN_INTERVAL = settings.TELEGRAM_MIN_LOG_INTERVAL
 
 
 BASE_URL = f"https://api.telegram.org/bot{API_TOKEN}"
@@ -27,7 +28,7 @@ def send_telegram_log(text):
 
 
 log_cache = defaultdict(int)
-MIN_INTERVAL = 10
+
 
 
 class LogMiddleware(MiddlewareMixin):

diff --git a/core/tests.py b/core/tests.py
@@ -9,6 +9,9 @@
 
 from django.test import TestCase
 from core.telegram import LogMiddleware
+from django.conf import settings
+
+TELEGRAM_MIN_LOG_INTERVAL = settings.TELEGRAM_MIN_LOG_INTERVAL
 
 from .constraints import (
     Attest,
@@ -314,8 +317,8 @@ def test_log_message_sent_to_telegram(self):
         self.assertEqual(res['ok'],  True)
         res = self.middleware.log_message("Test log message")
         self.assertEqual(res['ok'],  False)
-        # delay 11 seconds
-        time.sleep(11)
+        # delay
+        time.sleep(TELEGRAM_MIN_LOG_INTERVAL)
         res = self.middleware.log_message("Test log message")
         self.assertEqual(res['ok'],  True)
         res = self.middleware.log_message("Test log message")