update

ast_checker/engines/__init__.py

@@ -1,5 +1,6 @@
 from .function_call import CountFunctionCallEngine, MustCallFunctionEngine, MustNotCallFunctionEngine
 from .method_call import MustCallMethodEngine, MustNotCallMethodEngine
+from .nesting import MustHaveNestingEngine
 from .node_count import CountNodeEngine
 from .node_exists import MustExistNodeEngine, MustNotExistNodeEngine
 from .operator import MustUseOperatorEngine
@@ -14,6 +15,7 @@ ENGINES = {
     "must_call_method": MustCallMethodEngine(),
     "must_not_call_method": MustNotCallMethodEngine(),
     "must_use_operator": MustUseOperatorEngine(),
+    "must_have_nesting": MustHaveNestingEngine(),
 }
 
 

ast_checker/engines/nesting.py

@@ -0,0 +1,33 @@
+from .base import BaseEngine
+from ast_checker.labels import label
+
+
+class MustHaveNestingEngine(BaseEngine):
+    def _has_inner_in_subtree(self, node, inner_type):
+        for child in node.children:
+            if self.has_node(child, inner_type):
+                return True
+        return False
+
+    def _message(self, rule):
+        if rule.get("message"):
+            return rule["message"]
+        outer = rule.get("outer", "")
+        inner = rule.get("inner", "")
+        outer_label = label(outer)
+        inner_label = label(inner)
+        if outer == inner:
+            return f"必须使用 {outer_label} 嵌套"
+        return f"必须在 {outer_label} 中嵌套使用 {inner_label}"
+
+    def check(self, tree, rule, language, mapping):
+        outer_type = mapping.get(rule["outer"], rule["outer"])
+        inner_type = mapping.get(rule["inner"], rule["inner"])
+        outer_nodes = self.collect_nodes(tree.root_node, outer_type)
+        for outer_node in outer_nodes:
+            if self._has_inner_in_subtree(outer_node, inner_type):
+                return []
+        return [self._message(rule)]
+
+    def describe(self, rule, language, mapping):
+        return self._message(rule)

problem/serializers.py

@@ -131,9 +131,14 @@ class ProblemAdminSerializer(BaseProblemSerializer):
 
 
 class ProblemAdminListSerializer(BaseProblemSerializer):
+    has_ast_rules = serializers.SerializerMethodField()
+
+    def get_has_ast_rules(self, obj):
+        return bool(obj.ast_rules)
+
     class Meta:
         model = Problem
-        fields = ["_id", "id", "title", "created_by", "visible", "create_time", "difficulty", "tags"]
+        fields = ["_id", "id", "title", "created_by", "visible", "create_time", "difficulty", "tags", "has_ast_rules", "allow_flowchart", "show_flowchart"]
 
 
 class ProblemSerializer(BaseProblemSerializer):