wehub-resource-sync
84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
"""社交媒体内容解析器
|
|
|
|
解析朋友圈、微博、小红书、Instagram 等社交媒体截图或导出文件。
|
|
截图通过 Claude 的 Read 工具直接读取（支持图片），本工具处理文本导出。
|
|
|
|
Usage:
|
|
python3 social_parser.py --dir <screenshot_dir> --output <output_path>
|
|
"""
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def scan_directory(dir_path: str) -> dict:
|
|
"""扫描目录，按类型分类文件"""
|
|
files = {'images': [], 'texts': [], 'other': []}
|
|
|
|
image_exts = {'.jpg', '.jpeg', '.png', '.gif', '.webp', '.bmp'}
|
|
text_exts = {'.txt', '.md', '.json', '.csv'}
|
|
|
|
for root, dirs, filenames in os.walk(dir_path):
|
|
for fname in filenames:
|
|
fpath = os.path.join(root, fname)
|
|
ext = Path(fname).suffix.lower()
|
|
if ext in image_exts:
|
|
files['images'].append(fpath)
|
|
elif ext in text_exts:
|
|
files['texts'].append(fpath)
|
|
else:
|
|
files['other'].append(fpath)
|
|
|
|
return files
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='社交媒体内容解析器')
|
|
parser.add_argument('--dir', required=True, help='截图/文件目录')
|
|
parser.add_argument('--output', required=True, help='输出文件路径')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not os.path.isdir(args.dir):
|
|
print(f"错误：目录不存在 {args.dir}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
files = scan_directory(args.dir)
|
|
|
|
os.makedirs(os.path.dirname(args.output) or '.', exist_ok=True)
|
|
with open(args.output, 'w', encoding='utf-8') as f:
|
|
f.write("# 社交媒体内容扫描结果\n\n")
|
|
f.write(f"扫描目录：{args.dir}\n\n")
|
|
|
|
f.write(f"## 文件统计\n")
|
|
f.write(f"- 图片文件：{len(files['images'])} 个\n")
|
|
f.write(f"- 文本文件：{len(files['texts'])} 个\n")
|
|
f.write(f"- 其他文件：{len(files['other'])} 个\n\n")
|
|
|
|
if files['images']:
|
|
f.write("## 图片列表（需用 Read 工具逐一查看）\n")
|
|
for img in sorted(files['images']):
|
|
f.write(f"- {img}\n")
|
|
f.write("\n")
|
|
|
|
if files['texts']:
|
|
f.write("## 文本内容\n")
|
|
for txt in sorted(files['texts']):
|
|
f.write(f"\n### {os.path.basename(txt)}\n")
|
|
try:
|
|
with open(txt, 'r', encoding='utf-8', errors='ignore') as tf:
|
|
content = tf.read(5000)
|
|
f.write(f"```\n{content}\n```\n")
|
|
except Exception as e:
|
|
f.write(f"读取失败：{e}\n")
|
|
|
|
print(f"扫描完成，结果已写入 {args.output}")
|
|
print(f"提示：图片截图需使用 Read 工具查看，本工具仅列出文件路径")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|