update

ast_checker/engines/node_exists.py

@@ -2,18 +2,28 @@ from .base import BaseEngine
 
 
 class MustExistNodeEngine(BaseEngine):
+    def _message(self, rule):
+        return rule.get("message") or f"必须使用 {rule['target']}"
+
     def check(self, tree, rule, language, mapping):
-        target = rule["target"]
-        node_type = mapping.get(target, target)
+        node_type = mapping.get(rule["target"], rule["target"])
         if not self.has_node(tree.root_node, node_type):
-            return [rule.get("message", f"必须使用 {target}")]
+            return [self._message(rule)]
         return []
 
+    def describe(self, rule, language, mapping):
+        return self._message(rule)
+
 
 class MustNotExistNodeEngine(BaseEngine):
+    def _message(self, rule):
+        return rule.get("message") or f"不能使用 {rule['target']}"
+
     def check(self, tree, rule, language, mapping):
-        target = rule["target"]
-        node_type = mapping.get(target, target)
+        node_type = mapping.get(rule["target"], rule["target"])
         if self.has_node(tree.root_node, node_type):
-            return [rule.get("message", f"不能使用 {target}")]
+            return [self._message(rule)]
         return []
+
+    def describe(self, rule, language, mapping):
+        return self._message(rule)