254 lines
8.9 KiB
Python
254 lines
8.9 KiB
Python
# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
# Tree (Hierarchical) Nested Set Model (nsm)
|
|
#
|
|
# To use the nested set model,
|
|
# use the following pattern
|
|
# 1. name your parent field as "parent_item_group" if not have a property nsm_parent_field as your field name in the document class
|
|
# 2. have a field called "old_parent" in your fields list - this identifies whether the parent has been changed
|
|
# 3. call update_nsm(doc_obj) in the on_upate method
|
|
|
|
# ------------------------------------------
|
|
from __future__ import unicode_literals
|
|
|
|
import frappe
|
|
from frappe import msgprint, _
|
|
from frappe.model.document import Document
|
|
|
|
class NestedSetRecursionError(frappe.ValidationError): pass
|
|
class NestedSetMultipleRootsError(frappe.ValidationError): pass
|
|
class NestedSetChildExistsError(frappe.ValidationError): pass
|
|
class NestedSetInvalidMergeError(frappe.ValidationError): pass
|
|
|
|
# called in the on_update method
|
|
def update_nsm(doc_obj):
|
|
# get fields, data from the DocType
|
|
opf = 'old_parent'
|
|
|
|
if str(doc_obj.__class__)=='frappe.model.doc.Document':
|
|
# passed as a Document object
|
|
d = doc_obj
|
|
pf = "parent_" + frappe.scrub(d.doctype)
|
|
else:
|
|
# passed as a DocType object
|
|
d = doc_obj.doc
|
|
pf = "parent_" + frappe.scrub(d.doctype)
|
|
|
|
if hasattr(doc_obj,'nsm_parent_field'):
|
|
pf = doc_obj.nsm_parent_field
|
|
if hasattr(doc_obj,'nsm_oldparent_field'):
|
|
opf = doc_obj.nsm_oldparent_field
|
|
|
|
p, op = d.fields.get(pf) or None, d.fields.get(opf) or None
|
|
|
|
# has parent changed (?) or parent is None (root)
|
|
if not d.lft and not d.rgt:
|
|
update_add_node(d, p or '', pf)
|
|
elif op != p:
|
|
update_move_node(d, pf)
|
|
|
|
# set old parent
|
|
d.fields[opf] = p
|
|
frappe.db.set_value(d.doctype, d.name, opf, p or '')
|
|
|
|
# reload
|
|
d._loadfromdb()
|
|
|
|
def update_add_node(doc, parent, parent_field):
|
|
"""
|
|
insert a new node
|
|
"""
|
|
from frappe.utils import now
|
|
n = now()
|
|
|
|
doctype = doc.doctype
|
|
name = doc.name
|
|
|
|
# get the last sibling of the parent
|
|
if parent:
|
|
left, right = frappe.db.sql("select lft, rgt from `tab%s` where name=%s" \
|
|
% (doctype, "%s"), parent)[0]
|
|
validate_loop(doc.doctype, doc.name, left, right)
|
|
else: # root
|
|
right = frappe.db.sql("select ifnull(max(rgt),0)+1 from `tab%s` \
|
|
where ifnull(`%s`,'') =''" % (doctype, parent_field))[0][0]
|
|
right = right or 1
|
|
|
|
# update all on the right
|
|
frappe.db.sql("update `tab%s` set rgt = rgt+2, modified=%s where rgt >= %s" %
|
|
(doctype, '%s', '%s'), (n, right))
|
|
frappe.db.sql("update `tab%s` set lft = lft+2, modified=%s where lft >= %s" %
|
|
(doctype, '%s', '%s'), (n, right))
|
|
|
|
# update index of new node
|
|
if frappe.db.sql("select * from `tab%s` where lft=%s or rgt=%s"% (doctype, right, right+1)):
|
|
frappe.msgprint("Nested set error. Please send mail to support")
|
|
raise Exception
|
|
|
|
frappe.db.sql("update `tab{0}` set lft=%s, rgt=%s, modified=%s where name=%s".format(doctype),
|
|
(right,right+1,n,name))
|
|
return right
|
|
|
|
|
|
def update_move_node(doc, parent_field):
|
|
parent = doc.fields.get(parent_field)
|
|
|
|
if parent:
|
|
new_parent = frappe.db.sql("""select lft, rgt from `tab%s`
|
|
where name = %s""" % (doc.doctype, '%s'), parent, as_dict=1)[0]
|
|
|
|
validate_loop(doc.doctype, doc.name, new_parent.lft, new_parent.rgt)
|
|
|
|
# move to dark side
|
|
frappe.db.sql("""update `tab%s` set lft = -lft, rgt = -rgt
|
|
where lft >= %s and rgt <= %s"""% (doc.doctype, '%s', '%s'), (doc.lft, doc.rgt))
|
|
|
|
# shift left
|
|
diff = doc.rgt - doc.lft + 1
|
|
frappe.db.sql("""update `tab%s` set lft = lft -%s, rgt = rgt - %s
|
|
where lft > %s"""% (doc.doctype, '%s', '%s', '%s'), (diff, diff, doc.rgt))
|
|
|
|
# shift left rgts of ancestors whose only rgts must shift
|
|
frappe.db.sql("""update `tab%s` set rgt = rgt - %s
|
|
where lft < %s and rgt > %s"""% (doc.doctype, '%s', '%s', '%s'),
|
|
(diff, doc.lft, doc.rgt))
|
|
|
|
if parent:
|
|
new_parent = frappe.db.sql("""select lft, rgt from `tab%s`
|
|
where name = %s""" % (doc.doctype, '%s'), parent, as_dict=1)[0]
|
|
|
|
|
|
# set parent lft, rgt
|
|
frappe.db.sql("""update `tab%s` set rgt = rgt + %s
|
|
where name = %s"""% (doc.doctype, '%s', '%s'), (diff, parent))
|
|
|
|
# shift right at new parent
|
|
frappe.db.sql("""update `tab%s` set lft = lft + %s, rgt = rgt + %s
|
|
where lft > %s""" % (doc.doctype, '%s', '%s', '%s'),
|
|
(diff, diff, new_parent.rgt))
|
|
|
|
# shift right rgts of ancestors whose only rgts must shift
|
|
frappe.db.sql("""update `tab%s` set rgt = rgt + %s
|
|
where lft < %s and rgt > %s""" % (doc.doctype, '%s', '%s', '%s'),
|
|
(diff, new_parent.lft, new_parent.rgt))
|
|
|
|
|
|
new_diff = new_parent.rgt - doc.lft
|
|
else:
|
|
# new root
|
|
max_rgt = frappe.db.sql("""select max(rgt) from `tab%s`""" % doc.doctype)[0][0]
|
|
new_diff = max_rgt + 1 - doc.lft
|
|
|
|
# bring back from dark side
|
|
frappe.db.sql("""update `tab%s` set lft = -lft + %s, rgt = -rgt + %s
|
|
where lft < 0"""% (doc.doctype, '%s', '%s'), (new_diff, new_diff))
|
|
|
|
def rebuild_tree(doctype, parent_field):
|
|
"""
|
|
call rebuild_node for all root nodes
|
|
"""
|
|
# get all roots
|
|
frappe.db.auto_commit_on_many_writes = 1
|
|
|
|
right = 1
|
|
result = frappe.db.sql("SELECT name FROM `tab%s` WHERE `%s`='' or `%s` IS NULL ORDER BY name ASC" % (doctype, parent_field, parent_field))
|
|
for r in result:
|
|
right = rebuild_node(doctype, r[0], right, parent_field)
|
|
|
|
frappe.db.auto_commit_on_many_writes = 0
|
|
|
|
def rebuild_node(doctype, parent, left, parent_field):
|
|
"""
|
|
reset lft, rgt and recursive call for all children
|
|
"""
|
|
from frappe.utils import now
|
|
n = now()
|
|
|
|
# the right value of this node is the left value + 1
|
|
right = left+1
|
|
|
|
# get all children of this node
|
|
result = frappe.db.sql("SELECT name FROM `tab%s` WHERE `%s`=%s" %
|
|
(doctype, parent_field, '%s'), (parent))
|
|
for r in result:
|
|
right = rebuild_node(doctype, r[0], right, parent_field)
|
|
|
|
# we've got the left value, and now that we've processed
|
|
# the children of this node we also know the right value
|
|
frappe.db.sql("""UPDATE `tab{0}` SET lft=%s, rgt=%s, modified=%s
|
|
WHERE name=%s""".format(doctype), (left,right,n,parent))
|
|
|
|
#return the right value of this node + 1
|
|
return right+1
|
|
|
|
|
|
def validate_loop(doctype, name, lft, rgt):
|
|
"""check if item not an ancestor (loop)"""
|
|
if name in frappe.db.sql_list("""select name from `tab%s` where lft <= %s and rgt >= %s""" % (doctype,
|
|
"%s", "%s"), (lft, rgt)):
|
|
frappe.throw("""Item cannot be added to its own descendents.""", NestedSetRecursionError)
|
|
|
|
class DocTypeNestedSet(Document):
|
|
def on_update(self):
|
|
update_nsm(self)
|
|
self.validate_ledger()
|
|
|
|
def on_trash(self):
|
|
if not self.nsm_parent_field:
|
|
self.nsm_parent_field = frappe.scrub(self.doc.doctype) + "_parent"
|
|
|
|
parent = self.doc.fields[self.nsm_parent_field]
|
|
if not parent:
|
|
msgprint(_("Root ") + self.doc.doctype + _(" cannot be deleted."), raise_exception=1)
|
|
|
|
# cannot delete non-empty group
|
|
has_children = frappe.db.sql("""select count(name) from `tab{doctype}`
|
|
where `{nsm_parent_field}`=%s""".format(doctype=self.doc.doctype, nsm_parent_field=self.nsm_parent_field),
|
|
(self.doc.name,))[0][0]
|
|
if has_children:
|
|
frappe.throw("{cannot_delete}. {children_exist}: {name}.".format(
|
|
children_exist=_("Children exist for"), name=self.doc.name,
|
|
cannot_delete=_("Cannot delete")), NestedSetChildExistsError)
|
|
|
|
self.doc.fields[self.nsm_parent_field] = ""
|
|
update_nsm(self)
|
|
|
|
def before_rename(self, olddn, newdn, merge=False, group_fname="is_group"):
|
|
if merge:
|
|
is_group = frappe.db.get_value(self.doc.doctype, newdn, group_fname)
|
|
if self.doc.fields[group_fname] != is_group:
|
|
frappe.throw(_("""Merging is only possible between Group-to-Group or
|
|
Ledger-to-Ledger"""), NestedSetInvalidMergeError)
|
|
|
|
def after_rename(self, olddn, newdn, merge=False):
|
|
if merge:
|
|
parent_field = "parent_" + self.doc.doctype.replace(" ", "_").lower()
|
|
rebuild_tree(self.doc.doctype, parent_field)
|
|
|
|
def validate_one_root(self):
|
|
if not self.doc.fields[self.nsm_parent_field]:
|
|
if frappe.db.sql("""select count(*) from `tab%s` where
|
|
ifnull(%s, '')=''""" % (self.doc.doctype, self.nsm_parent_field))[0][0] > 1:
|
|
frappe.throw(_("""Multiple root nodes not allowed."""), NestedSetMultipleRootsError)
|
|
|
|
def validate_ledger(self, group_identifier="is_group"):
|
|
if self.doc.fields.get(group_identifier) == "No":
|
|
if frappe.db.sql("""select name from `tab%s` where %s=%s and docstatus!=2""" %
|
|
(self.doc.doctype, self.nsm_parent_field, '%s'), (self.doc.name)):
|
|
frappe.throw(self.doc.doctype + ": " + self.doc.name +
|
|
_(" can not be marked as a ledger as it has existing child"))
|
|
|
|
def get_root_of(doctype):
|
|
"""Get root element of a DocType with a tree structure"""
|
|
result = frappe.db.sql_list("""select name from `tab%s`
|
|
where lft=1 and rgt=(select max(rgt) from `tab%s` where docstatus < 2)""" %
|
|
(doctype, doctype))
|
|
return result[0] if result else None
|
|
|
|
def get_ancestors_of(doctype, name):
|
|
"""Get ancestor elements of a DocType with a tree structure"""
|
|
lft, rgt = frappe.db.get_value(doctype, name, ["lft", "rgt"])
|
|
result = frappe.db.sql_list("""select name from `tab%s`
|
|
where lft<%s and rgt>%s order by lft desc""" % (doctype, "%s", "%s"), (lft, rgt))
|
|
return result or []
|