问题描述
我为用户提供CRUD操作，只能通过admin角色执行。我有6种不同的角色，这些角色是通过创建组并将用户从管理面板分配给特定组来实现的。
我的问题是，当管理员创建新用户时，是否可以通过下拉菜单将这些组字段包含在下拉菜单中，以便他可以选择哪个组来分配新用户，而不是从管理面板中选择？
任何帮助将不胜感激：）
model.py
class CustomUserManager(BaseUserManager):
def create_user(self,email,password,**extra_fields):
if not email:
raise ValueError(_('The Email must be set'))
email = self.normalize_email(email)
user = self.model(email=email,**extra_fields)
user.set_password(password)
user.save()
return user
def create_superuser(self,**extra_fields):
extra_fields.setdefault('is_staff',True)
extra_fields.setdefault('is_superuser',True)
extra_fields.setdefault('is_active',True)
if extra_fields.get('is_staff') is not True:
raise ValueError(_('Superuser must have is_staff=True.'))
if extra_fields.get('is_superuser') is not True:
raise ValueError(_('Superuser must have is_superuser=True.'))
return self.create_user(email,**extra_fields)
class CustomUser(AbstractUser):
username = None
email = models.EmailField(_('email address'),unique=True)
USERNAME_FIELD = 'email'
required_FIELDS = []
objects = CustomUserManager()
def __str__(self):
return self.email
username = models.CharField(max_length=30,blank=True,default='')
is_superuser = models.BooleanField(default=True)
is_admin = models.BooleanField(default=True)
is_employee = models.BooleanField(default=True)
is_headofdepartment = models.BooleanField(default=True)
is_reception = models.BooleanField(default=True)
is_patient = models.BooleanField(default=True)
is_active = models.BooleanField(default=True)
is_staff = models.BooleanField(default=True)
forms.py
class UserForm(ModelForm):
class Meta:
model = CustomUser
fields = ['email','password',]
useradd.html
<h1 class="display-4">Add new user</h1>
<form action="" method="post" autocomplete="off">
{% csrf_token %}
{{ form.as_p }}
<button type="submit" class="btn btn-success">Save </button>
views.py
@login_required
@admin_only
def users(request):
context = {'users': CustomUser.objects.all()}
return render(request,'users.html',context)
@login_required
def users_add(request):
if request.method == "POST":
form = UserForm(request.POST)
if form.is_valid():
users = CustomUser.objects.create_user(**form.cleaned_data)
#login(request,users)
# redirect,or however you want to get to the main view
return redirect('Feedback:users')
else:
form = UserForm()
return render(request,'useradd.html',{'form': form})
解决方法
您可以使用ModelChoiceField在表单中包含select小部件。因此，请尝试以下代码：
forms.py
from django.contrib.auth.models import Group
from django.forms import ModelForm,ModelChoiceField
from app.model import CustomUser
class UserForm(ModelForm):
class Meta:
model = CustomUser
fields = [
'email','password',]
def __init__(self,*args,**kwargs):
super().__init__(*args,**kwargs)
self.fields['group'] = ModelChoiceField(
queryset=Group.objects.all(),empty_label='No group'
)
更改 model.py
中的CustomUserManager.create_user方法
from django.contrib.auth.models import Group
class CustomUserManager(BaseUserManager):
def create_user(self,email: str,password: str,group: Group,**extra_fields):
if not email:
raise ValueError(_('The Email must be set'))
email = self.normalize_email(email)
user = self.model(email=email,**extra_fields)
user.set_password(password)
user.save()
if group is not None:
group.user_set.add(user)
return user