BinaryONe commited on
Commit ·
93866aa
1
Parent(s): 98c1f4e
DB Changes
Browse files
FileStream/Database/database.py
CHANGED
|
@@ -151,6 +151,15 @@ class Database:
|
|
| 151 |
return file_info
|
| 152 |
except InvalidId:
|
| 153 |
raise FileNotFound
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 154 |
else:
|
| 155 |
return None
|
| 156 |
|
|
|
|
| 151 |
return file_info
|
| 152 |
except InvalidId:
|
| 153 |
raise FileNotFound
|
| 154 |
+
elif privacy_type == "PRIVATE":
|
| 155 |
+
try:
|
| 156 |
+
file_info = await self.tfiles.find_one({"_id": ObjectId(_id)})
|
| 157 |
+
if not file_info:
|
| 158 |
+
print('file not found')
|
| 159 |
+
#raise FileNotFound
|
| 160 |
+
return file_info
|
| 161 |
+
except InvalidId:
|
| 162 |
+
raise FileNotFound
|
| 163 |
else:
|
| 164 |
return None
|
| 165 |
|