lxbfYeaa/BookConverter
Archived
Template
1
0
Fork 0
forked from LiveCarta/BookConverter
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add abil. to replace value of attributes in presets

Browse Source
This commit is contained in:
Kiryl
2022-09-05 17:19:31 +03:00
parent 7cef389ba4
commit 3fcff462d3
3 changed files with 23 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
consumer.py
View File

@@ -14,7 +14,8 @@ from src.docx_converter.docx_solver import DocxBook
from src.epub_converter.epub_solver import EpubBook
def configure_file_logger(name: str, filename: str = "logs/converter.log", filemode: str ="w+", logging_level: int = logging.INFO) -> logging.Logger:
def configure_file_logger(name: str, filename: str = "logs/converter.log",
filemode: str = "w+", logging_level: int = logging.INFO) -> logging.Logger:
logger = logging.getLogger(name)
folder_path = os.path.dirname(os.path.abspath(__file__))

10
presets/presets.json
View File

@@ -81,11 +81,17 @@
"preset_name": "attr_replacer",
"rules": [
{
"attr": "xlink:href",
"attr": {
"name": "xlink:href",
"value": ".*"
},
"condition": {
"tags": ["^img$"]
},
"attr_to_replace": "src"
"attr_to_replace": {
"name": "src",
"value": null
}
}
]
},

19
src/epub_converter/html_epub_processor.py
View File

@@ -147,10 +147,17 @@ class HtmlEpubProcessor:
@staticmethod
def _replace_attr(**kwargs):
attr = kwargs["rule"]["attr"]
attr_to_replace = kwargs["rule"]["attr_to_replace"]
kwargs["tag"][attr_to_replace] = kwargs["tag"][attr]
del kwargs["tag"][attr]
attr, attr_value =\
kwargs["rule"]["attr"]["name"], kwargs["rule"]["attr"]["value"]
attr_to_replace, attr_value_to_replace =\
kwargs["rule"]["attr_to_replace"]["name"], kwargs["rule"]["attr_to_replace"]["value"]
if attr_to_replace:
kwargs["tag"][attr_to_replace] = kwargs["tag"][attr]
if attr_value_to_replace:
kwargs["tag"].attrs[attr_to_replace] = attr_value_to_replace
del kwargs["tag"][attr]
elif attr_value_to_replace:
kwargs["tag"].attrs[attr] = attr_value_to_replace
@staticmethod
def _unwrap_tag(**kwargs):
@@ -208,7 +215,7 @@ class HtmlEpubProcessor:
elif condition_on_tag[0] == "tags":
attr = rule["attr"]
for tag in chapter_tag.find_all([re.compile(tag) for tag in tags],
{attr: re.compile(r".*")}):
{attr['name']: re.compile(fr"{attr['value']}")}):
action(chapter_tag=chapter_tag, tag=tag, rule=rule)
else:
for tag in chapter_tag.find_all([re.compile(tag) for tag in tags]):
@@ -316,7 +323,7 @@ class HtmlEpubProcessor:
----------
title_str: str
chapter_tag: Tag, soup object
chapter_tag: BeautifulSoup, soup object
remove_title_from_chapter: bool