telegram_libs 0.1.27__py3-none-any.whl → 0.1.29__py3-none-any.whl

telegram_libs/locales/en.json

@@ -14,5 +14,8 @@
   "support": {
     "message": "Write down any questions, issues or suggestions you have, and we will resolve them as soon as possible 👇 ",
     "response": "Thank you! Our support team will contact you soon."
+  },
+  "rate_limit": {
+    "exceeded": "You have exceeded the free rate limit for today. Please try again tomorrow."
   }
 }

telegram_libs/locales/ru.json

@@ -14,5 +14,8 @@
   "support": {
     "message": "Напишите любые вопросы, проблемы, или предложения и мы решим их как можно скорее 👇 ",
     "response": "Спасибо! Наша служба поддержки свяжется с вами в ближайшее время."
+  },
+  "rate_limit": {
+    "exceeded": "Вы превысили лимит бесплатных запросов на сегодня. Пожалуйста, попробуйте снова завтра."
   }
 }

telegram_libs/utils.py

@@ -68,12 +68,8 @@ class RateLimitManager:
         self.mongo_manager = mongo_manager
         self.rate_limit = rate_limit
 
-    def check_limit(self, user_id: int) -> bool:
+    def check_limit(self, user_id: int) -> tuple[bool, dict]:
         """Check if user has exceeded the daily rate limit."""
-        # Premium users have no limits
-        if self.mongo_manager.check_subscription_status(user_id):
-            return True
-
         # Get today's date and reset time to midnight
         today = datetime.now().date()
 
@@ -90,18 +86,38 @@ class RateLimitManager:
                         "last_action_date": datetime.now().isoformat(),
                     },
                 )
-                return True
+                return True, user_data
 
         # Check if user has exceeded the limit
         actions_today = user_data.get("actions_today", 0)
         if actions_today >= self.rate_limit:
-            return False
+            return False, user_data
+
+        return True, user_data
+    
+    def check_and_increment(self, user_id: int) -> bool:
+        """Check if user can perform an action and increment the count if allowed."""
+        if self.mongo_manager.check_subscription_status(user_id):
+            return True
 
-        return True
+        can_perform, user_data = self.check_limit(user_id)
+        if can_perform:
+            self.increment_action_count(user_id, user_data)
+            return True
+        return False
+    
+    async def check_limit_with_response(self, update: Update, context: ContextTypes.DEFAULT_TYPE, user_id: int) -> None:
+        """Check if user can perform an action and handle the response."""
+        if not self.check_and_increment(user_id):
+            lang = self.mongo_manager.get_user_info(user_id).get("lang")
+            message = t("rate_limit.exceeded", lang, common=True)
+            await update.message.reply_text(message)
+            await get_subscription_keyboard(update, lang)
 
-    def increment_action_count(self, user_id: int) -> None:
+    def increment_action_count(self, user_id: int, user_data: dict = None) -> None:
         """Increment the daily action count for the user."""
-        user_data = self.mongo_manager.get_user_data(user_id)
+        if user_data is None:
+            user_data = self.mongo_manager.get_user_data(user_id)
         current_actions = user_data.get("actions_today", 0)
         self.mongo_manager.update_user_data(
             user_id,

{telegram_libs-0.1.27.dist-info → telegram_libs-0.1.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: telegram_libs
-Version: 0.1.27
+Version: 0.1.29
 Summary: Common libraries for Telegram bots
 Author: Andrey Gritsaenko gricaenko.95a@gmail.com
 Requires-Python: >=3.11,<4.0

{telegram_libs-0.1.27.dist-info → telegram_libs-0.1.29.dist-info}/RECORD

@@ -2,16 +2,16 @@ telegram_libs/__init__.py,sha256=xrsD5r6ZiJxPapHf1UhQ61z2gHtCCWrzW0CZHvlvXRc,82
 telegram_libs/constants.py,sha256=mATa2Pr4YzPGNrqw29kOHK7sSefWvENZ-byPPZCBTc8,958
 telegram_libs/error.py,sha256=uomabeEiSP4v4AEpKYbi_gR0l3G003sio6lKl72AinY,453
 telegram_libs/handlers.py,sha256=d5-4FvjyWcJeN1i3dHTYhF-2P4j7PbM1QElloFR_llg,2417
-telegram_libs/locales/en.json,sha256=4VVkME7lCKW6ZjkopaoA6Uq3DLbEVkWacGei3GNSXFM,843
-telegram_libs/locales/ru.json,sha256=LYI6rAlwdNLaTGgkrALn31Lt7jC3SZteei8P_i7ZpQI,1208
+telegram_libs/locales/en.json,sha256=5GT1mk1xskau1oaku0E6tZ7-aGf3F1qwkJzD_bN0uV8,960
+telegram_libs/locales/ru.json,sha256=KvlxUUpwNyiOtDBhUm3U13Gj7UtzLKX5SzHOdkPT6gY,1411
 telegram_libs/logger.py,sha256=oikvkZMk2kOLvQI9hVMUGMCoSnOPf_fhMvv2SbO-STo,891
 telegram_libs/mongo.py,sha256=j1UIGnldvaNZaFPltCvEUcUOOCoquBWvqPh7Q2HTrvY,4776
 telegram_libs/payment.py,sha256=cNb2vnBlYI_6KnTGz-PCM0uKahW2uGjHVn1mGYlYG-4,3564
 telegram_libs/subscription.py,sha256=NJlh1kcDfasX1XLpkVfe1UbZslpWSC1hXHaum0kfG8A,4693
 telegram_libs/support.py,sha256=a3BA7g3seVBUMasv65SzxebLayLigA069wvDcStYbCM,2748
 telegram_libs/translation.py,sha256=8Kb2cgqKKZH4X_i2Le0V_K1imZdoaCzYAca831DOBig,2049
-telegram_libs/utils.py,sha256=YnxeZFBLEb95swNlER6Vulv3-TXXVgUTGmzpYUROVY0,3796
-telegram_libs-0.1.27.dist-info/LICENSE,sha256=ZXkWPZbCc61L29Gz6ZHPwn1c4Pm0TnfIqtx8jGWi9F4,1069
-telegram_libs-0.1.27.dist-info/METADATA,sha256=DhniNOJ1-RW0DhJNn2QvH6lWL0tH56SvRNMrJPVUbOw,804
-telegram_libs-0.1.27.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-telegram_libs-0.1.27.dist-info/RECORD,,
+telegram_libs/utils.py,sha256=Y-WwncNugCbmzQjvDg95NEI-bHRIX9Ly6SVuySJjpjo,4698
+telegram_libs-0.1.29.dist-info/LICENSE,sha256=ZXkWPZbCc61L29Gz6ZHPwn1c4Pm0TnfIqtx8jGWi9F4,1069
+telegram_libs-0.1.29.dist-info/METADATA,sha256=Qi0yn7gicb9WnTwCqyF_EPSZWXCZqEt14vp5Caoi8hA,804
+telegram_libs-0.1.29.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+telegram_libs-0.1.29.dist-info/RECORD,,