68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
Warn = """\033[0;31;40m
|
|
=======警告=======
|
|
此脚本仅包含有限的“自动重命名”功能，它：
|
|
- 不会对Gerber文件进行解析
|
|
- 不会验证文件格式是否正确
|
|
- 不会处理钻孔文件
|
|
- 极有可能损坏你的文件或引起生产错误
|
|
- 需要你根据实际情况进行修改
|
|
如果你不知道自己在干什么，请手动修改文件而不是使用脚本
|
|
请编辑此脚本以解除警告
|
|
\033[0m"""
|
|
print(Warn)
|
|
raise Exception("READ THE WARNING")
|
|
|
|
import yaml, os, re
|
|
|
|
fconfig = open("config.yaml", "r", encoding="utf-8")
|
|
frule = open("rule.yaml", "r", encoding="utf-8")
|
|
Config = yaml.load(fconfig, Loader=yaml.FullLoader)
|
|
Rule = yaml.load(frule, Loader=yaml.FullLoader)
|
|
|
|
WorkDir = Config["WorkDir"]
|
|
DestDir = Config["DestDir"]
|
|
WorkDirFiles = os.listdir(WorkDir)
|
|
|
|
if not DestDir:
|
|
DestDir=os.path.join(WorkDir,'output')
|
|
if not os.path.exists(DestDir):
|
|
os.mkdir(DestDir)
|
|
|
|
#创建PCB下单必读文档
|
|
with open(os.path.join(DestDir, Config["TextFileName"]), "w") as textFile:
|
|
textFile.write(Config["TextFileContent"])
|
|
|
|
#检验文件是否齐全/重复匹配
|
|
for key, value in Rule.items():
|
|
matchFile = []
|
|
rePattern = re.compile(pattern=value)
|
|
|
|
for fileName in WorkDirFiles:
|
|
if rePattern.search(fileName):
|
|
matchFile.append(fileName)
|
|
|
|
if len(matchFile) < 1:
|
|
raise Exception(key + "匹配失败")
|
|
elif len(matchFile) > 1:
|
|
raise Exception(key + "重复匹配")
|
|
else:
|
|
print(key + " -> " + matchFile[0])
|
|
|
|
#改名和加头操作
|
|
for key, value in Rule.items():
|
|
matchFile = ""
|
|
rePattern = re.compile(pattern=value)
|
|
|
|
for fileName in WorkDirFiles:
|
|
if rePattern.search(fileName):
|
|
matchFile = fileName
|
|
|
|
with open(os.path.join(WorkDir, matchFile), "r") as file:
|
|
fileData = file.read()
|
|
with open(os.path.join(DestDir, Config["FileName"][key]), "w") as file:
|
|
file.write(Config["Header"])
|
|
file.write(fileData)
|
|
|
|
fconfig.close()
|
|
frule.close()
|
|
exit() |