Upload database.py

akn/utils/database.py

@@ -49,7 +49,9 @@ class Database:
         self.gpt_4_bot = self.db_tiktok["gpt_4_bot"]
         self.meta_bot = self.db_tiktok["metabot"]
         self.youtube_bot = self.db_tiktok["youtubebot"]
-
+        self.session_bot = self.db_tiktok["sessionbot"]
+        self.captcha_bot = self.db_tiktok["captchabot"]
+        
         # akeno
         self.afk = self.db["afk"]
         self.antiflood = self.db["antiflood"]
@@ -240,6 +242,12 @@ class Database:
     async def get_all_youtube_bot(self) -> list:
         return [i async for i in self.youtube_bot.find({})]
 
+    async def get_all_session_bot(self) -> list:
+        return [i async for i in self.session_bot.find({})]
+
+    async def get_all_captcha_bot(self) -> list:
+        return [i async for i in self.captcha_bot.find({})]
+
     async def is_gbanned(self, user_id: int) -> bool:
         if await self.gban.find_one({"user_id": user_id}):
             return True