def save(self, user, role=None):
data = self.data
try:
parent_name = data.pop('parent_name')
except:
parent_name = None
data.update({'user_id': user.id})
node = Node(**data).save()
if not parent_name:
if not NodeNode.get(parent_id=1, child_id=node.id):
NodeNode(parent_id=1, child_id=node.id).save()
else:
for name in parent_name:
parent = Node.get(name=name)
if parent:
if not NodeNode.get(parent_id=parent.id, child_id=node.id):
NodeNode(parent_id=parent.id, child_id=node.id).save()
return node
def init_node():
from collipa.models import Node
if not Node.get(id=1):
Node(name=u'根节点', urlname='root', description=u'一切的根源').save()
