[英]How can I filter users as a choice in a foreign key django forms?
我遇到了從 django forms 中的外鍵字段過濾項目經理的問題。 我想排除超級用戶和項目經理以外的其他用戶角色。
模型.py
class Projects(models.Model):
id=models.AutoField(primary_key=True)
project_name=models.CharField(max_length=255)
project_manager=models.ForeignKey(CustomUser,on_delete=models.CASCADE, limit_choices_to={'is_project_manager__eq : 1 '})
client_name=models.ForeignKey(Clients,on_delete=models.CASCADE, null=True)
project_pic=models.ImageField(upload_to='static/website/project-images')
project_start_date=models.DateField(null=True)
project_end_date=models.DateField(null=True)
project_description=models.TextField(null=True)
created_at=models.DateTimeField(auto_now_add=True)
updated_at=models.DateTimeField(auto_now=True)
is_draft = models.BooleanField(default=True)
objects=models.Manager()
class Meta:
verbose_name_plural = 'Project'
def __str__(self):
return f'{self.project_name}'
Forms.py
class ProjectForm(forms.ModelForm):
class Meta:
model = Projects
fields = ['project_name', 'project_manager', 'client_name', 'project_pic', 'project_start_date', 'project_end_date', 'project_description']
視圖.py
def project_list(request):
projects = Projects.objects.all()
if request.method == 'POST':
form = ProjectForm(request.POST, request.FILES)
if form.is_valid():
form.save()
project_name = form.cleaned_data.get('project_name')
messages.success(request, f'{project_name} has been successfully added.')
return redirect('project_list')
else:
form = ProjectForm()
context = {
'projects' : projects,
'form' : form,
}
template_name ='project-admin/project-list.html'
return render(request, template_name, context)
自定義用戶/模型.py
這是他們自己的角色,可將他們設置為項目經理、庫存管理員和客戶
class CustomUser(AbstractUser):
is_inventory_admin = models.BooleanField('Is Inventory Admin', default=False)
is_project_manager = models.BooleanField('Is Project Manager', default=False)
is_client = models.BooleanField('Is Client', default=False)
class Meta:
db_table = 'auth_user'
您檢查條件:
class Projects(models.Model):
# …
project_manager = models.ForeignKey(
CustomUser,
on_delete=models.CASCADE,
limit_choices_to={'is_project_manager': True}
)
# …
您還可以明確排除is_inventory_admin=True
和is_client=True
(如果CustomUser
可以同時是兩個或三個),具有:
class Projects(models.Model):
# …
project_manager = models.ForeignKey(
CustomUser,
on_delete=models.CASCADE,
limit_choices_to={'is_project_manager': True, 'is_inventory_admin': False, 'is_client': False}
)
# …
注意:通常使用
settings.AUTH_USER_MODEL
[Django-doc]引用用戶 model 比直接使用User
model [Django-doc] 更好。 有關更多信息,您可以查看參考文檔的User
model部分。
注意:通常 Django model 被賦予一個單數名稱,因此
Project
而不是。Projects
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.