104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
测试add_docs_to_mkdocs.py脚本的基本功能
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
# 测试get_all_md_files函数
|
||
|
|
def test_get_all_md_files():
|
||
|
|
print("测试get_all_md_files函数...")
|
||
|
|
md_files = []
|
||
|
|
for root, dirs, files in os.walk('docs'):
|
||
|
|
for file in files:
|
||
|
|
if file.endswith('.md'):
|
||
|
|
rel_path = os.path.relpath(os.path.join(root, file), 'docs')
|
||
|
|
rel_path = rel_path.replace('\\', '/')
|
||
|
|
md_files.append(rel_path)
|
||
|
|
|
||
|
|
print(f"找到 {len(md_files)} 个.md文件")
|
||
|
|
print("前10个文件:")
|
||
|
|
for i, f in enumerate(sorted(md_files)[:10], 1):
|
||
|
|
print(f" {i:2d}. {f}")
|
||
|
|
|
||
|
|
return sorted(md_files)
|
||
|
|
|
||
|
|
# 测试categorize_file函数
|
||
|
|
def test_categorize_file():
|
||
|
|
print("\n测试categorize_file函数...")
|
||
|
|
test_cases = [
|
||
|
|
('技术文档/API文档.md', '技术文档'),
|
||
|
|
('开发指南/快速开始.md', '开发指南'),
|
||
|
|
('学习笔记/Java学习.md', '学习笔记'),
|
||
|
|
('DevOps平台/Gerrit使用指南.md', 'DevOps平台'),
|
||
|
|
('cursor/cursor.md', 'Cursor工具'),
|
||
|
|
('Obsidian/高频命令.md', 'Obsidian笔记'),
|
||
|
|
('其他文件.md', '其他'),
|
||
|
|
]
|
||
|
|
|
||
|
|
for file_path, expected in test_cases:
|
||
|
|
# 模拟categorize_file函数
|
||
|
|
path_parts = file_path.split('/')
|
||
|
|
if len(path_parts) == 1:
|
||
|
|
category = "其他"
|
||
|
|
else:
|
||
|
|
category_map = {
|
||
|
|
'技术文档': '技术文档',
|
||
|
|
'开发指南': '开发指南',
|
||
|
|
'学习笔记': '学习笔记',
|
||
|
|
'DevOps平台': 'DevOps平台',
|
||
|
|
'cursor': 'Cursor工具',
|
||
|
|
'Obsidian': 'Obsidian笔记'
|
||
|
|
}
|
||
|
|
first_dir = path_parts[0]
|
||
|
|
category = category_map.get(first_dir, first_dir)
|
||
|
|
|
||
|
|
status = "✓" if category == expected else "✗"
|
||
|
|
print(f" {status} {file_path:40} -> {category:15} (期望: {expected})")
|
||
|
|
|
||
|
|
# 主测试
|
||
|
|
def main():
|
||
|
|
print("=" * 60)
|
||
|
|
print("测试add_docs_to_mkdocs.py脚本功能")
|
||
|
|
print("=" * 60)
|
||
|
|
|
||
|
|
# 测试1: 获取所有md文件
|
||
|
|
all_md_files = test_get_all_md_files()
|
||
|
|
|
||
|
|
# 测试2: 分类函数
|
||
|
|
test_categorize_file()
|
||
|
|
|
||
|
|
# 测试3: 检查当前mkdocs.yml中的文件
|
||
|
|
print("\n检查当前mkdocs.yml中的导航文件...")
|
||
|
|
try:
|
||
|
|
with open('mkdocs.yml', 'r', encoding='utf-8') as f:
|
||
|
|
content = f.read()
|
||
|
|
|
||
|
|
# 简单统计nav中的文件
|
||
|
|
import re
|
||
|
|
# 计算nav部分有多少行包含.md
|
||
|
|
nav_section = re.search(r'nav:\s*\n(.*?)(?=\n\w+:|$)', content, re.DOTALL)
|
||
|
|
if nav_section:
|
||
|
|
nav_content = nav_section.group(1)
|
||
|
|
md_count = nav_content.count('.md')
|
||
|
|
print(f"mkdocs.yml导航中大约有 {md_count} 个.md文件引用")
|
||
|
|
|
||
|
|
# 提取所有文件路径
|
||
|
|
file_paths = re.findall(r'\s+-\s+(.*?\.md)', nav_content)
|
||
|
|
print(f"实际提取到 {len(file_paths)} 个文件路径")
|
||
|
|
print("导航中的分类:")
|
||
|
|
lines = nav_content.split('\n')
|
||
|
|
for line in lines:
|
||
|
|
if line.strip() and not line.strip().startswith('-') and ':' in line:
|
||
|
|
print(f" - {line.strip()}")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"读取mkdocs.yml时出错: {e}")
|
||
|
|
|
||
|
|
print("\n" + "=" * 60)
|
||
|
|
print("测试完成")
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|