Backup_Repos/calibre
1
0
Fork 0
mirror of git://github.com/kovidgoyal/calibre.git synced 2026-05-02 21:03:17 +02:00
Code Issues Projects Releases Wiki Activity

Allow passing in file objects to add_file()

Browse source
This commit is contained in:
Kovid Goyal 2016-10-09 14:57:09 +05:30
parent 1255f81cfd
commit 7bf79bd650
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/calibre/ebooks/oeb/polish/container.py
View file

@ -335,7 +335,10 @@ def add_file(self, name, data, media_type=None, spine_index=None, modify_name_if
if not os.path.exists(base):
os.makedirs(base)
with lopen(path, 'wb') as f:
f.write(data)
if hasattr(data, 'read'):
shutil.copyfileobj(data, f)
else:
f.write(data)
mt = media_type or self.guess_type(name)
self.name_path_map[name] = path
self.mime_map[name] = mt