Changes in Database

FileStream/Database/database.py

@@ -286,6 +286,11 @@ class Database:
                                   {"$set": {
                                       "file_ids": file_ids
                                   }})
+    elif privacy_type=="PRIVATE":
+      await self.tfiles.update_one({"_id": ObjectId(_id)},
+                                  {"$set": {
+                                      "file_ids": file_ids
+                                  }})                              
     else:
       return None
 
@@ -307,7 +312,13 @@ class Database:
           }
       })
     else:
-      return None
+      await self.tfiles.update_one({"_id": ObjectId(_id)}, {
+          "$set": {
+              "message_id": file_info['message_id'],
+              "location": file_info['location'],
+              "file": file_info['file']
+          }
+      })