Fix #568 (encoding problem in fb2 upload)
This commit is contained in:
parent
82fe282e9b
commit
f81fbaf542
20
cps/fb2.py
20
cps/fb2.py
|
@ -20,41 +20,43 @@ def get_fb2_info(tmp_file_path, original_file_extension):
|
|||
def get_author(element):
|
||||
last_name = element.xpath('fb:last-name/text()', namespaces=ns)
|
||||
if len(last_name):
|
||||
last_name = last_name[0]
|
||||
last_name = last_name[0].encode('utf-8')
|
||||
else:
|
||||
last_name = u''
|
||||
middle_name = element.xpath('fb:middle-name/text()', namespaces=ns)
|
||||
if len(middle_name):
|
||||
middle_name = middle_name[0]
|
||||
middle_name = middle_name[0].encode('utf-8')
|
||||
else:
|
||||
middle_name = u''
|
||||
first_name = element.xpath('fb:first-name/text()', namespaces=ns)
|
||||
if len(first_name):
|
||||
first_name = first_name[0]
|
||||
first_name = first_name[0].encode('utf-8')
|
||||
else:
|
||||
first_name = u''
|
||||
return first_name + ' ' + middle_name + ' ' + last_name
|
||||
return (first_name.decode('utf-8') + u' '
|
||||
+ middle_name.decode('utf-8') + u' '
|
||||
+ last_name.decode('utf-8')).encode('utf-8')
|
||||
|
||||
author = str(", ".join(map(get_author, authors)))
|
||||
|
||||
title = tree.xpath('/fb:FictionBook/fb:description/fb:title-info/fb:book-title/text()', namespaces=ns)
|
||||
if len(title):
|
||||
title = str(title[0])
|
||||
title = str(title[0].encode('utf-8'))
|
||||
else:
|
||||
title = u''
|
||||
description = tree.xpath('/fb:FictionBook/fb:description/fb:publish-info/fb:book-name/text()', namespaces=ns)
|
||||
if len(description):
|
||||
description = str(description[0])
|
||||
description = str(description[0].encode('utf-8'))
|
||||
else:
|
||||
description = u''
|
||||
|
||||
return uploader.BookMeta(
|
||||
file_path=tmp_file_path,
|
||||
extension=original_file_extension,
|
||||
title=title.encode('utf-8').decode('utf-8'),
|
||||
author=author.encode('utf-8').decode('utf-8'),
|
||||
title=title.decode('utf-8'),
|
||||
author=author.decode('utf-8'),
|
||||
cover=None,
|
||||
description=description,
|
||||
description=description.decode('utf-8'),
|
||||
tags="",
|
||||
series="",
|
||||
series_id="",
|
||||
|
|
|
@ -379,7 +379,7 @@ def save_cover(url, book_path):
|
|||
img = requests.get(url)
|
||||
if img.headers.get('content-type') != 'image/jpeg':
|
||||
web.app.logger.error("Cover is no jpg file, can't save")
|
||||
return false
|
||||
return False
|
||||
|
||||
if ub.config.config_use_google_drive:
|
||||
tmpDir = gettempdir()
|
||||
|
@ -388,13 +388,13 @@ def save_cover(url, book_path):
|
|||
f.close()
|
||||
uploadFileToEbooksFolder(os.path.join(book_path, 'cover.jpg'), os.path.join(tmpDir, f.name))
|
||||
web.app.logger.info("Cover is saved on gdrive")
|
||||
return true
|
||||
return True
|
||||
|
||||
f = open(os.path.join(ub.config.config_calibre_dir, book_path, "cover.jpg"), "wb")
|
||||
f.write(img.content)
|
||||
f.close()
|
||||
web.app.logger.info("Cover is saved")
|
||||
return true
|
||||
return True
|
||||
|
||||
def do_download_file(book, book_format, data, headers):
|
||||
if ub.config.config_use_google_drive:
|
||||
|
|
|
@ -3096,7 +3096,7 @@ def edit_book(book_id):
|
|||
|
||||
if not error:
|
||||
if to_save["cover_url"]:
|
||||
if helper.save_cover(to_save["cover_url"], book.path) is true:
|
||||
if helper.save_cover(to_save["cover_url"], book.path) is True:
|
||||
book.has_cover = 1
|
||||
else:
|
||||
flash(_(u"Cover is not a jpg file, can't save"), category="error")
|
||||
|
|
Loading…
Reference in New Issue
Block a user