Upload database.py

akn/utils/database.py

@@ -100,6 +100,17 @@ class Database:
             upsert=True
         )
 
+    async def set_pic_in_allbot(self, id, link, is_pic):
+        return await self.privary_bot.update_one(
+            {"bot_id": id},
+            {"$set": {"link_pic": link, "is_pic": is_pic}},
+            upsert=True
+        )
+
+    async def get_pic_in_allbot(self, id):
+        user_data = await self.privary_bot.find_one({"bot_id": id})
+        return user_data.get("link_pic"), user_data.get("is_pic") if user_data else False
+
     async def fixed_maintance(self, is_maintance):
         return await self.maintance_bot.update_one(
             {"bot_id": 1},