Django模型，保存对象

Question

我正在开发一个小型交换系统，我想给每个加入这个应用程序的人随机数量 (1-10) 的 btc。当有人成功编译注册表时，会在钱包中收到随机数量的 Btc。这是我的代码，但不起作用

models.py

class Profile(models.Model):
    _id = ObjectIdField()
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    wallet = models.FloatField()

class Order(models.Model):
    _id = ObjectIdField()
    profile = models.ForeignKey(User, on_delete=models.CASCADE)
    datetime = models.DateTimeField(auto_now_add=True)
    price = models.FloatField()
    quantity = models.FloatField()

views.py

def registerPage(request):
    bonus = randint(1,10)
    form = CreateUserForm()
    if request.method == 'POST':
        form = CreateUserForm(request.POST)
        if form.is_valid():
            form.wallet = bonus
            form.save()

            return redirect('login')

    contex = {'form':form}
    return render(request, 'app/register.html', contex)

forms.py

class CreateUserForm(UserCreationForm):
    class Meta:
        model = User
        fields = ['username', 'email', 'password1', 'password2']

我不知道这是否不重要，但我正在使用 Mongodb。谢谢

Answer 1

试试这个：

def registerPage(request):
    form = CreateUserForm()
    if request.method == 'POST':
        form = CreateUserForm(request.POST)
        if form.is_valid():
            new_user = form.save()
            bonus = random.randint(1,10)
            new_profile = Profile.objects.create(user=new_user, wallet=bonus)
            new_profile.save() # unnecessary after calling `create` but I still do this just in case :)
            return redirect('login')
    else:
        form = CreateUserForm()
    contex = {'form':form}
    return render(request, 'app/register.html', contex)