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

remove print

This commit is contained in:
pycook 2019-12-03 22:13:14 +08:00
parent b0e6248cd1
commit 261f673e17
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
api/lib/cmdb/preference.py
View File

@ -116,30 +116,32 @@ class PreferenceManager(object):
@staticmethod @staticmethod
def get_relation_view(): def get_relation_view():
views = PreferenceRelationView.get_by(to_dict=True) views = PreferenceRelationView.get_by(to_dict=True)
view2cr_ids = dict()
result = dict() result = dict()
name2id = list() name2id = list()
for view in views: for view in views:
result.setdefault(view['name'], []).extend(json.loads(view['cr_ids'])) view2cr_ids.setdefault(view['name'], []).extend(json.loads(view['cr_ids']))
name2id.append([view['name'], view['id']]) name2id.append([view['name'], view['id']])
id2type = dict() id2type = dict()
for view_name in result: for view_name in view2cr_ids:
for i in result[view_name]: for i in view2cr_ids[view_name]:
id2type[i['parent_id']] = None id2type[i['parent_id']] = None
id2type[i['child_id']] = None id2type[i['child_id']] = None
topo = {i['child_id']: {i['parent_id']} for i in result[view_name]} topo = {i['child_id']: {i['parent_id']} for i in view2cr_ids[view_name]}
leaf = list(set(toposort.toposort_flatten(topo)) - set([j for i in topo.values() for j in i])) leaf = list(set(toposort.toposort_flatten(topo)) - set([j for i in topo.values() for j in i]))
leaf2show_types = {i: [t['child_id'] for t in CITypeRelation.get_by(parent_id=i)] for i in leaf} leaf2show_types = {i: [t['child_id'] for t in CITypeRelation.get_by(parent_id=i)] for i in leaf}
node2show_types = copy.deepcopy(leaf2show_types) node2show_types = copy.deepcopy(leaf2show_types)
def _find_parent(node_id): def _find_parent(_node_id):
parents = topo.get(node_id, {}) parents = topo.get(_node_id, {})
for parent in parents: for parent in parents:
node2show_types.setdefault(parent, []).extend(node2show_types.get(node_id, [])) node2show_types.setdefault(parent, []).extend(node2show_types.get(_node_id, []))
_find_parent(parent) _find_parent(parent)
if not parents: if not parents:
return return
for l in leaf: for l in leaf:
_find_parent(l) _find_parent(l)
@ -156,7 +158,7 @@ class PreferenceManager(object):
for type_id in id2type: for type_id in id2type:
id2type[type_id] = CITypeCache.get(type_id).to_dict() id2type[type_id] = CITypeCache.get(type_id).to_dict()
print(result)
return result, id2type, sorted(name2id, key=lambda x: x[1]) return result, id2type, sorted(name2id, key=lambda x: x[1])
@classmethod @classmethod