DB Pos changes

FileStream/Database/database.py

@@ -163,6 +163,16 @@ class Database:
 
   # ---------------------[ FIND FILE IN DB for Bot and APIs]---------------------#
   async def get_file(self, _id, privacy_type:str ):
+      try:
+        file_info = await self.files.find_one({"_id": ObjectId(_id)})
+        if not file_info:
+          file_info = await self.tfiles.find_one({"_id": ObjectId(_id)})
+          #raise FileNotFound
+        return file_info
+      except InvalidId:
+        raise FileNotFound
+
+  async def get_file_old(self, _id, privacy_type:str ):
     if privacy_type == "PUBLIC":
       try:
         file_info = await self.files.find_one({"_id": ObjectId(_id)})

FileStream/utils/FileProcessors/file_properties.py

@@ -83,7 +83,7 @@ async def get_file_ids(client: Client | bool, db_id: str, message) -> Optional[F
         log_msg = await send_file(FileStream, db_id, file_info, message, source)
         #updated_info = update_file_info(log_msg)
         await db.update_file_info(db_id, update_file_info(log_msg), file_info['privacy_type'])
-        await db.update_file_ids(db_id, await update_file_id(await db.get_file(db_id,file_info['privacy_type']),MULTI_CLIENTS), file_info['privacy_type'])
+        await db.update_file_ids(db_id, await update_file_id(await db.get_file(db_id),MULTI_CLIENTS), file_info['privacy_type'])
 
       logging.debug("Stored file_id of all clients in DB")
       if not client: