window环境的构建脚本

Author: 黄佰荣

qpython-docs/add-analytics-windown.py

@@ -1,13 +1,10 @@
-import sys, os
-
-with open(sys.argv[1], 'r', encoding='gbk') as f, \
-     open('./qpydoc.tmp', 'w', encoding='gbk') as g, \
-     open(os.path.dirname(os.path.abspath(__file__)) + '/extra.txt', 'r', encoding='gbk') as e:
+import sys,os
+with open(sys.argv[1], 'r') as f, open('./qpydoc.tmp', 'wb') as g, open(os.path.dirname(os.path.abspath(__file__))+'/extra.txt','r') as e:
     pth = sys.argv[1][1:]
-    extra = "".join(e.readlines()).replace("{{PTH}}", pth)
+    extra = "".join(e.readlines()).replace("{{PTH}}",pth)
     g.write('\n'.join(
         filter(lambda s: len(s),
                map(lambda s:
-                   ('', extra + "<hr/>")[s == '<div role="contentinfo">'] + s,
+                       ('',extra+"<hr/>")[s=='<div role="contentinfo">']+s,
                    map(str.strip, f.readlines())))))
-os.rename('./qpydoc.tmp', sys.argv[1])
+os.rename('./qpydoc.tmp', sys.argv[1])

qpython-docs/add-analytics.py

@@ -1,10 +1,41 @@
-import sys,os
-with open(sys.argv[1], 'r') as f, open('./qpydoc.tmp', 'wb') as g, open(os.path.dirname(os.path.abspath(__file__))+'/extra.txt','r') as e:
-    pth = sys.argv[1][1:]
-    extra = "".join(e.readlines()).replace("{{PTH}}",pth)
-    g.write('\n'.join(
-        filter(lambda s: len(s),
-               map(lambda s:
-                       ('',extra+"<hr/>")[s=='<div role="contentinfo">']+s,
-                   map(str.strip, f.readlines())))))
-os.rename('./qpydoc.tmp', sys.argv[1])
+import sys
+import os
+
+def process_file(input_file, output_file, extra_file):
+    # 确保 extra.txt 文件存在
+    if not os.path.exists(extra_file):
+        print(f"Error: The file '{extra_file}' does not exist.")
+        return
+
+    # 获取 extra.txt 文件内容
+    with open(extra_file, 'r', encoding='utf-8') as e:
+        extra = e.read().replace("{{PTH}}", input_file[1:])  # 用路径替换 {{PTH}}
+
+    # 逐行读取输入文件并处理内容
+    with open(input_file, 'r', encoding='utf-8') as f, open(output_file, 'w', encoding='utf-8') as g:
+        for line in f:
+            line = line.strip()  # 去除行首和行尾空白字符
+            if line:  # 忽略空行
+                # 如果是特定标签，添加 extra 和 <hr/> 内容
+                if line == '<div role="contentinfo">':
+                    g.write(extra + "<hr/>\n")
+                else:
+                    g.write(line + '\n')  # 写入处理后的行
+
+    # 替换原始文件
+    os.rename(output_file, input_file)
+
+def main():
+    if len(sys.argv) < 2:
+        print("Usage: python script.py <input_file>")
+        return
+
+    input_file = sys.argv[1]
+    output_file = './qpydoc.tmp'
+    extra_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'extra.txt')
+
+    # 处理文件
+    process_file(input_file, output_file, extra_file)
+
+if __name__ == '__main__':
+    main()