diff --git a/src/nonebot_plugin_liteperm/nodelib.py b/src/nonebot_plugin_liteperm/nodelib.py index 3c7c172..7d2459b 100644 --- a/src/nonebot_plugin_liteperm/nodelib.py +++ b/src/nonebot_plugin_liteperm/nodelib.py @@ -2,9 +2,13 @@ from copy import deepcopy from typing import Any +from typing_extensions import Self + class Permissions: permissions_data: dict[str, str | dict | bool] + node_dict: dict[str, bool] + __permissions_str: str = "" def __init__( self, permissions_data: dict[str, str | dict | bool] | None = None @@ -13,30 +17,29 @@ def __init__( permissions_data = {} self.permissions_data = permissions_data - __permissions_str: str = "" - def __str__(self): return json.dumps(self.permissions_data) - def __search_perm(self, data, parent_key="", result=None): + def __search_perm( + self, data: dict[str, Any], parent_key="", result=None + ) -> list[tuple[str, bool]]: if result is None: result = [] - for key in data: - node = data[key] + for key, node in data.items(): current_path = f"{parent_key}.{key}" if parent_key else key # 检查当前节点权限 if node.get("has_permission", False): - result.append(f"{current_path} true") + result.append((current_path, True)) elif node.get("explicit_hasnt", False): - result.append(f"{current_path} false") + result.append((current_path, False)) if node.get("children", {}) != {}: children = node.get("children", {}) self.__search_perm(children, current_path, result) return result - def __dump_to_str( + def __dump( self, overwrite: bool = False, ): @@ -44,31 +47,45 @@ def __dump_to_str( self.__permissions_str = "" data = self.permissions_data data = deepcopy(data) - for d in self.__search_perm(data): - self.__permissions_str += f"{d}\n" - - def del_permission(self, node: str): + node_dict = {} + for n, v in self.__search_perm(data): + self.__permissions_str += f"{n} {'true' if v else 'false'}\n" + node_dict[n] = v + for line in self.__permissions_str.splitlines(): + if line: + perm, arg = line.split(" ") + node_dict[perm] = arg == "true" + self.node_dict = node_dict + + def del_permission(self, node: str) -> Self: node_parts = node.split(".") current_children: dict[str, Any] = self.permissions_data # 当前层级的子节点字典 try: for i, part in enumerate(node_parts): if part not in current_children: - return # 节点不存在，无法删除 + return self # 节点不存在，无法删除 current_node = current_children[part] if i == len(node_parts) - 1: del current_children[part] current_children = current_node["children"] finally: - self.__dump_to_str(overwrite=True) + self.__dump(overwrite=True) + return self - def set_permission(self, node: str, has_permission: bool, has_parent: bool = False): + def set_permission( + self, node: str, has_permission: bool, has_parent: bool = False + ) -> Self: node_parts = node.split(".") current_children: dict[str, Any] = self.permissions_data # 当前层级的子节点字典 for i, part in enumerate(node_parts): # 不存在创建新节点 if part not in current_children: - current_children[part] = {"has_permission": has_parent, "children": {}} + current_children[part] = { + "has_permission": has_parent, + "children": {}, + "explicit_hasnt": False, + } current_node = current_children[part] # 最后一个部分设权 if i == len(node_parts) - 1: @@ -76,35 +93,32 @@ def set_permission(self, node: str, has_permission: bool, has_parent: bool = Fal current_node["explicit_hasnt"] = not has_permission # 下一层 current_children = current_node["children"] - self.__dump_to_str(overwrite=True) + self.__dump(overwrite=True) + return self def check_permission(self, node: str) -> bool: - node_parts = node.split(".") - current_children: dict[str, Any] = self.permissions_data # 当前层级的子节点字典 - current_node = None - - for part in node_parts: - if part in current_children: - current_node = current_children[part] - current_children = current_node["children"] - elif "*" in current_children: - current_node = current_children["*"] - current_children = current_node["children"] - else: - return False # 没有找到节点或通配符 - - # 返回最终节点的权限 - return current_node["has_permission"] if current_node else False + node = node.strip() + node_dict = self.node_dict + if node_dict.get(node): + return True + current_node = "" + for part in node.split("."): + if node_dict.get(current_node + "." + "*"): + return True + current_node += ("." if current_node else "") + part + if node_dict.get(current_node): + return True + return False def dump_to_file(self, filename: str): with open(filename, "w", encoding="utf-8") as f: json.dump(self.permissions_data, f, indent=4) - self.__dump_to_str(overwrite=True) + self.__dump(overwrite=True) def load_from_json(self, filename: str): with open(filename, encoding="utf-8") as f: self.permissions_data = json.load(f) - self.__dump_to_str(overwrite=True) + self.__dump(overwrite=True) def from_perm_str(self, perm_str: str): for line in perm_str.split("\n"): @@ -123,7 +137,7 @@ def data(self) -> dict[str, Any]: @data.setter def data(self, data: dict[str, Any]): self.permissions_data = data - self.__dump_to_str(overwrite=True) + self.__dump(overwrite=True) @property def perm_str(self) -> str: @@ -131,19 +145,15 @@ def perm_str(self) -> str: @property def permissions_str(self) -> str: - self.__dump_to_str(True) + self.__dump(True) return self.__permissions_str # 此处仅用于测试 if __name__ == "__main__": - permissions = Permissions({}) - permissions.set_permission("user.read", True) - permissions.set_permission("user.write", True) + permissions = Permissions() permissions.set_permission("user.*", True) - permissions.set_permission("user", False) - permissions.set_permission("children", True) - permissions.set_permission("children.read", True) - permissions.set_permission("children.children", True) + print(permissions.check_permission("user.a")) print(permissions.permissions_str) print(json.dumps(permissions.dump_data(), indent=4)) + print(permissions.node_dict)