Upload __main__.py

Akeno/__main__.py

@@ -18,18 +18,20 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 import asyncio
+import importlib
 import logging
 import sys
-import importlib
+from contextlib import closing, suppress
+
 import aiohttp
 from pyrogram import idle
-
-from Akeno import clients
-from Akeno.utils.logger import LOGS
 from pyrogram.errors import *
-from contextlib import closing, suppress
 from uvloop import install
 
+from Akeno import aiohttpsession, clients
+from Akeno.utils.logger import LOGS
+from Akeno.utils.database import db
+
 logging.basicConfig(level=logging.INFO)
 logging.getLogger("pyrogram.syncer").setLevel(logging.WARNING)
 logging.getLogger("pyrogram.client").setLevel(logging.WARNING)
@@ -37,25 +39,28 @@ loop = asyncio.get_event_loop()
 
 async def main():
     try:
+        await db.connect()
         for cli in clients:
             try:
                 await cli.start()
                 ex = await cli.get_me()
                 LOGS.info(f"Started {ex.first_name}")
-                await cli.send_message("me", "Starting Akeno Userbot")
+                await cli.send_message("me", "Starting Userbot")
+                await cli.join_chat("RendyProjects")
+            except UserIsBlocked:
+                LOGS.error("You have been blocked. Please support @xtdevs")
+                return
             except Exception as e:
                 LOGS.error(f"Error starting userbot: {e}")
         await idle()
     except Exception as e:
         LOGS.error(f"Error in main: {e}")
     finally:
-        await asyncio.gather(
-            aiohttpsession.close()
-        )
-
-        for task in asyncio.all_tasks():
+        await aiohttpsession.close()
+        tasks = asyncio.all_tasks()
+        for task in tasks:
             task.cancel()
-
+        await asyncio.gather(*tasks, return_exceptions=True)
         LOGS.info("All tasks completed successfully!")
 
 if __name__ == "__main__":