全流程运行测试完毕

source/README_up_analysis.md

@@ -115,6 +115,8 @@ python source/batch_ai_summary_from_report.py --input source\output\reports\up_t
 
 ```powershell
 python source/extract_keep_follow_doc.py
+
+python source/extract_keep_follow_doc.py --input source/19_06_all.md --output source/19_30_keep_follow.md      
 ```
 
 输出：

source/extract_keep_follow_doc.py

@@ -46,8 +46,8 @@ def main() -> int:
 
         action_m = re.search(r"-\s*建议动作:\s*(.+)", section)
         action = action_m.group(1).strip() if action_m else ""
-        # 仅保留“建议动作: 保留关注”的UP。
-        if action != "保留关注":
+        # 反逻辑：没有"建议动作: 可以取关"就保留
+        if action == "可以取关":
             continue
 
         ai_m = re.search(r"###\s*AI分析\s*\n([\s\S]*?)(?=\n###\s|\Z)", section)

source/quicksort.py

@@ -1,7 +0,0 @@
-def quicksort(arr):
-    if len(arr) <= 1:
-        return arr
-    pivot = arr[-1]
-    left = [x for x in arr[:-1] if x <= pivot]
-    right = [x for x in arr[:-1] if x > pivot]
-    return quicksort(left) + [pivot] + quicksort(right)

source/sort_up_main.py

@@ -1,27 +1,43 @@
+import argparse
 import re
 
-input_file = r'D:\Code\Python\followlist\source\output\reports\bbb.md'
-output_file = r'D:\Code\Python\followlist\source\output\reports\bbb.md'
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description="对UP主按首字母排序")
+    parser.add_argument(
+        "--input",
+        default="source/output/reports/up_analysis_full_auto.md",
+        help="输入报告路径",
+    )
+    parser.add_argument(
+        "--output",
+        help="输出报告路径（默认覆盖输入）",
+    )
+    return parser.parse_args()
 
-with open(input_file, 'r', encoding='utf-8') as f:
+def main():
+    args = parse_args()
+    input_file = args.input
+    output_file = args.output or input_file
+
+    with open(input_file, 'r', encoding='utf-8') as f:
         content = f.read()
 
-lines = content.split('\n')
+    lines = content.split('\n')
 
-header_lines = []
-section_starts = []
-for i, line in enumerate(lines):
+    header_lines = []
+    section_starts = []
+    for i, line in enumerate(lines):
         if line.startswith('## '):
             section_starts.append(i)
 
-if len(section_starts) < 2:
+    if len(section_starts) < 2:
         print('No sections found')
-    exit()
+        return 1
 
-header = '\n'.join(lines[:section_starts[0]])
-sections_data = []
+    header = '\n'.join(lines[:section_starts[0]])
+    sections_data = []
 
-for idx in range(len(section_starts)):
+    for idx in range(len(section_starts)):
         start = section_starts[idx]
         if idx + 1 < len(section_starts):
             end = section_starts[idx + 1]
@@ -32,10 +48,10 @@ for idx in range(len(section_starts)):
         section_text = '\n'.join(section_lines)
         sections_data.append(section_text)
 
-sections_data = sections_data[1:]
+    sections_data = sections_data[1:]
 
-parsed = []
-for sec in sections_data:
+    parsed = []
+    for sec in sections_data:
         match = re.match(r'^## (\d+)\. (.+) \(mid: (\d+)\)', sec)
         if match:
             num = int(match.group(1))
@@ -48,7 +64,7 @@ for sec in sections_data:
                 'content': sec
             })
 
-def sort_key(item):
+    def sort_key(item):
         name = item['name']
         first_char = name[0].lower() if name else ''
         if first_char.isdigit():
@@ -58,16 +74,20 @@ def sort_key(item):
         else:
             return '2' + first_char
 
-parsed.sort(key=sort_key)
+    parsed.sort(key=sort_key)
 
-new_content = header + '\n'
-for i, sec in enumerate(parsed):
+    new_content = header + '\n'
+    for i, sec in enumerate(parsed):
         new_content += sec['content'] + '\n'
 
-with open(output_file, 'w', encoding='utf-8') as f:
+    with open(output_file, 'w', encoding='utf-8') as f:
         f.write(new_content)
 
-print(f'Sorted {len(parsed)} sections')
-print('First 10:')
-for s in parsed[:10]:
+    print(f'Sorted {len(parsed)} sections')
+    print('First 10:')
+    for s in parsed[:10]:
         print(f'  {s["name"]}')
+    return 0
+
+if __name__ == "__main__":
+    raise SystemExit(main())