during the budget creation, I created a functionality to compare it w…

…ith the total amount also

cmd/root.go

@@ -9,7 +9,7 @@ import (
 	"github.com/spf13/cobra"
 )
 
-const version = "v0.1.102"
+const version = "v0.1.103"
 
 // rootCmd represents the base command when called without any subcommands
 var RootCmd = &cobra.Command{

framework_drivers/db/budget_db/budget_db.go

@@ -29,13 +29,23 @@ func (m MyBudgetDatabase) CreateBudget(bv *entities.BudgetVariables) error {
 		return err
 	}
 
-	totalAmount, ok := value[0].(int)
+	fullTotalAmount, ok := value[0].(int)
 	if !ok {
 		return errors.New("unable to convert to int")
 	}
 
+	budgetAmount, err := m.TakeBudgetAmounts()
+	if err != nil {
+		return nil
+	}
+
+	budgetTotalAmount := 0
+	for _, amount := range budgetAmount {
+		budgetTotalAmount += amount
+	}
+
 	for i := 0; i < len(includedCategory); i++ {
-		if len(bv.Category) != 0 && len(includedCategory) != 0 && includedCategory[i][0] == bv.Category && totalAmount != 0 {
+		if len(bv.Category) != 0 && len(includedCategory) != 0 && includedCategory[i][0] == bv.Category && fullTotalAmount != 0 && budgetTotalAmount+bv.Amount <= fullTotalAmount {
 			_, err = insert.Exec(bv.Category, bv.Amount, 0, 0)
 			if err != nil {
 				return err
@@ -44,7 +54,35 @@ func (m MyBudgetDatabase) CreateBudget(bv *entities.BudgetVariables) error {
 			return nil
 		}
 	}
-	return errors.New("enter the category in the total amount. see 'flow total-amount -h'")
+	return errors.New("enter the category in the total amount or set the budget below or equal to total amount. see 'flow total-amount -h'")
+}
+
+func (m MyBudgetDatabase) TakeBudgetAmounts() ([]int, error) {
+	bv := new(entities.BudgetVariables)
+	var amounts []int
+
+	db, err := m.Connection()
+	if err != nil {
+		return nil, err
+	}
+	defer db.Close()
+
+	query := "SELECT amounts FROM Budget"
+	rows, err := db.Query(query)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		err := rows.Scan(&bv.Amount)
+		if err != nil {
+			return nil, err
+		}
+		amounts = append(amounts, bv.Amount)
+	}
+
+	return amounts, nil
 }
 
 func (m MyBudgetDatabase) ViewBudget(category string) ([5]interface{}, error) {

interface_adapters/interfaces.go

@@ -39,6 +39,7 @@ type TotalAmountDatabase interface {
 
 type BudgetDatabase interface {
 	CreateBudget(bv *entities.BudgetVariables) error
+	TakeBudgetAmounts() ([]int, error)
 	ViewBudget(category string) ([5]interface{}, error)
 	RemoveBudget(category string) error
 	UpdateBudget(old, new string, amount int) error

usecases/app/spend/spend.go

@@ -49,9 +49,6 @@ func (m MySpending) SpendMoney(category string, spending_amount int) error {
 		BudgetCategoryRemainingAmount: budget_category_remaining_amount,
 	}
 
-	fmt.Println(included_categories_in_total_amount[1][0])
-	fmt.Println(included_categories_in_total_amount[0][0])
-
 	err = m.ValidBudgetValues(&sv)
 	if err != nil {
 		return err