循环在父ForeignKey上

时间:2013-02-27 21:14:56

标签: python django

我的Django项目中有一个模型如下:

class Category(models.Model):
   parentCategory = models.ForeignKey(Category,null=True,blank=True)
   category = models.CharField(max_length=255)

   def __unicode__(self):
       return self.category

我用它来设置类别heirachy。某些类别的父类别有一个父类别也有父类别等。我想知道如何在__unicode__方法中循环这个结构?

干杯, 本

更新 以下是我的类别表中的结构:

Parent
Parent -> Sub Parent
Parent -> Sub Parent -> Sub Sub Parent

" get_name" Aamir Adnan的职能回归:

Parent
Sub Parent -> Sub Sub Parent
Sub Sub Parent

我需要它显示如下:

Parent
Parent -> Sub Parent
Parent -> Sub Parent -> Sub Sub Parent

有什么想法吗?

3 个答案:

答案 0 :(得分:3)

怎么样..

   def __unicode__(self):
       str = self.category
       obj = self
       while obj.parentCateogry:
            str += " " + obj.parentCategory.category 
            obj = obj.parentCategory
       return str

答案 1 :(得分:3)

您所描述的是树结构,遗传迭代外键关系的所有父元素是一个非常昂贵的操作;父进程的每次查找都需要一次数据库命中,其他答案则说明了进行操作的方法。

我建议你做的是使用允许有效树查询的表结构。有几种方法可以做到这一点,但为了指出正确的方向,我建议您查看django-mpttdjango-treebeard

例如,使用django-mptt可以通过以下通用结构实现此目的,这只会导致一个额外的数据库命中,以查询目标类别的所有祖先。

from django.db import models
from mptt.models import MPTTModel, TreeForeignKey


class Category(MPTTModel):
    name = models.CharField(max_length=255)
    parent = TreeForeignKey('self', null=True, blank=True, related_name='children')

    def __unicode__(self):
        ancestors = self.get_ancestors(ascending=False, include_self=True)
        return ' -> '.join(category.name
                           for category in ancestors)

答案 2 :(得分:1)

你可以这样做:

class Category(models.Model):
    parentCategory = models.ForeignKey('self', null=True, blank=True,
        related_name="categories")
    category = models.CharField(max_length=255)

    def get_name(self, obj, name=''):
        name += ' - ' + obj.category
        categories = obj.categories.all()
        if categories:
            for category in categories:
                name = self.get_name(category, name)
            return name
        else:
            return name

    def __unicode__(self):
        return self.get_name(self).strip(' - ').strip()