BinaryONe commited on
Commit
93866aa
·
1 Parent(s): 98c1f4e

DB Changes

Browse files
Files changed (1) hide show
  1. FileStream/Database/database.py +9 -0
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