From 60d9d646a27e219a836bb23de93cc5b12109378a Mon Sep 17 00:00:00 2001 From: Ethan Dalool Date: Fri, 6 Sep 2019 15:38:51 -0700 Subject: [PATCH] Add argument _original_epub_filepath for better messaging later. Printing the tempdir is not very helpful, so if we need to print something about this epub file we can use _original_path to get the epub if it was a .open() or the tempdir if it was .new(). --- epubfile.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/epubfile.py b/epubfile.py index 0775b65..2beebbe 100644 --- a/epubfile.py +++ b/epubfile.py @@ -276,13 +276,15 @@ class NotInSpine(EpubfileException): class Epub: - def __init__(self, directory): + def __init__(self, directory, _original_epub_filepath=None): if isinstance(directory, tempfile.TemporaryDirectory): self._tempdir_reference = directory directory = directory.name self.root_directory = pathclass.Path(directory, force_sep='/') + self._original_path = _original_epub_filepath or self.root_directory.absolute_path + self.opf_filepath = None self.opf = None @@ -325,7 +327,7 @@ class Epub: def open(cls, epub_filepath): extract_to = tempfile.TemporaryDirectory(prefix='epubfile-') extract_epub(epub_filepath, extract_to.name) - return cls(extract_to) + return cls(extract_to, _original_epub_filepath=epub_filepath) def save(self, epub_filepath): self.write_opf()