From 5279d96c842b9dd9a924fad8d181d1bc1135a4a2 Mon Sep 17 00:00:00 2001 From: pycook Date: Fri, 22 Dec 2023 11:19:16 +0800 Subject: [PATCH] fix(api): add CI (#326) --- cmdb-api/api/lib/cmdb/ci.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/cmdb-api/api/lib/cmdb/ci.py b/cmdb-api/api/lib/cmdb/ci.py index 74d18c1..3aed0d7 100644 --- a/cmdb-api/api/lib/cmdb/ci.py +++ b/cmdb-api/api/lib/cmdb/ci.py @@ -395,8 +395,9 @@ class CIManager(object): k not in ci_type_attrs_alias and _no_attribute_policy == ExistPolicy.REJECT): return abort(400, ErrFormat.attribute_not_found.format(k)) - if limit_attrs and ci_type_attrs_name.get(k) not in limit_attrs and ( - ci_type_attrs_alias.get(k) not in limit_attrs): + _attr_name = ((ci_type_attrs_name.get(k) and ci_type_attrs_name[k].name) or + (ci_type_attrs_alias.get(k) and ci_type_attrs_alias[k].name)) + if limit_attrs and _attr_name not in limit_attrs: return abort(403, ErrFormat.ci_filter_perm_attr_no_permission.format(k)) ci_dict = {k: v for k, v in ci_dict.items() if k in ci_type_attrs_name or k in ci_type_attrs_alias}