Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages 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 @@ class Container(ContainerBase): # {{{
if not os.path.exists(base): if not os.path.exists(base):
os.makedirs(base) os.makedirs(base)
with lopen(path, 'wb') as f: 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) mt = media_type or self.guess_type(name)
self.name_path_map[name] = path self.name_path_map[name] = path
self.mime_map[name] = mt self.mime_map[name] = mt