deny for development

daemons/bot.py

@@ -1,13 +1,37 @@
 import os
 
 import telebot
+from cachetools import TTLCache
+from requests import get
 from telebot.types import Message
 
+import settings
 from helpers.mongo import mongo
 
 bot = telebot.TeleBot(os.getenv("TELEGRAM_TOKEN"))
 
 
+STAFF_CACHE = TTLCache(100, 60)
+
+
+def is_staff(telegram_id):
+    print('there')
+    staff_data = STAFF_CACHE.get(telegram_id)
+    if staff_data is None:
+        staff_data = get(
+            'https://platform.sprinthub.ru/is_staff',
+            headers={'X-Security-Token': settings.PLATFORM_SECURITY_TOKEN},
+            params={
+                'telegram_id': telegram_id
+            }
+        )
+        if staff_data.status_code != 200:
+            return False
+        staff_data = staff_data.json()['is_staff']
+        STAFF_CACHE[telegram_id] = staff_data
+    return staff_data
+
+
 @bot.message_handler(commands=['start'])
 def on_start(message: Message):
     mongo.users_collection.delete_many({"chat_id": message.chat.id})
@@ -16,5 +40,7 @@ def on_start(message: Message):
 
 @bot.message_handler()
 def do_action(message: Message):
+    if settings.STAGE == 'development' and not is_staff(message.chat.id):
+        return
     from helpers.answer import Answer
     Answer(message).process()