我正在尝试让一个已登录的用户填写一个表格来创建一个组。创建网上论坛时,我需要将用户自动添加到网上论坛。
对于此问题,我们正在使用两个模型-用户和组。
用户是Django提供的默认模型。
组的定义如下:
renderItem={() =>this.renderItem(library)}
我要重构的视图是:
class Group(models.Model):
name = models.CharField(max_length=255, unique=True)
admins = models.ManyToManyField(User, default=1, related_name='user_username')
all_users = models.ManyToManyField(User, default=1)
def __str__(self):
return self.name
def get_absolute_url(self):
return reverse('home')
def get_admins(self):
return ", ".join([u.username for u in self.admins.all()])
def add_admin(self, user):
self.admins.add(user)
def get_all_users(self):
return ", ".join([u.username for u in self.all_users.all()])
def add_user(self, user):
self.all_users.add(user)
self.save()
def is_admin(self, user):
if user in self.admins.all():
return True
else:
return False
这是GroupForm:
@login_required
def user_generated_group(request):
if request.method == 'POST':
form = GroupForm(request.POST)
user = request.user
if form.is_valid():
group = Group.objects.create(name=form.cleaned_data['name'])
group.add_admin(user)
group.add_user(user)
group.save()
return HttpResponseRedirect(reverse('home'))
else:
form = GroupForm()
context = {
'form': form,
'type': 'group',
'sidebar': Sidebar(request),
}
return render(request, 'form.html', context)
目标是利用Django的内置CreateView。到目前为止,重构后的视图看起来像:
class GroupForm(forms.ModelForm):
class Meta:
model = Group
fields = '__all__'
exclude = ['all_users', 'admins', ]
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields['name'].widget.attrs.update({'class': 'input'})
我还没有实现此视图的add_user和add_admin逻辑。希望我可以在其他地方管理这些方法。
我不确定从这里去哪里。这种逻辑应该由用户还是由组来处理?
如果是按小组,我应该使用经理吗?
如果由用户使用,我应该创建自定义用户模型吗?
答案 0 :(得分:1)
您可以覆盖form_valid
[Django-doc]方法,以将self.request.user
添加到组中,例如:
from django.http import HttpResponseRedirect
class CreateGroup(LoginRequiredMixin, CreateView):
model = Group
form_class = GroupForm
template_name = 'form.html'
def form_valid(self, form):
self.object = group = form.save()
group.all_users.add(self.request.user)
return HttpResponseRedirect(self.get_success_url())