hotfix(tgbot): add api connect with proxy

Author: 0rtemk

tgbot/uni_eduplatform_bot.py

@@ -10,6 +10,15 @@
 # Load .env
 load_dotenv()
 
+SOCKS5_PROXY = os.getenv('SOCKS5_PROXY')
+
+proxies = None
+if SOCKS5_PROXY:
+    proxies = {
+        'http': SOCKS5_PROXY,
+        'https': SOCKS5_PROXY,
+    }
+
 TOKEN = os.getenv('TG_BOT_TOKEN')
 TEACHER_CODE = os.getenv('TEACHER_CODE')
 PLATFORM_ADDRESS_SERVER = os.getenv('PLATFORM_ADDRESS_SERVER')
@@ -21,7 +30,13 @@
 if not PLATFORM_ADDRESS_SERVER:
     raise RuntimeError('PLATFORM_ADDRESS_SERVER not found in environment (.env: PLATFORM_ADDRESS_SERVER)')
 
-bot = telebot.TeleBot(TOKEN)
+if SOCKS5_PROXY:
+    bot = telebot.TeleBot(
+        TOKEN,
+        proxy=SOCKS5_PROXY
+    )
+else:
+    bot = telebot.TeleBot(TOKEN)
 
 # In-memory registration/change-password state per chat
 states: Dict[int, Dict[str, Any]] = {}
@@ -115,7 +130,7 @@ def graphql_query(query: str, variables: dict = None) -> dict:
 	if variables is not None:
 		payload['variables'] = variables
 	try:
-		r = requests.post(url, json=payload, headers=headers, timeout=10)
+		r = requests.post(url, json=payload, headers=headers, timeout=10, proxies=proxies)
 		try:
 			r.raise_for_status()
 		except requests.HTTPError as http_err:
@@ -389,8 +404,4 @@ def registration_flow(message):
 
 
 if __name__ == '__main__':
-	while True:
-		try:
-			bot.polling(none_stop = True)
-		except Exception as e:
-			time.sleep(2)
+	bot.polling()