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().
master
Ethan Dalool 2019-09-06 15:38:51 -07:00
parent 9821125f2c
commit 60d9d646a2
1 changed files with 4 additions and 2 deletions

View File

@ -276,13 +276,15 @@ class NotInSpine(EpubfileException):
class Epub: class Epub:
def __init__(self, directory): def __init__(self, directory, _original_epub_filepath=None):
if isinstance(directory, tempfile.TemporaryDirectory): if isinstance(directory, tempfile.TemporaryDirectory):
self._tempdir_reference = directory self._tempdir_reference = directory
directory = directory.name directory = directory.name
self.root_directory = pathclass.Path(directory, force_sep='/') 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_filepath = None
self.opf = None self.opf = None
@ -325,7 +327,7 @@ class Epub:
def open(cls, epub_filepath): def open(cls, epub_filepath):
extract_to = tempfile.TemporaryDirectory(prefix='epubfile-') extract_to = tempfile.TemporaryDirectory(prefix='epubfile-')
extract_epub(epub_filepath, extract_to.name) 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): def save(self, epub_filepath):
self.write_opf() self.write_opf()