Merge pull request O365#465 from l-kotzur/convert_to_pdf_patch

Improve convert_to_pdf integration
igp-gravity · Jun 5, 2020 · c9b90a6 · c9b90a6
2 parents 79c4aec + 2e71909
commit c9b90a6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/O365/drive.py b/O365/drive.py
@@ -81,8 +81,11 @@ def download(self, to_path=None, name=None, chunk_size='auto',
                                  "or any integer number representing bytes")
 
             params = {}
-            if convert_to_pdf and Path(name).suffix in ALLOWED_PDF_EXTENSIONS:
-                params['format'] = 'pdf'
+            if convert_to_pdf:
+                if Path(self.name).suffix in ALLOWED_PDF_EXTENSIONS:
+                    params['format'] = 'pdf'
+                else:
+                    raise ValueError("File is not included in 'ALLOWED_PDF_EXTENSIONS'")
 
             with self.con.get(url, stream=stream, params=params) as response:
                 if not response: