def balance_reset(request):
balance = Payment.get_price(Payment.objects)
balance_in_secondary_currency = None
if settings.SECONDARY_CURRENCY:
balance_in_secondary_currency = (CurrencyConverter().get_price(balance,
settings.DEFAULT_CURRENCY,
settings.SECONDARY_CURRENCY))
if request.method == "POST":
balance_reset_form = BalanceResetForm(request.POST)
if balance_reset_form.is_valid():
payment = Payment()
payment.title = settings.DEFAULT_TITLE_FOR_BALANCE_RESET
payment.currency = settings.DEFAULT_CURRENCY
payment.price = balance - balance_reset_form.cleaned_data["price"]
payment.price *= -1
payment.save()
payment.tags.add(*settings.DEFAULT_TAGS_FOR_BALANCE_RESET)
messages.success(request, "Balance was successfully reset!")
return redirect("core.index")
else:
balance_reset_form = BalanceResetForm()
return render(request, "balance_reset.html", {
"balance": balance,
"balance_in_secondary_currency": balance_in_secondary_currency,
"balance_reset_form": balance_reset_form,
})
def index(request):
if request.method == "POST":
payment_form = IndexForm(request.POST)
if payment_form.is_valid():
payment_form.save()
messages.success(request, "Payment was successfully added!")
return redirect("core.index")
else:
payment_form = IndexForm()
balance = Payment.get_price(Payment.objects)
balance_in_secondary_currency = None
if settings.SECONDARY_CURRENCY:
balance_in_secondary_currency = (CurrencyConverter().get_price(balance,
settings.DEFAULT_CURRENCY,
settings.SECONDARY_CURRENCY))
return render(request, "index.html", {
"payment_form": payment_form,
"balance": balance,
"balance_in_secondary_currency": balance_in_secondary_currency,
})
