mirror of
https://github.com/illian64/llm-translate.git
synced 2026-04-28 11:49:54 +00:00
323458ca27
* Prepare for docker, documentation. * Prepare for docker, documentation. * Prepare for docker, documentation. --------- Co-authored-by: APodoinikov <APodoynikov@detmir.ru>
82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
import logging
|
|
import os
|
|
import traceback
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
from app import file_processor
|
|
from app.app_core import AppCore
|
|
from app.dto import ProcessingFileDirReq, ProcessingFileResp, FileProcessingPluginInitInfo, ProcessingFileStruct
|
|
from app.file_processor_html import FileProcessorHtml
|
|
|
|
plugin_name = os.path.basename(__file__)[:-3] # calculating modname
|
|
logger = logging.getLogger('uvicorn')
|
|
|
|
|
|
def start(core: AppCore):
|
|
manifest = { # plugin settings
|
|
"name": "Translator for epub books", # name
|
|
"version": "1.0", # version
|
|
|
|
"default_options": {
|
|
"enabled": True,
|
|
"text_format": {
|
|
"original_tag": "",
|
|
"translate_tag": "emphasis",
|
|
"header_delimiter": " / ",
|
|
},
|
|
"header_tags": [],
|
|
"text_tags": ['p'],
|
|
"output_file_name_template": {
|
|
"preserve_original": "%%source%%__%%from_lang%%_%%to_lang%%",
|
|
"without_original": "%%source%%__%%to_lang%%",
|
|
},
|
|
"translate_only_first_paragraphs": 0,
|
|
"default_extension_processor": {
|
|
"fb2": True
|
|
},
|
|
},
|
|
|
|
"file_processing": {
|
|
"file_epub_translate": (init, file_processing, processed_file_name)
|
|
}
|
|
}
|
|
|
|
return manifest
|
|
|
|
|
|
def start_with_options(core: AppCore, manifest: dict):
|
|
pass
|
|
|
|
|
|
def init(core: AppCore) -> FileProcessingPluginInitInfo:
|
|
return FileProcessingPluginInitInfo(plugin_name=plugin_name, supported_extensions={"fb2"})
|
|
|
|
|
|
def file_processing(core: AppCore, file_struct: ProcessingFileStruct, req: ProcessingFileDirReq) -> ProcessingFileResp:
|
|
options = core.plugin_options(plugin_name)
|
|
html_processor = FileProcessorHtml(core=core, options=options)
|
|
|
|
try:
|
|
fb2_content = file_processor.read_file_with_fix_encoding(file_struct.path_file_in())
|
|
|
|
soup = BeautifulSoup(fb2_content, features="xml")
|
|
html_processor.process(req, soup, "body")
|
|
|
|
out_file_name = processed_file_name(core=core, file_struct=file_struct, req=req)
|
|
|
|
with open(file_struct.path_file_out(out_file_name), 'w+', encoding='utf-8') as fb2_put_file:
|
|
fb2_put_file.write(soup.decode())
|
|
|
|
return file_processor.get_processing_file_resp_ok(file_struct=file_struct, file_out=out_file_name)
|
|
except Exception as e:
|
|
traceback.print_tb(e.__traceback__, limit=10)
|
|
logging.error("Error with processing file %s: %s", file_struct.file_name_ext, str(e))
|
|
return file_processor.get_processing_file_resp_error(
|
|
file_in=file_struct.file_name_ext, path_in=file_struct.path_in, error_msg=str(e))
|
|
|
|
|
|
def processed_file_name(core: AppCore, file_struct: ProcessingFileStruct, req: ProcessingFileDirReq) -> str:
|
|
options = core.plugin_options(plugin_name)
|
|
|
|
return file_processor.file_name_from_template(file_struct=file_struct, req=req, options=options)
|