Django UserCreationForm添加request.user.pk?

时间:2016-02-13 10:33:34

标签: python django createuser

您好我需要向UserCreationForm添加请求用户pk。我尝试了很多次,但没有结果。在我的仪表板中,我有用户所有者,我有用户卖家。 所有3种用户的一种型号。所以所有者可以使用role = seller创建另一个用户,但为此我需要将param = request.user.pk的父字段添加到卖家

自定义UserModel

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


USER = 0
OWNER = 1
SELLER = 2
user_type_choice = (
    (USER, 'User'),
    (OWNER, 'Owner'),
    (SELLER, 'Seller'),
)


class User(AbstractUser):
    role = models.CharField(max_length=100, choices=user_type_choice, default=USER)
    idn = models.CharField(max_length=10, blank=True, default=None, null=True)
    mobile = models.CharField(max_length=15, blank=True, default=None, null=True)
    parent = models.ForeignKey('self', null=True, blank=True)
    tariff = models.IntegerField(default=None, null=True, blank=True)

    class Meta(AbstractUser.Meta):
        swappable = 'AUTH_USER_MODEL'

forms.py

class SellerRegisterForm(UserCreationForm):

email = forms.EmailField(required=True)
first_name = forms.CharField(required=False)
last_name = forms.CharField(required=False)

class Meta:
    model = User
    fields = ('username', 'email', 'password1', 'password2') """try add and remove 'parent' here in felds """

def save(self, commit=True):
    user = super(SellerRegisterForm, self).save(commit=False)
    user.email = self.cleaned_data['email']
    user.first_name = self.cleaned_data['first_name']
    user.last_name = self.cleaned_data['last_name']
    user.role = ROLE_SELLER
    # try add user.parent = self.request.user.pk 
    # but it does not work

    if commit:
        user.save()

    return user

views.py

def register_seller(request):
if request.method == 'POST':
    form = SellerRegisterForm(request.POST)     # create form object
    if form.is_valid():
        #try add new_form = form.save(commit=False)
        # new_form.parent = request.user.pk
        # new_form.save() but it not working

        form.save()
        return HttpResponseRedirect('/accounts/login')

else:
    form = SellerRegisterForm()
return render(request, 'register_owner.html', {'form': form})

1 个答案:

答案 0 :(得分:1)

您需要通过表单__init__方法传递用户对象。

class SellerRegisterForm(UserCreationForm):
    def __init__(self, *args, **kwargs):
        self.user = kwargs.pop('user', None)
        super(SellerRegisterForm, self).__init__(*args, **kwargs)

并在您看来:

form = SellerRegisterForm(request.POST, user=request.user)

然后只需在表单类的任何位置使用self.user