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().
This commit is contained in:
parent
9821125f2c
commit
60d9d646a2
1 changed files with 4 additions and 2 deletions
|
@ -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()
|
||||
|
|
Loading…
Reference in a new issue