diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-10-24 14:49:34 +0100 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-10-24 14:49:34 +0100 |
| commit | febe13fc7006db65f3a90bbb8e30d646fd0b72af (patch) | |
| tree | 15f72f0c7afed050a6c542d6e86d7922da64f525 | |
| parent | Move pdf operations to a separate module (diff) | |
| download | paper2remarkable-febe13fc7006db65f3a90bbb8e30d646fd0b72af.tar.gz paper2remarkable-febe13fc7006db65f3a90bbb8e30d646fd0b72af.zip | |
Move pdf file check to utils
| -rw-r--r-- | paper2remarkable/providers/_base.py | 11 | ||||
| -rw-r--r-- | paper2remarkable/utils.py | 12 |
2 files changed, 12 insertions, 11 deletions
diff --git a/paper2remarkable/providers/_base.py b/paper2remarkable/providers/_base.py index d427f9e..8e9223e 100644 --- a/paper2remarkable/providers/_base.py +++ b/paper2remarkable/providers/_base.py @@ -181,17 +181,6 @@ class Provider(metaclass=abc.ABCMeta): self.log("Created filename: %s" % name) return name - - def check_file_is_pdf(self, filename): - try: - fp = open(filename, "rb") - pdf = PyPDF2.PdfFileReader(fp, strict=False) - fp.close() - del pdf - return True - except PyPDF2.utils.PdfReadError: - exception("Downloaded file isn't a valid pdf file.") - def download_url(self, url, filename): """Download the content of an url and save it to a filename """ self.log("Downloading file at url: %s" % url) diff --git a/paper2remarkable/utils.py b/paper2remarkable/utils.py index af19d22..5188afb 100644 --- a/paper2remarkable/utils.py +++ b/paper2remarkable/utils.py @@ -9,6 +9,7 @@ Copyright: 2019, G.J.J. van den Burg """ +import PyPDF2 import sys GITHUB_URL = "https://github.com/GjjvdBurg/arxiv2remarkable" @@ -23,3 +24,14 @@ def exception(msg): % GITHUB_URL ) raise SystemExit(1) + + +def check_file_is_pdf(filename): + try: + fp = open(filename, "rb") + pdf = PyPDF2.PdfFileReader(fp, strict=False) + fp.close() + del pdf + return True + except PyPDF2.utils.PdfReadError: + exception("Downloaded file isn't a valid pdf file.") |
