7x31
/
cmdb
mirror of https://github.com/veops/cmdb.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: ci relation statistics

This commit is contained in:
pycook 2023-10-18 13:35:01 +08:00
parent 788dd684ab
commit 2a5091c51a
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
cmdb-api/api/lib/cmdb/search/ci_relation/search.py
View File

@ -35,7 +35,7 @@ class Search(object):
self.sort = sort or ("ci_id" if current_app.config.get("USE_ES") else None) self.sort = sort or ("ci_id" if current_app.config.get("USE_ES") else None)
self.root_id = root_id self.root_id = root_id
self.level = level self.level = level or 0
self.reverse = reverse self.reverse = reverse
def _get_ids(self): def _get_ids(self):
@ -104,16 +104,22 @@ class Search(object):
ci_ids=merge_ids).search() ci_ids=merge_ids).search()
def statistics(self, type_ids): def statistics(self, type_ids):
self.level = int(self.level)
_tmp = [] _tmp = []
ids = [self.root_id] if not isinstance(self.root_id, list) else self.root_id ids = [self.root_id] if not isinstance(self.root_id, list) else self.root_id
for l in range(0, int(self.level)): for lv in range(0, self.level):
if not l: if not lv:
_tmp = list(map(lambda x: list(json.loads(x).items()), if type_ids and lv == self.level - 1:
[i or '{}' for i in rd.get(ids, REDIS_PREFIX_CI_RELATION) or []])) _tmp = list(map(lambda x: [i for i in x if i[1] in type_ids],
(map(lambda x: list(json.loads(x).items()),
[i or '{}' for i in rd.get(ids, REDIS_PREFIX_CI_RELATION) or []]))))
else:
_tmp = list(map(lambda x: list(json.loads(x).items()),
[i or '{}' for i in rd.get(ids, REDIS_PREFIX_CI_RELATION) or []]))
else: else:
for idx, item in enumerate(_tmp): for idx, item in enumerate(_tmp):
if item: if item:
if type_ids and l == self.level - 1: if type_ids and lv == self.level - 1:
__tmp = list( __tmp = list(
map(lambda x: [(_id, type_id) for _id, type_id in json.loads(x).items() map(lambda x: [(_id, type_id) for _id, type_id in json.loads(x).items()
if type_id in type_ids], if type_id in type_ids],