def on_author_add(instance, created, **kwargs): if created: tree_item = TreeItem(title=instance.name, tree_id=2, url=instance.get_absolute_url(), author = instance) tree_item.save(force_insert=True) else: tree_item = TreeItem.objects.get(author = instance) tree_item.title = instance.name tree_item.url = instance.get_absolute_url() tree_item.save(force_update=True)
def on_category_add(instance, created, **kwargs): parent_treeitem_id = TreeItem.objects.get(url=instance.author.get_absolute_url()).id if created: tree_item = TreeItem( title=instance.title, tree_id=2, parent_id=parent_treeitem_id, url=instance.get_absolute_url()) tree_item.save(force_insert=True) else: tree_item = TreeItem.objects.get(category = instance) tree_item.title = instance.title tree_item.url = instance.get_absolute_url() tree_item.save(force_update=True)