153 lines
6 KiB
Python
153 lines
6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import ub
|
|
import db
|
|
import re
|
|
import web
|
|
from flask_babel import gettext as _
|
|
|
|
|
|
RET_FAIL = 0
|
|
RET_SUCCESS = 1
|
|
|
|
|
|
def versionKindle():
|
|
versions = _(u'not installed')
|
|
if os.path.exists(ub.config.config_converterpath):
|
|
try:
|
|
p = subprocess.Popen(ub.config.config_converterpath, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
p.wait()
|
|
for lines in p.stdout.readlines():
|
|
if isinstance(lines, bytes):
|
|
lines = lines.decode('utf-8')
|
|
if re.search('Amazon kindlegen\(', lines):
|
|
versions = lines
|
|
except Exception:
|
|
versions = _(u'Excecution permissions missing')
|
|
return {'kindlegen' : versions}
|
|
|
|
|
|
def versionCalibre():
|
|
versions = _(u'not installed')
|
|
if os.path.exists(ub.config.config_converterpath):
|
|
try:
|
|
p = subprocess.Popen([ub.config.config_converterpath, '--version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
p.wait()
|
|
for lines in p.stdout.readlines():
|
|
if isinstance(lines, bytes):
|
|
lines = lines.decode('utf-8')
|
|
if re.search('.*\(calibre', lines):
|
|
versions = lines
|
|
except Exception:
|
|
versions = _(u'Excecution permissions missing')
|
|
return {'Calibre converter' : versions}
|
|
|
|
|
|
def convert_kindlegen(file_path, book):
|
|
error_message = None
|
|
if not os.path.exists(ub.config.config_converterpath):
|
|
error_message = _(u"kindlegen binary %(kindlepath)s not found", kindlepath=ub.config.config_converterpath)
|
|
web.app.logger.error("convert_kindlegen: " + error_message)
|
|
return error_message, RET_FAIL
|
|
try:
|
|
p = subprocess.Popen((ub.config.config_converterpath + " \"" + file_path + u".epub\"").encode(sys.getfilesystemencoding()),
|
|
stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
|
|
except Exception as e:
|
|
error_message = _(u"kindlegen failed, no execution permissions")
|
|
web.app.logger.error("convert_kindlegen: " + error_message)
|
|
return error_message, RET_FAIL
|
|
# Poll process for new output until finished
|
|
while True:
|
|
nextline = p.stdout.readline()
|
|
if nextline == '' and p.poll() is not None:
|
|
break
|
|
if nextline != "\r\n":
|
|
# Format of error message (kindlegen translates its output texts):
|
|
# Error(prcgen):E23006: Language not recognized in metadata.The dc:Language field is mandatory.Aborting.
|
|
conv_error = re.search(".*\(.*\):(E\d+):\s(.*)", nextline)
|
|
# If error occoures, log in every case
|
|
if conv_error:
|
|
error_message = _(u"Kindlegen failed with Error %(error)s. Message: %(message)s",
|
|
error=conv_error.group(1), message=conv_error.group(2).decode('utf-8'))
|
|
web.app.logger.info("convert_kindlegen: " + error_message)
|
|
web.app.logger.info(nextline.strip('\r\n'))
|
|
else:
|
|
web.app.logger.debug(nextline.strip('\r\n'))
|
|
|
|
check = p.returncode
|
|
if not check or check < 2:
|
|
book.data.append(db.Data(
|
|
name=book.data[0].name,
|
|
book_format="MOBI",
|
|
book=book.id,
|
|
uncompressed_size=os.path.getsize(file_path + ".mobi")
|
|
))
|
|
db.session.commit()
|
|
if ub.config.config_use_google_drive:
|
|
os.remove(file_path + u".epub")
|
|
return file_path + ".mobi", RET_SUCCESS
|
|
else:
|
|
web.app.logger.info("convert_kindlegen: kindlegen failed with error while converting book")
|
|
if not error_message:
|
|
error_message = 'kindlegen failed, no excecution permissions'
|
|
return error_message, RET_FAIL
|
|
|
|
|
|
def convert_calibre(file_path, book):
|
|
error_message = None
|
|
if not os.path.exists(ub.config.config_converterpath):
|
|
error_message = _(u"Ebook-convert binary %(converterpath)s not found", converterpath=ub.config.config_converterpath)
|
|
web.app.logger.error("convert_calibre: " + error_message)
|
|
return error_message, RET_FAIL
|
|
try:
|
|
command = ("\""+ub.config.config_converterpath + "\" \"" + file_path + u".epub\" \""
|
|
+ file_path + u".mobi\" " + ub.config.config_calibre).encode(sys.getfilesystemencoding())
|
|
p = subprocess.Popen(command,stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
|
|
except Exception as e:
|
|
error_message = _(u"Ebook-convert failed, no execution permissions")
|
|
web.app.logger.error("convert_calibre: " + error_message)
|
|
return error_message, RET_FAIL
|
|
# Poll process for new output until finished
|
|
while True:
|
|
nextline = p.stdout.readline()
|
|
if nextline == '' and p.poll() is not None:
|
|
break
|
|
web.app.logger.debug(nextline.strip('\r\n').decode(sys.getfilesystemencoding()))
|
|
|
|
check = p.returncode
|
|
if check == 0 :
|
|
book.data.append(db.Data(
|
|
name=book.data[0].name,
|
|
book_format="MOBI",
|
|
book=book.id,
|
|
uncompressed_size=os.path.getsize(file_path + ".mobi")
|
|
))
|
|
db.session.commit()
|
|
if ub.config.config_use_google_drive:
|
|
os.remove(file_path + u".epub")
|
|
|
|
return file_path + ".mobi", RET_SUCCESS
|
|
else:
|
|
web.app.logger.info("convert_calibre: Ebook-convert failed with error while converting book")
|
|
if not error_message:
|
|
error_message = 'Ebook-convert failed, no excecution permissions'
|
|
return error_message, RET_FAIL
|
|
|
|
|
|
def versioncheck():
|
|
if ub.config.config_ebookconverter == 1:
|
|
return versionKindle()
|
|
elif ub.config.config_ebookconverter == 2:
|
|
return versionCalibre()
|
|
else:
|
|
return {'ebook_converter':''}
|
|
|
|
|
|
def convert_mobi(file_path, book):
|
|
if ub.config.config_ebookconverter == 2:
|
|
return convert_calibre(file_path, book)
|
|
else:
|
|
return convert_kindlegen(file_path, book)
|