style: format code with black
This commit is contained in:
@@ -8,11 +8,11 @@ from frappe import _
|
||||
|
||||
@frappe.whitelist()
|
||||
def get_all_nodes(method, company):
|
||||
'''Recursively gets all data from nodes'''
|
||||
"""Recursively gets all data from nodes"""
|
||||
method = frappe.get_attr(method)
|
||||
|
||||
if method not in frappe.whitelisted:
|
||||
frappe.throw(_('Not Permitted'), frappe.PermissionError)
|
||||
frappe.throw(_("Not Permitted"), frappe.PermissionError)
|
||||
|
||||
root_nodes = method(company=company)
|
||||
result = []
|
||||
@@ -21,14 +21,16 @@ def get_all_nodes(method, company):
|
||||
for root in root_nodes:
|
||||
data = method(root.id, company)
|
||||
result.append(dict(parent=root.id, parent_name=root.name, data=data))
|
||||
nodes_to_expand.extend([{'id': d.get('id'), 'name': d.get('name')} for d in data if d.get('expandable')])
|
||||
nodes_to_expand.extend(
|
||||
[{"id": d.get("id"), "name": d.get("name")} for d in data if d.get("expandable")]
|
||||
)
|
||||
|
||||
while nodes_to_expand:
|
||||
parent = nodes_to_expand.pop(0)
|
||||
data = method(parent.get('id'), company)
|
||||
result.append(dict(parent=parent.get('id'), parent_name=parent.get('name'), data=data))
|
||||
data = method(parent.get("id"), company)
|
||||
result.append(dict(parent=parent.get("id"), parent_name=parent.get("name"), data=data))
|
||||
for d in data:
|
||||
if d.get('expandable'):
|
||||
nodes_to_expand.append({'id': d.get('id'), 'name': d.get('name')})
|
||||
if d.get("expandable"):
|
||||
nodes_to_expand.append({"id": d.get("id"), "name": d.get("name")})
|
||||
|
||||
return result
|
||||
|
||||
Reference in New Issue
Block a user