update
This commit is contained in:
@@ -5,12 +5,15 @@ from ast_checker.labels import label
|
|||||||
class CountNodeEngine(BaseEngine):
|
class CountNodeEngine(BaseEngine):
|
||||||
def _message(self, rule, count):
|
def _message(self, rule, count):
|
||||||
target = rule["target"]
|
target = rule["target"]
|
||||||
|
name = rule.get("label") or label(target)
|
||||||
min_count = rule.get("min")
|
min_count = rule.get("min")
|
||||||
max_count = rule.get("max")
|
max_count = rule.get("max")
|
||||||
|
if min_count is not None and min_count == max_count and count != min_count:
|
||||||
|
return rule.get("message") or f"{name} 需要出现 {min_count} 次,当前 {count} 次"
|
||||||
if min_count is not None and count < min_count:
|
if min_count is not None and count < min_count:
|
||||||
return rule.get("message") or f"{rule.get('label') or label(target)} 至少出现 {min_count} 次,当前 {count} 次"
|
return rule.get("message") or f"{name} 至少出现 {min_count} 次,当前 {count} 次"
|
||||||
if max_count is not None and count > max_count:
|
if max_count is not None and count > max_count:
|
||||||
return rule.get("message") or f"{rule.get('label') or label(target)} 至多出现 {max_count} 次,当前 {count} 次"
|
return rule.get("message") or f"{name} 至多出现 {max_count} 次,当前 {count} 次"
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def check(self, tree, rule, language, mapping):
|
def check(self, tree, rule, language, mapping):
|
||||||
@@ -22,13 +25,16 @@ class CountNodeEngine(BaseEngine):
|
|||||||
|
|
||||||
def describe(self, rule, language, mapping):
|
def describe(self, rule, language, mapping):
|
||||||
target = rule["target"]
|
target = rule["target"]
|
||||||
|
name = rule.get("label") or label(target)
|
||||||
min_count = rule.get("min")
|
min_count = rule.get("min")
|
||||||
max_count = rule.get("max")
|
max_count = rule.get("max")
|
||||||
if rule.get("message"):
|
if rule.get("message"):
|
||||||
return rule["message"]
|
return rule["message"]
|
||||||
|
if min_count is not None and min_count == max_count:
|
||||||
|
return f"{name} 出现 {min_count} 次"
|
||||||
parts = []
|
parts = []
|
||||||
if min_count is not None:
|
if min_count is not None:
|
||||||
parts.append(f"至少 {min_count} 次")
|
parts.append(f"至少 {min_count} 次")
|
||||||
if max_count is not None:
|
if max_count is not None:
|
||||||
parts.append(f"至多 {max_count} 次")
|
parts.append(f"至多 {max_count} 次")
|
||||||
return f"{rule.get('label') or label(target)} " + "、".join(parts)
|
return f"{name} " + "、".join(parts)
|
||||||
|
|||||||
Reference in New Issue
Block a user