# -*- coding: utf-8 -*-
"""

    tbbench.models
    --------------

    django-treebeard benchmark models

    :copyright: 2008-2010 by Gustavo Picon
    :license: Apache License 2.0

"""

from django.db import models
from treebeard.mp_tree import MP_Node
from treebeard.al_tree import AL_Node
from treebeard.ns_tree import NS_Node
try:
    import mptt
except ImportError:
    mptt = None



class TbNode(MP_Node):
    numval = models.IntegerField()
    strval = models.CharField(max_length=255)


class TbSortedNode(MP_Node):
    node_order_by = ['numval', 'strval']

    numval = models.IntegerField()
    strval = models.CharField(max_length=255)


class AlNode(AL_Node):
    parent = models.ForeignKey('self',
                               related_name='children_set',
                               null=True,
                               db_index=True)
    sib_order = models.PositiveIntegerField()
    numval = models.IntegerField()
    strval = models.CharField(max_length=255)


class AlSortedNode(AL_Node):
    parent = models.ForeignKey('self',
                               related_name='children_set',
                               null=True,
                               db_index=True)
    node_order_by = ['numval', 'strval']
    numval = models.IntegerField()
    strval = models.CharField(max_length=255)


class NsNode(NS_Node):
    numval = models.IntegerField()
    strval = models.CharField(max_length=255)


class NsSortedNode(NS_Node):
    node_order_by = ['numval', 'strval']

    numval = models.IntegerField()
    strval = models.CharField(max_length=255)


if mptt:
    class MpttNode(models.Model):
        numval = models.IntegerField()
        strval = models.CharField(max_length=255)
        parent = models.ForeignKey('self',
                                   null=True,
                                   blank=True,
                                   related_name='children')
    mptt.register(MpttNode)

    class MpttSortedNode(models.Model):
        numval = models.IntegerField()
        strval = models.CharField(max_length=255)
        parent = models.ForeignKey('self',
                                   null=True,
                                   blank=True,
                                   related_name='children')
    mptt.register(MpttSortedNode, order_insertion_by=['numval', 'strval'])

else:
    MpttNode, MpttSortedNode = None, None