用Python写的一个脚本，机床拷贝的文件ALL-PROG.TXT分割成独立的NC文件

jiahuajing

1. # -*- coding: utf-8 -*-
   
2. """
   
3. NC/G代码文件分割工具
   
4. - 以 OXXXX 开头，以 M30 结束
   
5. - 每个分割文件以该程序第一行的字符作为文件名
   
6. """
   
7. 
   
8. import os
   
9. import re
   
10. 
    
11. 
    
12. def sanitize_filename(name: str) -> str:
    
13.     """清理文件名中的非法字符（Windows 不允许 \ / : * ? " < > |）"""
    
14.     # 替换非法字符为下划线
    
15.     name = re.sub(r'[\\/:*?"<>|]', '_', name)
    
16.     # 去除首尾空白和点
    
17.     name = name.strip(' .')
    
18.     # 限制长度，避免文件名过长
    
19.     return name[:100] if name else 'unnamed'
    
20. 
    
21. 
    
22. def split_nc_file(input_path: str, output_dir: str = None) -> list:
    
23.     # 默认输出目录：输入文件同目录下的 split_output
    
24.     if output_dir is None:
    
25.         output_dir = os.path.join(os.path.dirname(os.path.abspath(input_path)), 'split_output')
    
26.     os.makedirs(output_dir, exist_ok=True)
    
27. 
    
28.     # 多编码兼容读取
    
29.     content = None
    
30.     for enc in ('utf-8', 'gbk', 'gb2312', 'latin-1'):
    
31.         try:
    
32.             with open(input_path, 'r', encoding=enc) as f:
    
33.                 content = f.read()
    
34.             break
    
35.         except UnicodeDecodeError:
    
36.             continue
    
37.     if content is None:
    
38.         raise ValueError("无法识别文件编码")
    
39. 
    
40.     lines = content.splitlines()
    
41.     generated = []
    
42. 
    
43.     current_lines = []
    
44.     first_line = None
    
45.     in_program = False
    
46. 
    
47.     for line in lines:
    
48.         stripped = line.strip()
    
49. 
    
50.         # 检测程序开始：行首 O + 数字
    
51.         if re.match(r'^O\d+', stripped, re.IGNORECASE):
    
52.             in_program = True
    
53.             current_lines = [line]
    
54.             first_line = stripped   # 记录第一行（去掉首尾空白）
    
55.             continue
    
56. 
    
57.         if in_program:
    
58.             current_lines.append(line)
    
59. 
    
60.             # 检测程序结束：行中包含 M30
    
61.             if re.search(r'M30', stripped, re.IGNORECASE):
    
62.                 # 用第一行字符作为文件名
    
63.                 filename = sanitize_filename(first_line) + '.nc'
    
64.                 out_path = os.path.join(output_dir, filename)
    
65. 
    
66.                 # 处理重名：自动加序号
    
67.                 counter = 1
    
68.                 while os.path.exists(out_path):
    
69.                     name, ext = os.path.splitext(filename)
    
70.                     out_path = os.path.join(output_dir, f"{name}_{counter}{ext}")
    
71.                     counter += 1
    
72. 
    
73.                 # 写入文件
    
74.                 with open(out_path, 'w', encoding='utf-8') as f:
    
75.                     f.write('\n'.join(current_lines) + '\n')
    
76. 
    
77.                 generated.append(out_path)
    
78.                 print(f"已生成: {out_path}")
    
79. 
    
80.                 # 重置状态
    
81.                 in_program = False
    
82.                 current_lines = []
    
83.                 first_line = None
    
84. 
    
85.     return generated
    
86. 
    
87. 
    
88. if __name__ == '__main__':
    
89.     path = input("请输入文件路径: ").strip().strip('"')
    
90.     if not os.path.isfile(path):
    
91.         print(f"文件不存在: {path}")
    
92.     else:
    
93.         results = split_nc_file(path)
    
94.         print(f"\n分割完成，共生成 {len(results)} 个文件。")

复制代码