platform

main.py

@@ -3,129 +3,40 @@ import sys
 from random import randrange, choice
 
 import telebot
-from cachetools import TTLCache
-from requests import get
 from telebot.types import Message
 
 from mongo import mongo
+from platform import PlatformClient
 from storage import set_values, get_chat_info
 
-
-DEFAULT_ANSWERS = {
-    "a": "Хуй на!",
-    "da": "Pizda!",
-    "а": "Хуй на!",
-    "200": "Отсоси на месте!",
-    "300": "Отсоси у тракториста!",
-    "dа": "Pizda!",
-    "gde": "V pizde!",
-    "net": "Pidora otvet!",
-    "дa": "Пизда!",
-    "чe": "Хуй через плечо!",
-    "чo": "Хуй через плечо!",
-    "gdе": "V pizde!",
-    "heт": "Пидора ответ!",
-    "nеt": "Pidora otvet!",
-    "да": "Пизда!",
-    "че": "Хуй через плечо!",
-    "чо": "Хуй через плечо!",
-    "чё": "Хуй через плечо!",
-    "hет": "Пидора ответ!",
-    "гдe": "В пизде!",
-    "нeт": "Пидора ответ!",
-    "ага": "В жопе нога!",
-    "где": "В пизде!",
-    "как": "Жопой об косяк!",
-    "кто": "Конь в пальто!",
-    "нет": "Пидора ответ!",
-    "ога": "В жопе нога!",
-    "опа": "Срослась пизда и жопа!",
-    "угу": "Иди в пизду!",
-    "слыш": "За углом поссышь!",
-    "ладно": "Прохладно!",
-    "слышь": "За углом поссышь!",
-    "двести": "Отсоси на месте!",
-    "триста": "Отсоси у тракториста!",
-    "здрасте": "Пизду покрасьте!",
-    "здрасьте": "Пизду покрасьте!"
-}
-
-DEFAULT_REPLIES = [
-    "Пошел нахуй!",
-    "Я твою маму ебал!",
-    "Ты пидор!",
-]
-
 bot = telebot.TeleBot(os.getenv("TELEGRAM_TOKEN"))
 security_token = os.getenv("PLATFORM_SECURITY_TOKEN")
-stage = os.getenv("STAGE")
+stage = os.getenv("STAGE", 'local')
+
+
+platform = PlatformClient(
+    security_token,
+    'Pizda Bot',
+    stage,
+    ['constants', 'answers', 'replies'],
+    [],
+    need_poll=True,
+)
 
 
 all_letters = "йцукенгшщзхъёфывапролджэячсмитьбюЙЦУКЕНГШЩЗХЪЁФЫВАПРОЛДЖЭЯЧСМИТЬБЮQWERTYUIOPASDFGHJKLZXCVBNMqwertyuiopasdfghjklzxcvbnm1234567890 "
 
 
-CACHE_CONFIG = TTLCache(1, 60)
-SELF_NAME = None
-
-
 def get_self_name():
-    global SELF_NAME
-    if SELF_NAME:
-        return SELF_NAME
-    answers_response = get(
-        'https://platform.sprinthub.ru/configs/get',
-        headers={'X-Security-Token': security_token},
-        params={
-            'project': "Pizda Bot",
-            'stage': stage,
-            'name': "constants"
-        }
-    )
-    if answers_response.status_code != 200:
-        print("ERROR")
-        return 'answer_da_bot'
-    SELF_NAME = answers_response.json()['self_name']
-    return SELF_NAME
+    return platform.get_config('constants')['self_name']
 
 
 def get_answers():
-    answers = CACHE_CONFIG.get('answers')
-    if answers is None:
-        answers_response = get(
-            'https://platform.sprinthub.ru/configs/get',
-            headers={'X-Security-Token': security_token},
-            params={
-                'project': "Pizda Bot",
-                'stage': stage,
-                'name': "answers"
-            }
-        )
-        if answers_response.status_code != 200:
-            print("ERROR")
-            return DEFAULT_ANSWERS
-        answers = answers_response.json()
-        CACHE_CONFIG['answers'] = answers
-    return answers
+    return platform.get_config('answers')
 
 
 def get_replies():
-    answers = CACHE_CONFIG.get('replies')
-    if answers is None:
-        answers_response = get(
-            'https://platform.sprinthub.ru/configs/get',
-            headers={'X-Security-Token': security_token},
-            params={
-                'project': "Pizda Bot",
-                'stage': stage,
-                'name': "replies"
-            }
-        )
-        if answers_response.status_code != 200:
-            print("ERROR")
-            return DEFAULT_REPLIES
-        answers = answers_response.json()
-        CACHE_CONFIG['answers'] = answers
-    return answers
+    return platform.get_config('replies')
 
 
 @bot.message_handler(commands=['setprobability'])