|
|
@@ -0,0 +1,185 @@
|
|
|
+#!/usr/bin/env python3
|
|
|
+# -*- coding: utf-8 -*-
|
|
|
+"""
|
|
|
+从 Oracle PL/SQL 存储过程源码中提取 v_o_errmsg 赋值语句，
|
|
|
+生成 I18n/Templates.xml 可用的正则模板条目。
|
|
|
+
|
|
|
+用法：
|
|
|
+ python extract_db_templates.py <input.sql> [<input2.sql> ...] -o output.xml
|
|
|
+
|
|
|
+生成的 XML 形如：
|
|
|
+ <Template zh="序列号:(.*?) 已被工单(.*?)绑定"
|
|
|
+ en="SN:{0} has already been bound to WO {1}"
|
|
|
+ vi="SN:{0} đã được gán vào WO {1}"
|
|
|
+ id="SN:{0} sudah terikat ke WO {1}"/>
|
|
|
+"""
|
|
|
+import argparse
|
|
|
+import re
|
|
|
+import os
|
|
|
+import sys
|
|
|
+from xml.sax.saxutils import escape
|
|
|
+
|
|
|
+
|
|
|
+def tokenize_assignment(expr: str):
|
|
|
+ """
|
|
|
+ 把形如 'xxx'||var||'yyy'||var||'zzz' 的表达式拆成：
|
|
|
+ [('txt', 'xxx'), ('var', 'var'), ('txt', 'yyy'), ...]
|
|
|
+ """
|
|
|
+ tokens = []
|
|
|
+ i = 0
|
|
|
+ expr = expr.strip()
|
|
|
+ while i < len(expr):
|
|
|
+ # 跳过空白和 ||
|
|
|
+ if expr[i] in " \t\r\n":
|
|
|
+ i += 1
|
|
|
+ continue
|
|
|
+ if expr.startswith("||", i):
|
|
|
+ i += 2
|
|
|
+ continue
|
|
|
+
|
|
|
+ # 字符串字面量
|
|
|
+ if expr[i] == "'":
|
|
|
+ j = i + 1
|
|
|
+ literal = []
|
|
|
+ while j < len(expr):
|
|
|
+ if expr[j] == "'" and j + 1 < len(expr) and expr[j + 1] == "'":
|
|
|
+ literal.append("'")
|
|
|
+ j += 2
|
|
|
+ elif expr[j] == "'":
|
|
|
+ j += 1
|
|
|
+ break
|
|
|
+ else:
|
|
|
+ literal.append(expr[j])
|
|
|
+ j += 1
|
|
|
+ tokens.append(("txt", "".join(literal)))
|
|
|
+ i = j
|
|
|
+ continue
|
|
|
+
|
|
|
+ # 变量 / 函数调用 / 表达式
|
|
|
+ # 取到下一个 || 或字符串字面量或分号为止
|
|
|
+ j = i
|
|
|
+ while j < len(expr):
|
|
|
+ if expr[j] in " \t\r\n":
|
|
|
+ pass
|
|
|
+ elif expr.startswith("||", j):
|
|
|
+ break
|
|
|
+ elif expr[j] == "'":
|
|
|
+ break
|
|
|
+ elif expr[j] == ";":
|
|
|
+ break
|
|
|
+ j += 1
|
|
|
+ var_expr = expr[i:j].strip()
|
|
|
+ if var_expr:
|
|
|
+ tokens.append(("var", var_expr))
|
|
|
+ i = j
|
|
|
+
|
|
|
+ return tokens
|
|
|
+
|
|
|
+
|
|
|
+def build_template(tokens):
|
|
|
+ """
|
|
|
+ 根据 token 序列生成中文正则模式和英文/越文/印尼文占位符模板。
|
|
|
+ """
|
|
|
+ fixed_parts = []
|
|
|
+ var_count = 0
|
|
|
+ has_variable = False
|
|
|
+
|
|
|
+ for kind, value in tokens:
|
|
|
+ if kind == "txt":
|
|
|
+ fixed_parts.append(re.escape(value))
|
|
|
+ else:
|
|
|
+ has_variable = True
|
|
|
+ fixed_parts.append(r"(.*?)")
|
|
|
+ var_count += 1
|
|
|
+
|
|
|
+ if not has_variable:
|
|
|
+ return None, None # 纯静态提示，不需要模板
|
|
|
+
|
|
|
+ zh_pattern = "".join(fixed_parts)
|
|
|
+ placeholders = "{" + "} {".join(str(i) for i in range(var_count)) + "}"
|
|
|
+
|
|
|
+ # 默认英文模板：保留固定中文占位，后续需人工翻译
|
|
|
+ fixed_texts = [v for k, v in tokens if k == "txt"]
|
|
|
+ en_parts = []
|
|
|
+ var_idx = 0
|
|
|
+ for kind, value in tokens:
|
|
|
+ if kind == "txt":
|
|
|
+ # 这里先用原中文占位，实际使用时应替换为英文
|
|
|
+ en_parts.append(value)
|
|
|
+ else:
|
|
|
+ en_parts.append("{" + str(var_idx) + "}")
|
|
|
+ var_idx += 1
|
|
|
+ en_template = "".join(en_parts)
|
|
|
+
|
|
|
+ return zh_pattern, en_template
|
|
|
+
|
|
|
+
|
|
|
+def extract_assignments(sql_text: str):
|
|
|
+ """提取所有 v_o_errmsg := ... ; 语句。"""
|
|
|
+ # 去掉 PL/SQL 行注释
|
|
|
+ sql_text = re.sub(r"--.*?\n", "\n", sql_text)
|
|
|
+ # 去掉 /* */ 块注释
|
|
|
+ sql_text = re.sub(r"/\*.*?\*/", "", sql_text, flags=re.DOTALL)
|
|
|
+
|
|
|
+ # 匹配 v_o_errmsg:=...;（允许跨行）
|
|
|
+ pattern = re.compile(
|
|
|
+ r"v_o_errmsg\s*:=\s*(.*?);",
|
|
|
+ re.IGNORECASE | re.DOTALL,
|
|
|
+ )
|
|
|
+ return [m.group(1).strip() for m in pattern.finditer(sql_text)]
|
|
|
+
|
|
|
+
|
|
|
+def escape_attr(value: str) -> str:
|
|
|
+ """XML 属性转义。"""
|
|
|
+ return escape(value, {'"': """})
|
|
|
+
|
|
|
+
|
|
|
+def main():
|
|
|
+ parser = argparse.ArgumentParser(description="从 PL/SQL 提取动态错误模板")
|
|
|
+ parser.add_argument("sql_files", nargs="+", help="输入的 .sql 文件路径")
|
|
|
+ parser.add_argument("-o", "--output", default="Templates.xml", help="输出 XML 文件")
|
|
|
+ args = parser.parse_args()
|
|
|
+
|
|
|
+ all_templates = []
|
|
|
+ seen = set()
|
|
|
+
|
|
|
+ for path in args.sql_files:
|
|
|
+ if not os.path.exists(path):
|
|
|
+ print(f"[WARN] File not found: {path}", file=sys.stderr)
|
|
|
+ continue
|
|
|
+ with open(path, "r", encoding="utf-8") as f:
|
|
|
+ text = f.read()
|
|
|
+
|
|
|
+ for expr in extract_assignments(text):
|
|
|
+ # 跳过纯函数调用，如 SQLERRM、dbms_utility.format_error_backtrace
|
|
|
+ if re.match(r"^[A-Za-z0-9_\(\)\.\s\|\|]+$", expr) and "'" not in expr:
|
|
|
+ continue
|
|
|
+ tokens = tokenize_assignment(expr)
|
|
|
+ zh_pattern, en_template = build_template(tokens)
|
|
|
+ if zh_pattern and zh_pattern not in seen:
|
|
|
+ seen.add(zh_pattern)
|
|
|
+ all_templates.append((zh_pattern, en_template))
|
|
|
+
|
|
|
+ lines = [
|
|
|
+ '<?xml version="1.0" encoding="utf-8"?>',
|
|
|
+ '<!-- 由 scripts/extract_db_templates.py 从存储过程自动生成 -->',
|
|
|
+ '<!-- 请把 en/vi/id 属性中的中文占位文本替换为真实翻译 -->',
|
|
|
+ '<Templates>',
|
|
|
+ ]
|
|
|
+ for zh, en in all_templates:
|
|
|
+ lines.append(
|
|
|
+ f' <Template zh="{escape_attr(zh)}"'
|
|
|
+ f' en="{escape_attr(en)}"'
|
|
|
+ f' vi="{escape_attr(en)}"'
|
|
|
+ f' id="{escape_attr(en)}"/>'
|
|
|
+ )
|
|
|
+ lines.append("</Templates>")
|
|
|
+
|
|
|
+ with open(args.output, "w", encoding="utf-8") as f:
|
|
|
+ f.write("\n".join(lines))
|
|
|
+
|
|
|
+ print(f"Generated {len(all_templates)} templates -> {args.output}")
|
|
|
+
|
|
|
+
|
|
|
+if __name__ == "__main__":
|
|
|
+ main()
|
callm