Initial commit

novels/round3/final_expand_4.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Final pass - fix hz function and expand all
+import os, re
+
+d = "/app/working/workspaces/default/novels/round3/01-越躺平越强"
+
+# CORRECT regex - use raw string
+r = re.compile(r'[\u4e00-\u9fff]')
+
+def hz(s):
+    return len(r.findall(s))
+
+def read(fname):
+    path = os.path.join(d, fname)
+    with open(path, 'r', encoding='utf-8') as f:
+        return f.read()
+
+def write(fname, content):
+    path = os.path.join(d, fname)
+    with open(path, 'w', encoding='utf-8') as f:
+        f.write(content)
+
+def expand(fname, text):
+    c = read(fname)
+    cn = hz(c)
+    write(fname, c + text)
+    cn_new = hz(read(fname))
+    print(f"  {fname}: {cn} -> {cn_new}汉字")
+    return cn_new
+
+# First, check all chapters
+print("当前状态:")
+for f in sorted(os.listdir(d)):
+    if f.endswith('.md') and f.startswith('第'):
+        cn = hz(read(f))
+        print(f"  {f}: {cn}汉字")
+
+print("\n开始修复和扩写...")