highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-02 11:09:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

Return empty string for encrypted PDF (#369)

Browse Source
This commit is contained in:
Yuhong Sun 2023-08-31 16:59:28 -07:00 committed by GitHub
parent d6e87df548
commit 80a08bbf0c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/danswer/connectors/google_drive/connector.py
View File

@ -308,9 +308,13 @@ def extract_text(file: dict[str, str], service: discovery.Resource) -> str:
response = service.files().get_media(fileId=file["id"]).execute() response = service.files().get_media(fileId=file["id"]).execute()
pdf_stream = io.BytesIO(response) pdf_stream = io.BytesIO(response)
pdf_reader = PdfReader(pdf_stream) pdf_reader = PdfReader(pdf_stream)
if pdf_reader.is_encrypted: if pdf_reader.is_encrypted:
logger.warning(f"Google drive file: {file['name']} is encrypted danswer will ignore it's content") logger.warning(
else: f"Google drive file: {file['name']} is encrypted - Danswer will ignore it's content"
)
return ""
return "\n".join(page.extract_text() for page in pdf_reader.pages) return "\n".join(page.extract_text() for page in pdf_reader.pages)