def add_funds():
"""
Allows the user to add funds to their account.
"""
if 'username' not in session:
return redirect(url_for('login'))
else:
if session['type_of_user'] == "user":
cc = str(User.get_user_info(session['username'])["credit_card"])
cc = "******" + cc[-4:]
balance = 0
if session['type_of_user'] == "developer":
balance = Developer.get_info(session['username'])["balance"]
else:
balance = Client.get_info(session['username'])["balance"]
form = AddFundsForm()
if request.method == "GET":
return render_template("addFunds.html",
cc=cc,
form=form,
balance=balance,
added=False)
elif request.method == "POST":
if form.amount.validate(form):
if session['type_of_user'] == "developer":
Developer.add_earnings(session['username'],
form.amount.data)
balance = Developer.get_info(
session['username'])["balance"]
else:
Client.add_to_balance(session['username'],
form.amount.data)
balance = Client.get_info(
session['username'])["balance"]
return render_template("addFunds.html",
cc=cc,
form=form,
balance=balance,
added=True)
else:
return render_template("addFunds.html",
cc=cc,
form=form,
balance=balance,
added=False)
else:
return render_template("access_denied.html")
