Upload database.py

Akeno/utils/database.py

@@ -24,6 +24,7 @@ class Database:
         self.gachabots = self.db["gachabots"]
         self.cohere = self.db["cohere"]
         self.chatbot = self.db["chatbot"]
+        self.backup_chatbot = self.db["backupchatbot"]
         self.antiarabic = self.db["antiarabic"]
         self.gban = self.db["gban"]
         self.gmute = self.db["gmute"]
@@ -635,4 +636,16 @@ class Database:
         unset_data = {"user_id": None}
         return await self.chatbot.update_one({"chat_id": chat_id}, {"$unset": unset_data})
 
+    async def _update_chatbot_chat_in_db(self, user_id, chatbot_chat):
+        await self.backup_chatbot.update_one(
+            {"user_id": user_id},
+            {"$set": {"chatbot_chat": chatbot_chat}},
+            upsert=True
+        )
+
+    async def _get_chatbot_chat_from_db(self, user_id):
+        user_data = await self.backup_chatbot.find_one({"user_id": user_id})
+        return user_data.get("chatbot_chat", []) if user_data else []
+
+
 db = Database(MONGO_URL)