github-starred/burp-ui
2
0
Fork 0
mirror of https://github.com/ziirish/burp-ui.git synced 2026-05-15 14:16:08 -06:00
Code Issues 1 Projects Packages Wiki Activity Actions

fix: looks like an old indentation issue just showed up

Browse source
This commit is contained in:
ziirish 2019-02-04 22:25:02 +01:00
parent 2c1b4bb7ef
commit 87fc4cb4bc
1 changed files with 55 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

110
burpui/api/client.py
View file

@ -457,66 +457,66 @@ class ClientTreeAll(Resource):
@staticmethod
def _get_tree_all(name, backup, server):
json = bui.client.get_tree(name, backup, '*', agent=server)
tree = {}
rjson = []
roots = []
json = bui.client.get_tree(name, backup, '*', agent=server)
tree = {}
rjson = []
roots = []
def __expand_json(js):
res = {}
for entry in js:
# /!\ after marshalling, 'fullname' will be 'key'
res[entry['fullname']] = marshal(entry, node_fields)
return res
def __expand_json(js):
res = {}
for entry in js:
# /!\ after marshalling, 'fullname' will be 'key'
res[entry['fullname']] = marshal(entry, node_fields)
return res
tree = __expand_json(json)
tree = __expand_json(json)
# TODO: we can probably improve this at some point
redo = True
while redo:
redo = False
for key, entry in tree.items():
parent = entry['parent']
if not entry['children']:
entry['children'] = None
if parent:
if parent not in tree:
parent2 = parent
last = False
while parent not in tree and not last:
if not parent2:
last = True
json = bui.client.get_tree(
name,
backup,
parent2,
agent=server
)
if parent2 == '/':
parent2 = ''
else:
parent2 = os.path.dirname(parent2)
tree2 = __expand_json(json)
tree.update(tree2)
roots = []
redo = True
break
node = tree[parent]
if not node['children']:
node['children'] = []
elif entry in node['children']:
continue
node['children'].append(entry)
if node['folder']:
node['lazy'] = False
node['expanded'] = False
else:
roots.append(entry['key'])
# TODO: we can probably improve this at some point
redo = True
while redo:
redo = False
for key, entry in tree.items():
parent = entry['parent']
if not entry['children']:
entry['children'] = None
if parent:
if parent not in tree:
parent2 = parent
last = False
while parent not in tree and not last:
if not parent2:
last = True
json = bui.client.get_tree(
name,
backup,
parent2,
agent=server
)
if parent2 == '/':
parent2 = ''
else:
parent2 = os.path.dirname(parent2)
tree2 = __expand_json(json)
tree.update(tree2)
roots = []
redo = True
break
node = tree[parent]
if not node['children']:
node['children'] = []
elif entry in node['children']:
continue
node['children'].append(entry)
if node['folder']:
node['lazy'] = False
node['expanded'] = False
else:
roots.append(entry['key'])
for fullname in roots:
rjson.append(tree[fullname])
for fullname in roots:
rjson.append(tree[fullname])
return rjson
return rjson
@ns.route('/report/<name>',