python:xml.etree 生成思维导图 Freemind文件

请参阅:java : pdfbox 读取 PDF文件内书签 或者 python:从PDF中提取目录

请注意:书的目录.txt 编码:UTF-8,推荐用 Notepad++ 转换编码。

xml 是 python 标准库,在 D:\Python39\Lib\xml\etree

python 用 xml.etree.ElementTree 生成思维导图 Freemind(.mm)文件

编辑 txt_xml_etree_mm.py 如下

python 复制代码
# -*- coding: utf-8 -*-
""" 读目录.txt文件,使用 xml.etree 生成思维导图 Freemind(.mm)文件"""
import os
import sys
import codecs
import xml.etree.ElementTree as et

if len(sys.argv) ==2:
    f1 = sys.argv[1]
else:
    print('usage: txt_xml_etree_mm.py  file1.txt')
    sys.exit(1)

if not os.path.exists(f1):
    print(f"ERROR: {f1} not found.")
    sys.exit(1)

fn,ext = os.path.splitext(f1)
if ext.lower() != '.txt':
    print('ext is not .txt')
    sys.exit(2)

# 创建根节点
map1 = et.Element("map")
map1.set('version', '1.0.1')

fp = codecs.open(f1, mode="r", encoding="utf-8")
# 读取第一行:书名
title = fp.readline()
# 创建主题节点
root = et.SubElement(map1, "node")
root.set('ID', '1')
root.set('STYLE', 'bubble') # 泡框
root.set('TEXT', title.strip())

# 定义连线的颜色:红色
edge = et.SubElement(root, "edge")
edge.set('COLOR', "#ff0000")

# 用缩排表现层级关系,假设最多5个层级
indent1 = ' '*2
indent2 = ' '*4
indent3 = ' '*6
indent4 = ' '*8

n = 2
for line in fp:
    txt = line.strip()
    if len(txt) ==0:
        continue
    txt = txt[0:-3] # 去掉行尾的页数
    if len(txt) >0 and line[0] !=' ':
        # 创建主题的子节点(1级节点)
        node1 = et.SubElement(root, "node")
        node1.set('ID', str(n))
        node1.set('POSITION', "right")
        node1.set('TEXT', txt)
        p_node = node1 # 寄存父节点
    elif line.startswith(indent1) and line[2] !=' ':
        # 创建node1的子节点(2级节点)
        try: type(node1)
        except NameError: node2 = et.SubElement(root, "node")
        else: node2 = et.SubElement(node1, "node")
        node2.set('ID', str(n))
        node2.set('TEXT', txt)
        p_node = node2
    elif line.startswith(indent2) and line[4] !=' ':
        # 创建node2的子节点(3级节点)
        try: type(node2)
        except NameError: node3 = et.SubElement(node1, "node")
        else: node3 = et.SubElement(node2, "node")
        node3.set('FLODED', "true")
        node3.set('ID', str(n))
        node3.set('TEXT', txt)
        p_node = node3
    elif line.startswith(indent3) and line[6] !=' ':
        # 创建node3的子节点(4级节点)
        try: type(node3)
        except NameError: node4 = et.SubElement(node2, "node")
        else: node4 = et.SubElement(node3, "node")
        node4.set('ID', str(n))
        node4.set('TEXT', txt)
        p_node = node4
    elif line.startswith(indent4) and line[8] !=' ':
        # 创建node4的子节点(5级节点)
        try: type(node4)
        except NameError: node5 = et.SubElement(p_node, "node")
        else: node5 = et.SubElement(node4, "node")
        node5.set('ID', str(n))
        node5.set('TEXT', txt)
    else:
        print(txt)
    n += 1
fp.close()

# 转换成 str,方便导出
map_bytes = et.tostring(map1, encoding="utf-8")

# 导出到 .mm 格式的文件中
f2 = fn +'.mm'
with open(f2, 'w+b') as fp:
    fp.write(map_bytes)

print(f"line number: {n}")

运行 python txt_xml_etree_mm.py your_pdf_dir.txt

生成 your_pdf_dir.mm

相关推荐
唐古乌梁海40 分钟前
Flask项目中CSRF Token实现的解决方案
python·flask·csrf
月疯41 分钟前
FLASK与JAVA的文件互传(单文件互传亲测)
后端·python·flask
程序猿小D1 小时前
【完整源码+数据集+部署教程】医疗设备显示器图像分割系统: yolov8-seg-C2f-SCConv
python·yolo·计算机视觉·数据集·yolov8·医疗设备显示器图像分割系统
Derrick__11 小时前
Python常用内建模块——XML
xml·python
梵得儿SHI1 小时前
Java 操作 XML 及动态生成报告:从解析到实战
xml·java·jaxb·dom4j·xml解析·操作xml·报告生成
weixin_434169601 小时前
【机器学习】 在Jupyter Notebook 中如何指定Python环境
python·机器学习·jupyter
我是华为OD~HR~栗栗呀3 小时前
Java面经(22届考研-华oD)
java·后端·python·华为od·华为
毕设源码-朱学姐3 小时前
【开题答辩全过程】以 python基于Hadoop的服装穿搭系统的设计与实现为例,包含答辩的问题和答案
开发语言·hadoop·python
十二月将至3 小时前
python读取文件的常用操作
网络·python
爱砸键盘的懒洋洋3 小时前
Python第四课:数据类型与转换
开发语言·python