Django: получение записи базы данных на основе пользовательского экземпляра - PullRequest
Новая
       22

Django: получение записи базы данных на основе пользовательского экземпляра

0 голосов
/ 03 апреля 2019

У меня есть приложение, которое позволяет пациентам бронировать встречи с врачом.На портале врача я хочу получить все встречи, которые связаны с доктором, используя портал.

Проблема в том, что «Невозможно запросить« Имя человека »: должен быть экземпляр« Доктор ». После отладки я понял, чтоdoctor на самом деле относится к upin, указанному в модели Доктора вместо username.

Models.py: 

class User(AbstractUser):
    users = (
        ('doctor', 'Doctor'),
        ('patient', 'Patient'),
        ('assistant', 'Assistant'),
    )

    user_type = models.CharField(choices=users, max_length=9, default='patient')

    #return the users' name
    def __str__(self):
        return self.first_name  + " " + self.last_name

class Doctor(models.Model):
    upin = models.AutoField(primary_key=True) #unique physician identification number
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    user.user_type = "doctor"
    specialty = models.CharField(max_length=20)
    appointments_per_hour = models.IntegerField(null=True)
    picture = models.ImageField(upload_to = '', default = '')

    #return the doctors' name
    def __str__(self):
        return str(self.user)

class Patient(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    user.user_type = "patient"
    pid = models.AutoField(unique=True, primary_key=True) #patient identification

    #return the patients' name
    def __str__(self):
        return str(self.user)

@receiver(post_save, sender=User)
def create_user_profile(sender, instance, created, **kwargs):
    if created:
        if instance.user_type == "doctor":
            Doctor.objects.create(user=instance)
        elif instance.user_type == "patient":
            Patient.objects.create(user=instance)

@receiver(post_save, sender=User)
def save_user_profile(sender, instance, **kwargs):
    if instance.user_type == "doctor":
        instance.doctor.save()
    elif instance.user_type == "patient":
        instance.patient.save()

class Appointment(models.Model):
    TIMESLOT_LIST = (
        (1, '10:00 – 11:00'),
        (2, '11:00 – 12:00'),
        (3, '12:00 – 13:00'),
        (4, '13:00 – 14:00'),
        (5, '14:00 – 15:00'),
        (6, '15:00 – 16:00'),
        (7, '16:00 – 17:00'),
        (8, '17:00 – 18:00'),
        (8, '18:00 – 19:00'),
    )

    timeslot = models.IntegerField(choices=TIMESLOT_LIST, null=True)
    doctor = models.ForeignKey(Doctor, on_delete = models.CASCADE, related_name='doctor')
    patient = models.ForeignKey(Patient, on_delete = models.CASCADE, related_name='patient')


    def __str__(self):
        return str(self.doctor) + " " + str(self.patient) + " - " + str(self.get_timeslot_display())

Views.py: 

def doctor_portal(request):
    appointments = Appointment.objects.get(doctor=request.user)
    return TemplateResponse(request, 'scheduling/doctor.html', {"appointments" : appointments})

Для критериев запроса я могу использовать либо doctor, либо doctor.id. Как и ожидалось, doctor.id относится к upin в модели Доктора. Но почему doctor не относится к username доктора?

1 Ответ

0 голосов
/ 03 апреля 2019

Ваше Appointment поле модели doctor = models.ForeignKey(Doctor, on_delete = models.CASCADE, related_name='doctor') связано с Doctor, а не User моделью.

Так попробуйте это: 

def doctor_portal(request):
    appointments = Appointment.objects.get(doctor=request.user.doctor)
    return TemplateResponse(request, 'scheduling/doctor.html', {"appointments" : appointments})
...