| # Copyright (c) 2021, Frappe Technologies Pvt. Ltd. and Contributors |
| # MIT License. See license.txt |
| def get_all_nodes(method, company): |
| """Recursively gets all data from nodes""" |
| method = frappe.get_attr(method) |
| if method not in frappe.whitelisted: |
| frappe.throw(_("Not Permitted"), frappe.PermissionError) |
| root_nodes = method(company=company) |
| data = method(root.id, company) |
| result.append(dict(parent=root.id, parent_name=root.name, data=data)) |
| [{"id": d.get("id"), "name": d.get("name")} for d in data if d.get("expandable")] |
| 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)) |
| nodes_to_expand.append({"id": d.get("id"), "name": d.get("name")}) |