如何根据 Django forms 中的登录用户过滤值？

Question

我有 Rank、UserProfile 和 Company 模型。 这些类与外键相连。 排名为“Lead”的用户可以从他的仪表板创建新用户。 我想根据 UserProfile 的公司过滤排名模型。 在注册表单中将有一个下拉列表来选择新用户的排名。 应该只有属于 UserProfile 公司的排名。 这些是我的模型：

class CompanyProfile(models.Model):
    comp_id = models.UUIDField(default=uuid.uuid4, editable=False, unique=True)
    comp_name = models.CharField(max_length=200)
    country = models.CharField(max_length=200, default='')

    def __str__(self):
        return self.comp_name

class Rank(models.Model):
    rank_name = models.CharField(max_length=200)
    company = models.ForeignKey(CompanyProfile, on_delete=models.CASCADE, null=True, unique=False)

    def __str__(self):
        return self.rank_name


class UserProfile(AbstractUser):

    company = models.ForeignKey(CompanyProfile, on_delete=models.CASCADE, null=True, unique=False)
    user_id = models.UUIDField(default=uuid.uuid4(), editable=False, unique=True)
    username = models.CharField(max_length=500, unique=True)
    first_name = models.CharField(max_length=200)
    last_name = models.CharField(max_length=200)
    password = models.CharField(max_length=250)
    email = models.EmailField(max_length=254)
    rank = models.ForeignKey(Rank, on_delete=models.CASCADE, null=True, unique=False)
    image = models.ImageField(upload_to='profile_image', blank=True, null= True, default='profile.png')

    def __str__(self):
        return self.username

这是我的表格：

class SignUpForm(forms.ModelForm):


    password1 = forms.CharField(max_length=250)
    password2 = forms.CharField(max_length=250)
    

    class Meta:
        model = UserProfile
        fields = (
             'username', 'first_name', 'last_name', 'email', 'password1', 'password2','rank', 'image')

这是views.py：

@user_passes_test(is_lead)
@login_required
def signup(request):
    form_class = SignUpForm
    if request.method == 'POST':
  
        form = SignUpForm(request.POST)
        if form.is_valid() :

            user = form.save()
            user.refresh_from_db()  # load the profile instance created by the signal
            user.is_active = False
            if form.cleaned_data['password1'] != "":
                user.set_password(form.cleaned_data['password1'])
                user.save()
            return redirect('home')
    else:
        form = form_class()

    return render(request, 'signup.html', {'form': form})

Answer 1

您需要覆盖rank中的SignUpForm 。

类似于以下内容：

choices = [(rank.id, rank.name) for rank in Rank.objects.filter(company=...)]

rank = models.CharField(
    max_length=200,
    choices=choices,
)

我假设您以某种方式将公司传递到表单中，因此您需要调整filter以过滤您为其创建配置文件的公司。