askdocs / components /filechecker.py
Junaidb's picture
Update components/filechecker.py
9d9b572 verified
raw
history blame contribute delete
592 Bytes
from components.mongoconnection import provideClient
mc=provideClient()
db=mc["askmydoc_db"]
coll=db["askmydoc_col"]
def checkFile(owner,filetype):
document = coll.find_one({"owner": owner})
if document is None or "files" not in document or len(document["files"]) == 0:
return {"status": False, "files": []}
else:
cleaned_files=[]
for file in document["files"]:
if file["filetype"]==filetype:
cleaned_files.append({"file_name":file["filename"]})
return {"status": True, "files": cleaned_files}