58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""验证：权限树形化(parent_code) - seed 回填 + /permissions 返回父子结构"""
|
||
|
|
import json, urllib.request, urllib.error
|
||
|
|
|
||
|
|
BASE = "http://127.0.0.1:8902"
|
||
|
|
TOKEN = None
|
||
|
|
|
||
|
|
def req(method, path, payload=None):
|
||
|
|
url = BASE + path
|
||
|
|
data = json.dumps(payload).encode() if payload is not None else None
|
||
|
|
r = urllib.request.Request(url, data=data, method=method)
|
||
|
|
r.add_header("Content-Type", "application/json")
|
||
|
|
if TOKEN:
|
||
|
|
r.add_header("Authorization", "Bearer " + TOKEN)
|
||
|
|
try:
|
||
|
|
with urllib.request.urlopen(r, timeout=20) as resp:
|
||
|
|
return json.loads(resp.read().decode())
|
||
|
|
except urllib.error.HTTPError as e:
|
||
|
|
return {"_http": e.code, "_body": e.read().decode()[:300]}
|
||
|
|
|
||
|
|
def main():
|
||
|
|
global TOKEN
|
||
|
|
d = req("POST", "/api/auth/login", {"username": "admin", "password": "123456"})
|
||
|
|
TOKEN = (d.get("data") or {}).get("token")
|
||
|
|
assert TOKEN, f"登录失败: {d}"
|
||
|
|
print("✓ 登录成功")
|
||
|
|
|
||
|
|
# 执行 seed 回填 parent_code
|
||
|
|
sd = req("POST", "/api/rbac/seed")
|
||
|
|
print("✓ seed 完成:", sd.get("data") or sd)
|
||
|
|
|
||
|
|
# 拉取权限，校验 parent_code
|
||
|
|
pl = req("GET", "/api/permissions")
|
||
|
|
listp = (pl.get("data") or {}).get("list") or []
|
||
|
|
menus = [p for p in listp if p["type"] == "MENU"]
|
||
|
|
buttons = [p for p in listp if p["type"] == "BUTTON"]
|
||
|
|
print(f"权限总数 {len(listp)}：菜单 {len(menus)}，按钮 {len(buttons)}")
|
||
|
|
|
||
|
|
# 校验每个按钮的 parent_code 是否为有效菜单 code 或为空(全局)
|
||
|
|
menu_codes = {m["code"] for m in menus}
|
||
|
|
bad = [p for p in buttons if p.get("parentCode") and p["parentCode"] not in menu_codes]
|
||
|
|
global_btns = [p for p in buttons if not p.get("parentCode")]
|
||
|
|
print(f" 按钮中全局(无父)数: {len(global_btns)} -> {[p['code'] for p in global_btns]}")
|
||
|
|
print(f" 按钮 parent_code 有误(指向不存在菜单): {[p['code'] for p in bad]}")
|
||
|
|
assert not bad, "存在 parent_code 指向无效菜单"
|
||
|
|
print("✓ 所有按钮 parent_code 均有效(或为全局空)")
|
||
|
|
|
||
|
|
# 展示各菜单下挂的按钮（模拟前端树）
|
||
|
|
for m in menus:
|
||
|
|
kids = [p["name"] for p in buttons if p.get("parentCode") == m["code"]]
|
||
|
|
print(f" [{m['name']}] 挂按钮: {kids if kids else '(无)'}")
|
||
|
|
|
||
|
|
passed = not bad and len(global_btns) >= 1
|
||
|
|
print("\n==== 验证 %s ====" % ("通过" if passed else "失败"))
|
||
|
|
return passed
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
raise SystemExit(0 if main() else 1)
|