1

npm install -g markmap-cli

1
2

pip install playwright
playwright install chromium

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

import os
import subprocess
import asyncio
from playwright.async_api import async_playwright

async def capture_map(html_file, output_png):
    """
    渲染并截取高分辨率思维导图
    """
    async with async_playwright() as p:
        
        browser = await p.chromium.launch(headless=True)
        
        context = await browser.new_context(device_scale_factor=2)
        page = await context.new_page()
        
        abs_path = os.path.abspath(html_file)
        await page.goto(f"file://{abs_path}")
        
        try:
            
            svg_selector = "svg.markmap"
            await page.wait_for_selector(svg_selector, timeout=10000)
            
            
            await asyncio.sleep(2)
            
            
            element = page.locator(svg_selector)
            await element.screenshot(path=output_png, omit_background=False)
            
        except Exception as e:
            print(f"渲染过程出错: {e}")
        finally:
            await browser.close()

def run_conversion(input_md):
    if not os.path.exists(input_md):
        print(f"找不到输入文件: {input_md}")
        return

    base_name = os.path.splitext(input_md)[0]
    temp_html = f"{base_name}_temp.html"
    output_png = f"{base_name}.png"

    try:
        
        subprocess.run(["markmap", input_md, "-o", temp_html, "--offline"], check=True)
        
        
        asyncio.run(capture_map(temp_html, output_png))
        
        
        if os.path.exists(temp_html):
            os.remove(temp_html)
            
        print(f"完成: {output_png}")
    except Exception as e:
        print(f"执行失败: {e}")

if __name__ == "__main__":
    import sys
    if len(sys.argv) < 2:
        print("用法: python md2img.py test.md")
    else:
        run_conversion(sys.argv[1])