diff --git a/pdfx/__init__.py b/pdfx/__init__.py index 6042e26..af7e52a 100644 --- a/pdfx/__init__.py +++ b/pdfx/__init__.py @@ -205,3 +205,13 @@ def download_pdfs(self, target_dir): # Download urls as a set to avoid duplicates download_urls(urls, dir_referenced_pdfs) + + def close(self): + if self.stream: + self.stream.close() + + def __enter__(self): + return self + + def __exit__(self, exc_type, exc_val, exc_tb): + self.close()