Upload database.py

akn/utils/database.py

@@ -47,6 +47,7 @@ class Database:
         self.magic_bot = self.db_tiktok["magicbot"]
         self.gemini_bot = self.db_tiktok["geminibot"]
         self.gpt_4_bot = self.db_tiktok["gpt_4_bot"]
+        self.meta_bot = self.db_tiktok["metabot"]
 
         # akeno
         self.afk = self.db["afk"]
@@ -216,9 +217,19 @@ class Database:
         data = await self.session.find_one({"user_id": user_id})
         return data
 
+    async def add_bot_token_meta(self, user_id, bot_token):
+        add_bot_clone = {
+            "bot_token": bot_token,
+            "user_id": user_id
+        }
+        return await self.meta_bot.update_one({"user_id": user_id}, {"$set": add_bot_clone}, upsert=True)
+    
     async def get_all_sessions(self) -> list:
         return [i async for i in self.session.find({})]
 
+    async def get_all_meta_bot(self) -> list:
+        return [i async for i in self.meta_bot.find({})]
+        
     async def get_all_magic_bot(self) -> list:
         return [i async for i in self.magic_bot.find({})]