python - 需要外键约束

Question

我正在 Django 中创建一个大学管理应用程序。

这是我的模型。文件：accounts/model.py

from django.db import models
from django.contrib.auth.models import AbstractUser

class CustomUser(AbstractUser):
    ROLE = {('student', 'student'),
            ('staff', 'staff'),
            ('account_manager', 'account_manager'),
            ('Admin', 'Admin')}
    role = models.CharField(choices=ROLE, default='student',
                            max_length=20, blank=True, null=True)

我正在为所有用户（员工、学生、HOD 和校长）使用内置用户类。我们可以通过角色来识别用户。

不，我想创建一个课程数据库，其中的 staff_id 将是CustomUser表的外键。有没有办法选择具有外键角色的用户？

class Course(models.Model):
    course = models.CharField(max_length=150)
    start_date = models.DateField()
    end_date = models.DateField()
    instructor = models.ForeignKey(
        CustomUser, on_delete=models.CASCADE, related_name='instructor_name')
    examinar = models.ForeignKey(
        CustomUser, on_delete=models.CASCADE, related_name='examinar_name')

    def __str__(self):
        return f'{self.course.name} Batch No: {self.batch_no}'

这里都指的是同一个CustomUser外键。这就是我添加相关名称的原因。（这是正确的方法吗？）

但是在管理页面上，如果我想添加一门新课程，我会获得所有用户。像这样：

] 1

我只想在角色是员工时才显示用户。可能吗？

标签： pythonsqldjangodjango-modelsdjango-orm

Answer 1