def budgets(): """Manage budgets""" # User reached route via GET if request.method == "GET": # Get the users income income = tendie_account.getIncome(session["user_id"]) # Get the users current budgets budgets = tendie_budgets.getBudgets(session["user_id"]) # Get the users total budgeted amount budgeted = tendie_budgets.getTotalBudgeted(session["user_id"]) return render_template("budgets.html", income=income, budgets=budgets, budgeted=budgeted, deletedBudgetName=None) # User reached route via POST else: # Get the name of the budget the user wants to delete budgetName = request.form.get("delete").strip() # Delete the budget deletedBudgetName = tendie_budgets.deleteBudget( budgetName, session["user_id"]) # Render the budgets page with a success message, otherwise throw an error/apology if deletedBudgetName: # Get the users income, current budgets, and sum their budgeted amount unless they don't have any budgets (same steps as a GET for this route) income = tendie_account.getIncome(session["user_id"]) budgets = tendie_budgets.getBudgets(session["user_id"]) budgeted = tendie_budgets.getTotalBudgeted(session["user_id"]) return render_template("budgets.html", income=income, budgets=budgets, budgeted=budgeted, deletedBudgetName=deletedBudgetName) else: return apology("Uh oh! Your budget could not be deleted.")
def updatebudget(urlvar_budgetname): """Update a budget""" # User reached route via POST if request.method == "POST": # Get all of the budget info provided from the HTML form formData = list(request.form.items()) # Generate data structure to hold budget info from form budgetDict = tendie_budgets.generateBudgetFromForm(formData) # Render error message if budget name or categories contained invalid data if "apology" in budgetDict: return apology(budget["apology"]) else: # Update budget in the DB for user budget = tendie_budgets.updateBudget(urlvar_budgetname, budgetDict, session["user_id"]) # Render error message if budget name is a duplicate of another budget the user has if "apology" in budget: return apology(budget["apology"]) else: return render_template("budgetcreated.html", results=budget) # User reached route via GET else: # Get the budget details from the DB based on the budget name provided via URL. Throw an apology/error if budget can't be found. budgetID = tendie_budgets.getBudgetID(urlvar_budgetname, session["user_id"]) if budgetID is None: return apology("'" + urlvar_budgetname + "' budget does not exist") else: budget = tendie_budgets.getBudgetByID(budgetID, session["user_id"]) # Get the users income income = tendie_account.getIncome(session["user_id"]) # Get the users total budgeted amount budgeted = tendie_budgets.getTotalBudgeted(session["user_id"]) # Generate the full, updatable budget data structure (name, amount for budget w/ all categories and their budgeted amounts) budget = tendie_budgets.getUpdatableBudget(budget, session["user_id"]) # Render the budget update page return render_template("updatebudget.html", income=income, budgeted=budgeted, budget=budget)
def createbudget(): """Create a budget""" # User reached route via POST if request.method == "POST": # Make sure user has no more than 20 budgets (note: 20 is an arbitrary value) if tendie_budgets.getBudgets(session["user_id"]) is not None: budgetCount = len(tendie_budgets.getBudgets(session["user_id"])) if budgetCount >= 20: return apology("You've reached the max amount of budgets'") # Get all of the budget info provided from the HTML form formData = list(request.form.items()) # Generate data structure to hold budget info from form budgetDict = tendie_budgets.generateBudgetFromForm(formData) # Render error message if budget name or categories contained invalid data if "apology" in budgetDict: return apology(budget["apology"]) else: # Add budget to DB for user budget = tendie_budgets.createBudget(budgetDict, session["user_id"]) # Render error message if budget name is a duplicate of another budget the user has if "apology" in budget: return apology(budget["apology"]) else: return render_template("budgetcreated.html", results=budget) else: # Get the users income income = tendie_account.getIncome(session["user_id"]) # Get the users total budgeted amount budgeted = tendie_budgets.getTotalBudgeted(session["user_id"]) # Get the users spend categories categories = tendie_categories.getSpendCategories(session["user_id"]) return render_template("createbudget.html", income=income, budgeted=budgeted, categories=categories)