首页 > 解决方案 > 如何在 Django Admin 中显示关系

问题描述

我正在使用 Django 2.1,我很难弄清楚如何在 Admin 上显示关系。我有两个示范医院和单位,第三个是考试。他们是这样设置的,医院可以有多个单元,一个单元属于一个医院。考试模型具有考试名称和医院名称等考试属性。

class Hospital(models.Model):

    name = models.CharField(max_length=200)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    def __str__(self):
        return self.name

class Unit(models.Model):

    unit_name = models.CharField(max_length=200)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)
    parent_company = models.ForeignKey(
        Hospital, related_name='unit_parent', on_delete=models.CASCADE, verbose_name="Hospital")


class Exam(models.Model):

    unit_name = models.ForeignKey(
        Hospital, related_name='hospital_pk', on_delete=models.CASCADE)
    exam_name = models.CharField(max_length=255, blank=True,
                                 null=True, verbose_name="Exam's name")
    date = models.DateField()
    time = models.TimeField()

    def __str__(self):
        return self.exam_name


class CustomExamAdmin(admin.ModelAdmin):

    model = Exam
    list_display = ('exam_name', 'time', 'alert',
                    'document', 'notes', 'report', 'patients_name', 'unit_name')
    list_filter = ('exam_name', 'time')
    list_select_related = ('unit_name',)
    fieldsets = (
        ("Hospital's Name:", {
            'fields': ('unit_name',)}),
        ('Types of Exams:', {
            'fields': ('exam_name', 'modality', 'model_used')}),
        ("Patient's name:", {
            'fields': ('patients_name',)}),
     )

因此,在管理员的考试页面中,我可以选择医院名称,但是我无法选择属于医院的单位。我需要选择医院以及所属的单位。我是否应该在考试中添加另一个外键,如下所示: unit_selected = models.ForeignKey( Unit, related_name='unit_pk', on_delete=models.CASCADE)

例如: 医院:布鲁克代尔大学医院单位:市中心单位

我得到了什么: 在此处输入图像描述

标签: djangodjango-modelsdjango-admindjango-admin-filters

解决方案


为了在 中选择UnitAdmin您需要先将其添加到Exam模型中。您有一个名为 的字段unit_name,但您已将其定义为Hospital模型的外键,而不是Unit模型。如果你有一个unit_name字段,你真的不需要一个单独的hospital字段,因为你可以很容易地从unit_name.parent_company. 所以你可以这样做:

class Exam(models.Model):

    unit_name = models.ForeignKey(
        Unit, related_name='exams', on_delete=models.CASCADE)
    exam_name = models.CharField(max_length=255, blank=True,
                                 null=True, verbose_name="Exam's name")
    date = models.DateField()
    time = models.TimeField()

    def __str__(self):
        return self.exam_name

对于管理员,我认为最简单的方法是在管理员中选择医院名称时将医院名称作为单元名称的一部分,而不是单独选择医院和单位,例如Brookdale University Hospital|Downtown Unit. 这样你只需要一个下拉字段。你可以这样做:

from django import forms


class UnitChoiceField(forms.ModelChoiceField):
     def label_from_instance(self, obj):
         return "{}|{}".format(obj.parent_company.name, obj.unit_name)


class CustomExamAdmin(admin.ModelAdmin):

    model = Exam
    list_display = ('exam_name', 'time', 'alert',
                    'document', 'notes', 'report', 'patients_name', 'get_hospital', 'unit_name')
    list_filter = ('exam_name', 'time')
    list_select_related = ('unit_name__parent_company',)
    fieldsets = (
        ("Hospital's Name:", {
            'fields': ('unit_name',)}),
        ('Types of Exams:', {
            'fields': ('exam_name', 'modality', 'model_used')}),
        ("Patient's name:", {
            'fields': ('patients_name',)}),
     )

    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == 'unit_name':
            return UnitChoiceField(
                queryset=Unit.objects.all().select_related('parent_company').order_by(
                'parent_company.name', 'unit_name'))
        return super().formfield_for_foreignkey(db_field, request, **kwargs)

    def get_hospital(self, obj):
    #  required to add hospital name to list_display
        return obj.unit_name.parent_company.name           
    get_hospital.short_description = 'Hospital'

另一个注意事项:related_nameaForeignKeyField是从外部模型到您正在定义的模型的关系。所以在Unit模型中

parent_company = models.ForeignKey(
    Hospital, related_name='unit_parent', on_delete=models.CASCADE, verbose_name="Hospital")

应该

 parent_company = models.ForeignKey(
    Hospital, related_name='units', on_delete=models.CASCADE, verbose_name="Hospital")

然后从Hospital模型中您可以将单位列表称为units


推荐阅读