fix(message): Fixed rate limiter

Nickelza · Sep 16, 2023 · e569712 · e569712
1 parent 0628c94
commit e569712
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -5,7 +5,7 @@
 import pytz
 from telegram import Update
 from telegram.ext import Application, CommandHandler, MessageHandler, filters, Defaults, CallbackQueryHandler, \
-    ContextTypes, BaseRateLimiter
+    ContextTypes, AIORateLimiter
 
 import constants as c
 import resources.Environment as Env
@@ -51,6 +51,9 @@ def main() -> None:
         logger.addHandler(logging.StreamHandler())
         logger.setLevel(logging.DEBUG)
 
+    # Disable httpx logging info
+    logging.getLogger('httpx').setLevel(logging.WARNING)
+
     logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                         level=logging.INFO, stream=sys.stdout)
 
@@ -61,7 +64,7 @@ def main() -> None:
                    .token(Env.BOT_TOKEN.get())
                    .post_init(post_init)
                    .defaults(defaults)
-                   .rate_limiter(BaseRateLimiter())
+                   .rate_limiter(AIORateLimiter())
                    .build())
 
     # Chat id handler

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 python-dotenv==1.0.0
-python-telegram-bot==20.4
+python-telegram-bot[rate-limiter]==20.4
 peewee==3.16.3
 asyncpraw~=7.7.1
 APScheduler==3.10.4