d4rktg 0.5.0__tar.gz → 0.5.1__tar.gz

{d4rktg-0.5.0 → d4rktg-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: d4rktg
-Version: 0.5.0
+Version: 0.5.1
 Summary: A module for create with easy and fast
 Author: D4rkShell
 Author-email: premiumqtrst@gmail.com

d4rktg-0.5.1/VERSION.txt

@@ -0,0 +1 @@
+0.5.1

{d4rktg-0.5.0 → d4rktg-0.5.1}/d4rk/Utils/_round.py

@@ -1,5 +1,6 @@
 from functools import wraps
 from pyrogram.types import Message
+import asyncio
 
 # Track last bot index per chat
 last_index_per_chat = {}
@@ -7,6 +8,8 @@ last_index_per_chat = {}
 bot_order_per_chat = {}
 # Track messages already responded to: chat_id -> set of message_ids
 responded_messages = {}
+# Lock per chat to prevent race conditions
+chat_locks = {}
 
 def round_robin():
     def decorator(func):
@@ -24,26 +27,28 @@ def round_robin():
                 bot_order_per_chat[chat_id] = [client.me.id]
                 last_index_per_chat[chat_id] = 0
                 responded_messages[chat_id] = set()
+                chat_locks[chat_id] = asyncio.Lock()
 
-            # Skip if message already responded by any bot
-            if msg_id in responded_messages[chat_id]:
-                return
-
-            # Add new bots if not in the chat
+            # Add new bot if not in the chat
             if client.me.id not in bot_order_per_chat[chat_id]:
                 bot_order_per_chat[chat_id].append(client.me.id)
 
-            current_index = last_index_per_chat[chat_id]
-            selected_bot_id = bot_order_per_chat[chat_id][current_index]
-
-            # Only selected bot responds
-            if client.me.id == selected_bot_id:
-                result = await func(client, message, *args, **kwargs)
-                # Mark message as responded
-                responded_messages[chat_id].add(msg_id)
-                # Rotate index
-                last_index_per_chat[chat_id] = (current_index + 1) % len(bot_order_per_chat[chat_id])
-                return result
+            async with chat_locks[chat_id]:
+                # Skip if message already responded
+                if msg_id in responded_messages[chat_id]:
+                    return
+
+                # Decide which bot should respond
+                current_index = last_index_per_chat[chat_id]
+                selected_bot_id = bot_order_per_chat[chat_id][current_index]
+
+                if client.me.id == selected_bot_id:
+                    result = await func(client, message, *args, **kwargs)
+                    # Mark message as responded
+                    responded_messages[chat_id].add(msg_id)
+                    # Rotate for next message
+                    last_index_per_chat[chat_id] = (current_index + 1) % len(bot_order_per_chat[chat_id])
+                    return result
 
         return wrapper
     return decorator

{d4rktg-0.5.0 → d4rktg-0.5.1}/d4rktg.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: d4rktg
-Version: 0.5.0
+Version: 0.5.1
 Summary: A module for create with easy and fast
 Author: D4rkShell
 Author-email: premiumqtrst@gmail.com

d4rktg-0.5.0/VERSION.txt

@@ -1 +0,0 @@
-0.5.0