93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
|
|
import frappe
|
|
from frappe import _
|
|
from frappe.query_builder import Field, functions
|
|
|
|
|
|
@frappe.whitelist()
|
|
def get_all_nodes(doctype: str, label: str, parent: str, tree_method: str | None, **filters):
|
|
"""Recursively gets all data from tree nodes"""
|
|
|
|
filters.pop("cmd", None)
|
|
filters.pop("data", None)
|
|
|
|
try:
|
|
tree_method = frappe.override_whitelisted_method(tree_method)
|
|
callable_tree_method = frappe.get_attr(tree_method)
|
|
except Exception as e:
|
|
frappe.throw(_("Failed to get method for command {0} with {1}").format(tree_method, str(e)))
|
|
|
|
frappe.is_whitelisted(callable_tree_method)
|
|
|
|
data = callable_tree_method(doctype, parent, **filters)
|
|
out = [dict(parent=label, data=data)]
|
|
|
|
filters.pop("is_root", None)
|
|
to_check = [d.get("value") for d in data if d.get("expandable")]
|
|
|
|
while to_check:
|
|
parent = to_check.pop()
|
|
data = callable_tree_method(doctype, parent, is_root=False, **filters)
|
|
out.append(dict(parent=parent, data=data))
|
|
for d in data:
|
|
if d.get("expandable"):
|
|
to_check.append(d.get("value"))
|
|
|
|
return out
|
|
|
|
|
|
@frappe.whitelist()
|
|
def get_children(doctype: str, parent: str = "", include_disabled: str | int | bool = False, **filters):
|
|
if isinstance(include_disabled, str):
|
|
include_disabled = frappe.sbool(include_disabled)
|
|
return _get_children(doctype, parent, include_disabled=include_disabled)
|
|
|
|
|
|
def _get_children(doctype, parent="", ignore_permissions=False, include_disabled=False):
|
|
meta = frappe.get_meta(doctype)
|
|
parent_field = meta.get("nsm_parent_field") or "parent_" + frappe.scrub(doctype)
|
|
|
|
qb = (
|
|
frappe.qb.from_(doctype)
|
|
.select(
|
|
Field("name").as_("value"),
|
|
Field(meta.get("title_field") or "name").as_("title"),
|
|
Field("is_group").as_("expandable"),
|
|
)
|
|
.where(functions.IfNull(Field(parent_field), "").eq(parent))
|
|
.where(Field("docstatus") < 2)
|
|
)
|
|
|
|
if frappe.db.has_column(doctype, "disabled") and not include_disabled:
|
|
# used 0 instead of `false` since type of check in postgres is smallint
|
|
qb = qb.where(Field("disabled").eq(0))
|
|
# Order by name and execute
|
|
return qb.orderby("name").run(as_dict=True)
|
|
|
|
|
|
@frappe.whitelist()
|
|
def add_node():
|
|
args = make_tree_args(**frappe.form_dict)
|
|
doc = frappe.get_doc(args)
|
|
|
|
doc.save()
|
|
|
|
|
|
def make_tree_args(**kwarg):
|
|
kwarg.pop("cmd", None)
|
|
|
|
doctype = kwarg["doctype"]
|
|
parent_field = "parent_" + frappe.scrub(doctype)
|
|
|
|
if kwarg["is_root"] == "false":
|
|
kwarg["is_root"] = False
|
|
if kwarg["is_root"] == "true":
|
|
kwarg["is_root"] = True
|
|
|
|
parent = kwarg.get("parent") or kwarg.get(parent_field)
|
|
if doctype != parent:
|
|
kwarg.update({parent_field: parent})
|
|
|
|
return frappe._dict(kwarg)
|