116 lines
3.3 KiB
Python
116 lines
3.3 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import os
|
||
import re
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
from pypinyin import lazy_pinyin
|
||
|
||
INPUT_DIR = "/Users/gavin/kman/export"
|
||
OUTPUT_DIR = os.path.join(INPUT_DIR, "out")
|
||
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
||
|
||
def slugify(title):
|
||
# 拼音 + 只保留字母 + 最多15个字符
|
||
py = "".join(lazy_pinyin(title))
|
||
s = re.sub("[^a-zA-Z]", "", py)
|
||
return s[:15]
|
||
|
||
def parse_filename(fname):
|
||
# 去掉 note_bleaf 前缀
|
||
name = fname[len("note_bleaf"):]
|
||
# 分 title / subtitle
|
||
if "：" in name:
|
||
title, rest = name.split("：", 1)
|
||
# 去掉尾部可能的 -xxxx.md
|
||
subtitle = re.sub(r'-.*\.md$', '', rest)
|
||
else:
|
||
title = re.sub(r'-.*\.md$', '', name).replace(".md","")
|
||
subtitle = ""
|
||
return title.strip(), subtitle.strip()
|
||
|
||
def parse_file(filepath):
|
||
"""逐行解析，保证 CONTENT 中有 | 也能处理"""
|
||
entries = []
|
||
with open(filepath, "r", encoding="utf-8") as f:
|
||
lines = f.readlines()
|
||
for line in lines:
|
||
line = line.strip()
|
||
if not line or line.startswith("--") or line.startswith("TYPE"):
|
||
continue
|
||
parts = line.split("|", 4) # 只分成 5 个字段
|
||
if len(parts) < 5:
|
||
print(f"⚠️ 跳过解析失败行: {line}")
|
||
continue
|
||
type_, bookname, author, marktime, content = parts
|
||
try:
|
||
marktime_dt = datetime.strptime(marktime.strip(), "%Y/%m/%d %H:%M:%S")
|
||
except:
|
||
try:
|
||
marktime_dt = datetime.strptime(marktime.strip(), "%Y/%m/%d %H:%M")
|
||
except:
|
||
marktime_dt = None
|
||
entries.append({
|
||
"TYPE": type_.strip(),
|
||
"BOOKNAME": bookname.strip(),
|
||
"AUTHOR": author.strip(),
|
||
"MARKTIME": marktime_dt,
|
||
"CONTENT": content.strip()
|
||
})
|
||
return entries
|
||
|
||
def convert_file(filepath):
|
||
fname = os.path.basename(filepath)
|
||
title, subtitle = parse_filename(fname)
|
||
entries = parse_file(filepath)
|
||
if not entries:
|
||
print(f"❌ 解析失败 {filepath}")
|
||
return
|
||
|
||
# 最早的时间
|
||
times = [e["MARKTIME"] for e in entries if e["MARKTIME"]]
|
||
ddd = min(times).strftime("%Y-%m-%d %H:%M:%S") if times else datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||
|
||
slug = slugify(title)
|
||
output_fname = f"note_{title}.md"
|
||
output_path = os.path.join(OUTPUT_DIR, output_fname)
|
||
|
||
lines = [
|
||
"---",
|
||
f"title: '{title}'",
|
||
f"subtitle: '{subtitle}'",
|
||
f"author: '大童'",
|
||
f"date: {ddd}",
|
||
f"slug: {slug}",
|
||
"description:",
|
||
"---",
|
||
""
|
||
]
|
||
|
||
# 作者显示第一条的 AUTHOR
|
||
author_line = f"作者：{entries[0]['AUTHOR']}"
|
||
lines.append(author_line)
|
||
lines.append("")
|
||
|
||
for e in entries:
|
||
if e["TYPE"] == "HL":
|
||
lines.append(e["CONTENT"])
|
||
lines.append("")
|
||
elif e["TYPE"] == "NT":
|
||
lines.append(f"> {e['CONTENT']}")
|
||
content_text = "\n".join(lines)
|
||
|
||
with open(output_path, "w", encoding="utf-8") as f:
|
||
f.write(content_text)
|
||
|
||
print(f"✅ 生成 {output_path}")
|
||
|
||
def main():
|
||
for fname in os.listdir(INPUT_DIR):
|
||
if fname.startswith("note_bleaf") and fname.endswith(".md"):
|
||
convert_file(os.path.join(INPUT_DIR, fname))
|
||
|
||
if __name__ == "__main__":
|
||
main()
|