提交 2c48ee08 编写于 作者: L luxin

add group field for config file

上级 9d2a1cf3
...@@ -70,7 +70,8 @@ def ensure_config(path): ...@@ -70,7 +70,8 @@ def ensure_config(path):
if not os.path.exists(config_path): if not os.path.exists(config_path):
node = {"keywords": [], node = {"keywords": [],
"keywords_must": [], "keywords_must": [],
"keywords_forbid": []} "keywords_forbid": [],
"group": 0}
dump_json(config_path, node, exist_ok=True, override=False) dump_json(config_path, node, exist_ok=True, override=False)
return node return node
else: else:
...@@ -130,7 +131,8 @@ class TreeWalker: ...@@ -130,7 +131,8 @@ class TreeWalker:
"keywords": root.get("keywords", []), "keywords": root.get("keywords", []),
"children": [], "children": [],
"keywords_must": root.get("keywords_must", []), "keywords_must": root.get("keywords_must", []),
"keywords_forbid": root.get("keywords_forbid", []) "keywords_forbid": root.get("keywords_forbid", []),
"group": root.get("group", 0)
} }
self.tree[root["tree_name"]] = root_node self.tree[root["tree_name"]] = root_node
self.load_levels(root_node) self.load_levels(root_node)
...@@ -186,7 +188,8 @@ class TreeWalker: ...@@ -186,7 +188,8 @@ class TreeWalker:
"keywords": config["keywords"], "keywords": config["keywords"],
"children": [], "children": [],
"keywords_must": config.get("keywords_must", []), "keywords_must": config.get("keywords_must", []),
"keywords_forbid": config.get("keywords_forbid", []) "keywords_forbid": config.get("keywords_forbid", []),
"group": config.get("group", 0)
} }
} }
...@@ -239,7 +242,8 @@ class TreeWalker: ...@@ -239,7 +242,8 @@ class TreeWalker:
"keywords": [], "keywords": [],
"node_id": self.gen_node_id(), "node_id": self.gen_node_id(),
"keywords_must": [], "keywords_must": [],
"keywords_forbid": [] "keywords_forbid": [],
"group": 0
} }
dump_json(config_path, config, exist_ok=True, override=True) dump_json(config_path, config, exist_ok=True, override=True)
else: else:
...@@ -271,7 +275,8 @@ class TreeWalker: ...@@ -271,7 +275,8 @@ class TreeWalker:
"node_id": self.gen_node_id(), "node_id": self.gen_node_id(),
"keywords": [], "keywords": [],
"keywords_must": [], "keywords_must": [],
"keywords_forbid": [] "keywords_forbid": [],
"group": 0
} }
dump_json(config_path, config, exist_ok=True, override=True) dump_json(config_path, config, exist_ok=True, override=True)
else: else:
...@@ -290,7 +295,8 @@ class TreeWalker: ...@@ -290,7 +295,8 @@ class TreeWalker:
"children": [], "children": [],
"export": [], "export": [],
"keywords_must": [], "keywords_must": [],
"keywords_forbid": [] "keywords_forbid": [],
"group": 0
} }
dump_json(config_path, config, exist_ok=True, override=True) dump_json(config_path, config, exist_ok=True, override=True)
else: else:
...@@ -340,7 +346,8 @@ class TreeWalker: ...@@ -340,7 +346,8 @@ class TreeWalker:
"keywords": config["keywords"], "keywords": config["keywords"],
"children": [], "children": [],
"keywords_must": config.get("keywords_must", []), "keywords_must": config.get("keywords_must", []),
"keywords_forbid": config.get("keywords_forbid", []) "keywords_forbid": config.get("keywords_forbid", []),
"group": config.get("group", 0)
} }
} }
return num, result return num, result
...@@ -354,7 +361,8 @@ class TreeWalker: ...@@ -354,7 +361,8 @@ class TreeWalker:
"keywords": config.get("keywords", []), "keywords": config.get("keywords", []),
"children": config.get("children", []), "children": config.get("children", []),
"keywords_must": config.get("keywords_must", []), "keywords_must": config.get("keywords_must", []),
"keywords_forbid": config.get("keywords_forbid", []) "keywords_forbid": config.get("keywords_forbid", []),
"group": config.get("group", 0)
} }
} }
# if "children" in config: # if "children" in config:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册