работа с шаблонами
This commit is contained in:
@@ -1,16 +1,22 @@
|
||||
from typing import Optional
|
||||
|
||||
import typer
|
||||
from pathlib import Path
|
||||
from .core.converter import process_document
|
||||
|
||||
app = typer.Typer()
|
||||
|
||||
import logging
|
||||
|
||||
@app.command()
|
||||
def convert(
|
||||
input_dir: Path = typer.Option(..., "--input", "-i", help="Папка с исходниками"),
|
||||
output_dir: Path = typer.Option(..., "--output", "-o", help="Папка для результата"),
|
||||
style_config: Path = typer.Option("resources/style_config.yaml", "--style", "-s", help="Файл стилей")
|
||||
style_config: Path = typer.Option("resources/style_config.yaml", "--style", "-s", help="Файл стилей"),
|
||||
template: Optional[Path] = typer.Option(None, "--template", "-t", help="Шаблон .dotx"),
|
||||
verbose: bool = typer.Option(False, "--verbose", "-v", help="Включить отладочные сообщения"),
|
||||
):
|
||||
process_document(input_dir, output_dir, style_config)
|
||||
log_level = logging.DEBUG if verbose else logging.INFO
|
||||
logging.basicConfig(level=log_level, format='%(levelname)s: %(message)s')
|
||||
|
||||
if __name__ == "__main__":
|
||||
app()
|
||||
process_document(input_dir, output_dir, style_config, template)
|
||||
@@ -1,30 +1,43 @@
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from docx import Document
|
||||
from .style_registry import StyleRegistry
|
||||
from .renderer import render_markdown_to_docx
|
||||
|
||||
def process_document(input_dir: Path, output_dir: Path, style_config_path: Path):
|
||||
style_reg = StyleRegistry(style_config_path)
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
def process_document(input_dir: Path, output_dir: Path, style_config_path: Path, template_path: Optional[Path] = None):
|
||||
for doc_dir in input_dir.iterdir():
|
||||
if not doc_dir.is_dir():
|
||||
continue
|
||||
# Пропускаем папку images
|
||||
if doc_dir.name == "images":
|
||||
continue
|
||||
|
||||
doc_name = doc_dir.name
|
||||
md_path = doc_dir / "document.md"
|
||||
images_dir = doc_dir / "images"
|
||||
|
||||
if not md_path.exists():
|
||||
print(f"⚠️ Пропущено: {doc_name} — нет document.md")
|
||||
logger.warning(f"Пропущено: {doc_name} — нет document.md")
|
||||
continue
|
||||
|
||||
with open(md_path, encoding="utf-8") as f:
|
||||
md_content = f.read()
|
||||
|
||||
doc = Document()
|
||||
style_reg.ensure_styles_in_doc(doc) # добавляем стили
|
||||
if template_path and template_path.exists():
|
||||
doc = Document(template_path)
|
||||
else:
|
||||
doc = Document()
|
||||
|
||||
style_reg = StyleRegistry(style_config_path)
|
||||
style_reg.ensure_styles_in_doc(doc)
|
||||
render_markdown_to_docx(doc, md_content, images_dir, style_reg)
|
||||
|
||||
output_path = output_dir / f"{doc_name}.docx"
|
||||
output_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
doc.save(output_path)
|
||||
print(f"✅ {doc_name} → {output_path}")
|
||||
logger.info(f"Создан документ: {output_path}")
|
||||
@@ -7,11 +7,15 @@ def render_markdown_to_docx(doc, md_content: str, images_dir, style_reg: StyleRe
|
||||
ast = markdown(md_content)
|
||||
|
||||
def walk(node):
|
||||
handler = get_handler(node["type"])
|
||||
if handler:
|
||||
handler(node, doc, images_dir, style_reg)
|
||||
elif node.get("children"):
|
||||
for child in node["children"]:
|
||||
walk(child)
|
||||
if isinstance(node, list):
|
||||
for item in node:
|
||||
walk(item)
|
||||
elif isinstance(node, dict):
|
||||
handler = get_handler(node["type"])
|
||||
if handler:
|
||||
handler(node, doc, images_dir, style_reg)
|
||||
elif node.get("children"):
|
||||
for child in node["children"]:
|
||||
walk(child)
|
||||
|
||||
walk(ast)
|
||||
@@ -10,7 +10,10 @@ from docx.enum.text import WD_ALIGN_PARAGRAPH
|
||||
|
||||
@register_handler("heading")
|
||||
def handle_heading(node: dict, doc, images_dir: Path, style_reg):
|
||||
level = min(node["level"], 3)
|
||||
# mistune v3: node["attrs"]["level"]
|
||||
# тесты: node["level"]
|
||||
level = node.get("attrs", {}).get("level") or node.get("level")
|
||||
level = min(level, 3)
|
||||
text = node["children"][0]["raw"] if node["children"] else ""
|
||||
doc.add_heading(text, level=level)
|
||||
p = doc.paragraphs[-1]
|
||||
|
||||
Reference in New Issue
Block a user