如何在Django Admin中显示关系

时间:2019-04-26 13:49:47

标签: django django-models django-admin django-admin-filters

我正在使用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',)}),
     )

因此,在Admin的“检查”页面上,我可以选择医院名称,但是无法选择医院所属的单位。我需要选择医院以及它所属的单位。 我应该在考试中添加另一个外键吗?  unit_selected = models.ForeignKey(         单位,related_name ='unit_pk',on_delete = models.CASCADE)

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

我得到的是: enter image description here

1 个答案:

答案 0 :(得分:0)

要在Unit中选择Admin,需要首先将其添加到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_name中的ForeignKeyField是从外部模型回到您定义的模型的关系。因此,在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