|
@@ -67,35 +67,23 @@ def main():
|
|
|
|
|
|
|
|
for idx, item in enumerate(result):
|
|
for idx, item in enumerate(result):
|
|
|
print(f"处理结果项 {idx}: {type(item)}", file=sys.stderr)
|
|
print(f"处理结果项 {idx}: {type(item)}", file=sys.stderr)
|
|
|
- print(f" - 有 save 方法: {hasattr(item, 'save')}", file=sys.stderr)
|
|
|
|
|
|
|
+ print(f" - 有 save_to_markdown 方法: {hasattr(item, 'save_to_markdown')}", file=sys.stderr)
|
|
|
|
|
|
|
|
- if hasattr(item, 'save'):
|
|
|
|
|
- item_save_path = output_dir / image_basename
|
|
|
|
|
- print(f" - 保存到: {item_save_path}", file=sys.stderr)
|
|
|
|
|
- item.save(str(item_save_path))
|
|
|
|
|
|
|
+ if hasattr(item, 'save_to_markdown'):
|
|
|
|
|
+ print(f" - 保存到: {output_dir}", file=sys.stderr)
|
|
|
|
|
+ # 使用官方 API 保存 markdown
|
|
|
|
|
+ item.save_to_markdown(save_path=str(output_dir))
|
|
|
|
|
|
|
|
# 列出保存后的文件
|
|
# 列出保存后的文件
|
|
|
- if item_save_path.exists():
|
|
|
|
|
- files = list(item_save_path.iterdir())
|
|
|
|
|
|
|
+ if output_dir.exists():
|
|
|
|
|
+ files = list(output_dir.iterdir())
|
|
|
print(f" - 保存后的文件列表: {[f.name for f in files]}", file=sys.stderr)
|
|
print(f" - 保存后的文件列表: {[f.name for f in files]}", file=sys.stderr)
|
|
|
|
|
|
|
|
- # 查找生成的 markdown 文件(多种可能的路径)
|
|
|
|
|
- possible_paths = [
|
|
|
|
|
- item_save_path / f"{image_basename}.md",
|
|
|
|
|
- output_dir / f"{image_basename}.md",
|
|
|
|
|
- item_save_path / "result.md",
|
|
|
|
|
- ]
|
|
|
|
|
-
|
|
|
|
|
- for markdown_file in possible_paths:
|
|
|
|
|
- print(f" - 检查路径: {markdown_file} -> {markdown_file.exists()}", file=sys.stderr)
|
|
|
|
|
- if markdown_file.exists():
|
|
|
|
|
- print(f"SUCCESS: {markdown_file}")
|
|
|
|
|
- sys.exit(0)
|
|
|
|
|
-
|
|
|
|
|
- # 如果没找到,尝试递归查找所有 .md 文件
|
|
|
|
|
|
|
+ # 递归查找所有 .md 文件
|
|
|
print(f" - 递归查找 .md 文件...", file=sys.stderr)
|
|
print(f" - 递归查找 .md 文件...", file=sys.stderr)
|
|
|
- md_files = list(item_save_path.rglob("*.md")) if item_save_path.exists() else []
|
|
|
|
|
- print(f" - 找到 {len(md_files)} 个 .md 文件", file=sys.stderr)
|
|
|
|
|
|
|
+ md_files = list(output_dir.rglob("*.md"))
|
|
|
|
|
+ print(f" - 找到 {len(md_files)} 个 .md 文件: {[f.name for f in md_files]}", file=sys.stderr)
|
|
|
|
|
+
|
|
|
if md_files:
|
|
if md_files:
|
|
|
print(f"SUCCESS: {md_files[0]}")
|
|
print(f"SUCCESS: {md_files[0]}")
|
|
|
sys.exit(0)
|
|
sys.exit(0)
|