39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
from .base import BaseEngine
|
|
|
|
CALL_NODE_TYPES = {
|
|
"Python3": "call",
|
|
"C": "call_expression",
|
|
}
|
|
|
|
|
|
class _MethodCallBase(BaseEngine):
|
|
def _find_method_calls(self, root, method_name, language):
|
|
if language == "C":
|
|
return []
|
|
call_type = CALL_NODE_TYPES.get(language, "call")
|
|
calls = self.collect_nodes(root, call_type)
|
|
matches = []
|
|
for call in calls:
|
|
func_node = call.child_by_field_name("function")
|
|
if func_node and func_node.type == "attribute":
|
|
attr_node = func_node.child_by_field_name("attribute")
|
|
if attr_node and attr_node.text.decode() == method_name:
|
|
matches.append(call)
|
|
return matches
|
|
|
|
|
|
class MustCallMethodEngine(_MethodCallBase):
|
|
def check(self, tree, rule, language, mapping):
|
|
target = rule["target"]
|
|
if not self._find_method_calls(tree.root_node, target, language):
|
|
return [rule.get("message", f"必须调用 .{target}()")]
|
|
return []
|
|
|
|
|
|
class MustNotCallMethodEngine(_MethodCallBase):
|
|
def check(self, tree, rule, language, mapping):
|
|
target = rule["target"]
|
|
if self._find_method_calls(tree.root_node, target, language):
|
|
return [rule.get("message", f"不能调用 .{target}()")]
|
|
return []
|