mirror of
https://github.com/illian64/llm-translate.git
synced 2026-04-28 11:49:54 +00:00
0a70da3b98
* book translate * files processing * files processing * files processing * files processing --------- Co-authored-by: APodoinikov <APodoynikov@detmir.ru>
88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
import os
|
|
|
|
from app import file_processor, params
|
|
from app.app_core import AppCore
|
|
from app.dto import ProcessingFileDirReq, ProcessingFileResp, FileProcessingPluginInitInfo, ProcessingFileStruct
|
|
|
|
plugin_name = os.path.basename(__file__)[:-3] # calculating modname
|
|
|
|
|
|
def start(core: AppCore):
|
|
manifest = { # plugin settings
|
|
"name": "Translator for txt files", # name
|
|
"version": "1.0", # version
|
|
|
|
"default_options": {
|
|
"enabled": True,
|
|
"markdown_output": False,
|
|
"text_format": {
|
|
"original_prefix": "",
|
|
"original_postfix": "",
|
|
"translate_prefix": "*",
|
|
"translate_postfix": "*",
|
|
},
|
|
"new_line_delimiter": "\n",
|
|
"output_file_name_template": {
|
|
"preserve_original": "%source%__%from_lang%_%to_lang%",
|
|
"without_original": "%source%__%to_lang%",
|
|
},
|
|
"default_extension_processor": {
|
|
"txt": True
|
|
},
|
|
},
|
|
|
|
"file_processing": {
|
|
"file_txt_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={"txt"})
|
|
|
|
|
|
def file_processing(core: AppCore, file_struct: ProcessingFileStruct, req: ProcessingFileDirReq) -> ProcessingFileResp:
|
|
options = core.plugin_options(plugin_name)
|
|
markdown_output: bool = options["markdown_output"]
|
|
new_line_delimiter: str = options["new_line_delimiter"]
|
|
text_format = params.read_plugin_file_processing_text_format(options)
|
|
new_line_delimiter_count = 2 if markdown_output else 1
|
|
|
|
result_lines: list[str] = []
|
|
file_content = file_processor.read_file_with_fix_encoding(file_struct.path_file_in())
|
|
lines: list[str] = file_content.splitlines()
|
|
for line in lines:
|
|
if line == '':
|
|
result_lines.append(new_line_delimiter)
|
|
continue
|
|
|
|
if req.preserve_original_text:
|
|
result_lines.append(text_format.original_text(line) +
|
|
new_line_delimiter * new_line_delimiter_count)
|
|
|
|
translate_req = req.translate_req(line)
|
|
translate_txt = core.translate(translate_req).result
|
|
translate_txt_format = text_format.translate_text(translate_txt)
|
|
result_lines.append(translate_txt_format + new_line_delimiter * new_line_delimiter_count)
|
|
|
|
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=options["encoding_output"]) as f:
|
|
f.write((''.join(result_lines)))
|
|
|
|
return file_processor.get_processing_file_resp_ok(file_struct=file_struct, file_out=out_file_name)
|
|
|
|
|
|
def processed_file_name(core: AppCore, file_struct: ProcessingFileStruct, req: ProcessingFileDirReq) -> str:
|
|
options = core.plugin_options(plugin_name)
|
|
|
|
file_name = file_processor.file_name_from_template(file_struct=file_struct, req=req, options=options)
|
|
if options["markdown_output"]:
|
|
file_name = file_name[:-3] + "md"
|
|
|
|
return file_name
|