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

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,30 @@ 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
+        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
+
+        can_perform, user_data = self.check_limit(user_id)
+        if can_perform:
+            self.increment_action_count(user_id, user_data)
+            return True
+        return False
 
-    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.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: telegram_libs
-Version: 0.1.27
+Version: 0.1.28
 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.28.dist-info}/RECORD

@@ -10,8 +10,8 @@ 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=--X13_1iqIEKydwie7_Ak0V0rO04XHtKNK1DHrzGelY,4197
+telegram_libs-0.1.28.dist-info/LICENSE,sha256=ZXkWPZbCc61L29Gz6ZHPwn1c4Pm0TnfIqtx8jGWi9F4,1069
+telegram_libs-0.1.28.dist-info/METADATA,sha256=FpuMWElD6Lc-wVVP8U9Etg9gBI6my4oXu7EzkUjT9Pc,804
+telegram_libs-0.1.28.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+telegram_libs-0.1.28.dist-info/RECORD,,