django-categories/categories/models.py
2009-10-06 08:09:04 -04:00

34 lines
1.2 KiB
Python

import re
from django.core.urlresolvers import reverse
from django.db import models
from django.utils.encoding import force_unicode
import mptt
class Category(models.Model):
parent = models.ForeignKey('self',
blank=True,
null=True,
related_name="children",
help_text="Leave this blank for an Category Tree",
verbose_name='Parent')
name = models.CharField(max_length=100)
description = models.CharField(blank=True, null=True, max_length=255)
order = models.IntegerField(blank=True, null=True)
slug = models.SlugField()
def get_absolute_url(self):
"""Return a path"""
prefix = self.category_tree.get_absolute_url()
ancestors = self.get_ancestors()
return prefix + '/'.join([force_unicode(i.slug) for i in ancestors])
class Meta:
verbose_name_plural = 'categories'
unique_together = ('parent', 'name')
ordering = ('tree_id','lft')
def __unicode__(self):
ancestors = self.get_ancestors()
return ' > '.join([force_unicode(i.name) for i in ancestors]+[self.name,])
mptt.register(Category, order_insertion_by=['name'])