def delete_income(income_id): '''Delete income entry''' current_user_id = session.get('logged_in_user') incomes = Income(current_user_id) # is it valid? income = incomes.get_income(income_id) if income: # revert accounts = Accounts(current_user_id) accounts.modify_account_balance(amount=-float(income.amount), account_id=income.credit_to) incomes.delete_income(income_id) flash('Income deleted') else: flash('Not a valid income entry', 'error') return redirect(url_for('income.index'))