django - 在管理员的内联表单中限制选择中的外键选择

Question

模型的逻辑是:

• A Building有很多Rooms
• A Room可能在另一个里面 Room (例如，壁橱——“self”上的 ForeignKey)
• A Room只能在另一个Room里面在同一栋楼里(这是棘手的部分)

这是我的代码:

#spaces/models.py
from django.db import models    

class Building(models.Model):
    name=models.CharField(max_length=32)
    def __unicode__(self):
        return self.name

class Room(models.Model):
    number=models.CharField(max_length=8)
    building=models.ForeignKey(Building)
    inside_room=models.ForeignKey('self',blank=True,null=True)
    def __unicode__(self):
        return self.number

和:

#spaces/admin.py
from ex.spaces.models import Building, Room
from django.contrib import admin

class RoomAdmin(admin.ModelAdmin):
    pass

class RoomInline(admin.TabularInline):
    model = Room
    extra = 2

class BuildingAdmin(admin.ModelAdmin):
    inlines=[RoomInline]

admin.site.register(Building, BuildingAdmin)
admin.site.register(Room)

内联将仅显示当前建筑物中的房间(这正是我想要的)。但是，问题在于 inside_room下拉，它会显示 Rooms 表中的所有房间(包括其他建筑物中的房间)。

在 rooms 的内联中, 我需要限制 inside_room选择只有rooms在当前building (建筑记录目前正在被主 BuildingAdmin 表格更改)。

我想不出用 limit_choices_to 来做到这一点的方法在模型中，我也不知道如何正确地覆盖管理员的内联表单集(我觉得我应该以某种方式创建自定义内联表单，将主表单的 building_id 传递给自定义内联，然后限制查询集field 的选择基于此——但我就是不知道该怎么做)。

也许这对于管理站点来说太复杂了，但它似乎通常是有用的......

Answer 1

使用请求实例作为 obj 的临时容器。重写的内联方法 formfield_for_foreignkey修改查询集。这至少适用于 django 1.2.3。

class RoomInline(admin.TabularInline):

    model = Room

    def formfield_for_foreignkey(self, db_field, request=None, **kwargs):

        field = super(RoomInline, self).formfield_for_foreignkey(db_field, request, **kwargs)

        if db_field.name == 'inside_room':
            if request._obj_ is not None:
                field.queryset = field.queryset.filter(building__exact = request._obj_)  
            else:
                field.queryset = field.queryset.none()

        return field



class BuildingAdmin(admin.ModelAdmin):

    inlines = (RoomInline,)

    def get_form(self, request, obj=None, **kwargs):
        # just save obj reference for future processing in Inline
        request._obj_ = obj
        return super(BuildingAdmin, self).get_form(request, obj, **kwargs)