d4rktg 0.5.0__py3-none-any.whl → 0.5.2__py3-none-any.whl

d4rk/Handlers/_bot.py

@@ -12,6 +12,9 @@ from pyrogram.errors.exceptions.bad_request_400 import AccessTokenExpired
 
 from d4rk.Logs import setup_logger
 
+logs_sent = False
+logs_lock = asyncio.Lock()
+
 logger = setup_logger(__name__)
 
 class BotManager(Client):
@@ -98,22 +101,35 @@ class BotManager(Client):
         await self.setup_webserver()
 
     async def powerdown(self, *args):
+        global logs_sent, logs_lock
         logger.info("Initiating APP to stop...")
         if self._rename:await super().set_bot_info(lang_code='en',name=self.app_name + " (Offline)")
         self.stop_scheduler()
         today = self.TZ_now.strftime("%Y-%m-%d")
         if hasattr(self, '_web_runner') and self._web_runner:
             await self.web_server.cleanup()
-        if self._is_connected:
-            logger.info("Stopping bot client...")
+        
+        logger.info("Stopping bot client...")
+        if self._is_connected and self.LOGS:
             try:
-                if self.LOGS:
-                    await self.send_message(chat_id=self.LOGS, text=f"{self._bot_info.mention} stopping...")
+                await self.send_message(chat_id=self.LOGS, text=f"{self._bot_info.mention} stopping...")
+            except Exception as e:
+                logger.error(f"Error sending stop message for {self._bot_info.first_name}: {e}")
+
+        async with logs_lock:
+            if not logs_sent and self._is_connected:
+                logs_sent = True
+                try:
                     await self.send_document(chat_id=self.LOGS, document=f"logs/log-{today}.txt")
-                    await self.send_message(chat_id=self.LOGS, text=f"{self._bot_info.mention} stopped successfully!")
+                    logger.info("Log document sent successfully")
+                except Exception as e:
+                    logger.error(f"Error sending log document: {e}")
+
+        if self._is_connected and self.LOGS:
+            try:
+                await self.send_message(chat_id=self.LOGS, text=f"{self._bot_info.mention} stopped successfully!")
             except Exception as e:
-                logger.error(f"Error sending stop notification: {e}")
-            logger.info(f"{self._bot_info.first_name} - @{self._bot_info.username} Stoped")
+                logger.error(f"Error sending stop confirmation for {self._bot_info.first_name}: {e}")
             await super().stop()
             sys.exit(0)
 

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.dist-info → d4rktg-0.5.2.dist-info}/METADATA

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

{d4rktg-0.5.0.dist-info → d4rktg-0.5.2.dist-info}/RECORD

@@ -2,7 +2,7 @@ d4rk/__init__.py,sha256=Xq5qqX-FwosTJBLrSPF53FPD6fEISH750hbImZpSu5k,202
 d4rk/Database/__init__.py,sha256=TQB5D8PBDCq80jPq6rsC2G939yYYKTh_bCcOWsZ-nA8,18
 d4rk/Database/db.py,sha256=5T-dbHPQp9JF2rQb707SLSSkAaz8ghX4lO7g_Siy7oA,1870
 d4rk/Handlers/__init__.py,sha256=lO1b7Tnu3GWgwcJmX5qepiNqaBBi6qTgjj0SlzVUyEA,63
-d4rk/Handlers/_bot.py,sha256=HVCh_qKAJkVNutzOL3sfG520oVw8KdeSBNUj0NAUcw0,8489
+d4rk/Handlers/_bot.py,sha256=zQgo3NTofTMgDNV2YE-VDbJyq05ajJetuimfd7kR2zQ,9015
 d4rk/Handlers/_scheduler.py,sha256=AyqexO4nxZlIzRfU9vWTfJtTWQVQmP4de7GRPg-3JkA,1236
 d4rk/Logs/__init__.py,sha256=mXWD5jXnyH3_AvS7K_ki3iw5BpoEAvrDFbmr-iEFNnY,22
 d4rk/Logs/_logger.py,sha256=lqfVvCO0vZ_IaGOdIE4HA2KAUQZh7yW2iAHZcBz7F4o,4120
@@ -16,9 +16,9 @@ d4rk/Utils/_fonts.py,sha256=CQsDqPgvp27t3f75Cxod0EmZogEssLwIpyesH-YY5KM,7518
 d4rk/Utils/_ip.py,sha256=KJJW2QSngshIVWCO5YPXF1wj4IPQzVN5oFofpfzlU5w,559
 d4rk/Utils/_movie_parser.py,sha256=QEPd3z04p4pk2vxZ-2fYlRxHmQbmwQd2mCIlBsBM5AY,7039
 d4rk/Utils/_ractions.py,sha256=wOVPyoFnbDuMgoP6NF_gLO1DYcfhERC0trdAK1jWSE8,2170
-d4rk/Utils/_round.py,sha256=lKnkj8OC6rjYGnX5rY9-d6TliR43t2ICImuk2j5NLXk,1868
+d4rk/Utils/_round.py,sha256=jMFTG0eLcL3RVQZrgdSpKhLLus0Nc1WZFgJCET5Fdaw,2100
 d4rk/Utils/_terminal.py,sha256=Anu4OcffY3v6LMOrCskP1cHrJIliomo1Hjownbhh2sQ,125
-d4rktg-0.5.0.dist-info/METADATA,sha256=4cVq1vjwin8rD0zg-LhDjfD07_ZT_C4ixvuU743ULOI,652
-d4rktg-0.5.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-d4rktg-0.5.0.dist-info/top_level.txt,sha256=qs1qTnKWImmGi7E0FoJS0OAEOHoVZA9vHRS3Pm6ncAo,5
-d4rktg-0.5.0.dist-info/RECORD,,
+d4rktg-0.5.2.dist-info/METADATA,sha256=FlotNejt4GX6EIWcQb68jZBehFZ9-JwU-RgvCHsQUrM,652
+d4rktg-0.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+d4rktg-0.5.2.dist-info/top_level.txt,sha256=qs1qTnKWImmGi7E0FoJS0OAEOHoVZA9vHRS3Pm6ncAo,5
+d4rktg-0.5.2.dist-info/RECORD,,