30 lines
956 B
Python
30 lines
956 B
Python
# Copyright (c) 2021, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
from __future__ import unicode_literals
|
|
import frappe
|
|
from frappe import _
|
|
|
|
@frappe.whitelist()
|
|
def get_all_nodes(parent, parent_name, 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)
|
|
|
|
data = method(parent, company)
|
|
result = [dict(parent=parent, parent_name=parent_name, data=data)]
|
|
|
|
nodes_to_expand = [{'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))
|
|
for d in data:
|
|
if d.get('expandable'):
|
|
nodes_to_expand.append({'id': d.get('id'), 'name': d.get('name')})
|
|
|
|
return result
|