from django.contrib.auth.models import User
class Balance(models.Model):
    amount = models.DecimalField(max_digits=12, decimal_places=2)
    owner = models.ForeignKey(User, on_delete=models.CASCADE)

    class Meta:
        verbose_name_plural = 'Balance'

@login_required
def withdraw(request):
    if request.method == 'POST':
        form = WithdrawBalance(request.POST, request.FILES)
        if form.is_valid():
            form.save()
            messages.success(request, f'Your request has been submitted. We'll soon send you money.')
            return redirect('index')
    else:
        form = WithdrawBalance()
    context = {'form': form}
    return render(request, 'nextone/withdraw.html', context)

class WithdrawBalance(forms.ModelForm):
    class Meta:
        model = WithdrawPayment
        fields = ['payment']