64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# This file is part of the Calibre-Web (https://github.com/janeczku/calibre-web)
|
|
# Copyright (C) 2024 OzzieIsaacs
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
from uuid import uuid4
|
|
import os
|
|
|
|
from .file_helper import get_temp_dir
|
|
from .subproc_wrapper import process_open
|
|
from . import logger, config
|
|
from .constants import SUPPORTED_CALIBRE_BINARIES
|
|
|
|
log = logger.create()
|
|
|
|
|
|
def do_calibre_export(book_id, book_format):
|
|
try:
|
|
quotes = [3, 5, 7, 9]
|
|
tmp_dir = get_temp_dir()
|
|
calibredb_binarypath = get_calibre_binarypath("calibredb")
|
|
temp_file_name = str(uuid4())
|
|
my_env = os.environ.copy()
|
|
if config.config_calibre_split:
|
|
my_env['CALIBRE_OVERRIDE_DATABASE_PATH'] = os.path.join(config.config_calibre_dir, "metadata.db")
|
|
library_path = config.config_calibre_split_dir
|
|
else:
|
|
library_path = config.config_calibre_dir
|
|
opf_command = [calibredb_binarypath, 'export', '--dont-write-opf', '--with-library', library_path,
|
|
'--to-dir', tmp_dir, '--formats', book_format, "--template", "{}".format(temp_file_name),
|
|
str(book_id)]
|
|
p = process_open(opf_command, quotes, my_env)
|
|
_, err = p.communicate()
|
|
if err:
|
|
log.error('Metadata embedder encountered an error: %s', err)
|
|
return tmp_dir, temp_file_name
|
|
except OSError as ex:
|
|
# ToDo real error handling
|
|
log.error_or_exception(ex)
|
|
return None, None
|
|
|
|
|
|
def get_calibre_binarypath(binary):
|
|
binariesdir = config.config_binariesdir
|
|
if binariesdir:
|
|
try:
|
|
return os.path.join(binariesdir, SUPPORTED_CALIBRE_BINARIES[binary])
|
|
except KeyError as ex:
|
|
log.error("Binary not supported by Calibre-Web: %s", SUPPORTED_CALIBRE_BINARIES[binary])
|
|
pass
|
|
return ""
|