add finance_management

Freezepop · Dec 8, 2024 · 9271ec1 · 9271ec1
1 parent beafe68
commit 9271ec1
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 21 deletions.
diff --git a/src/finance_management/authentication.java b/src/finance_management/authentication.java
@@ -27,7 +27,9 @@ public void run(Scanner scanner, Connection cursor) throws SQLException {
                 if (auth_result == 1) {
                     new processsing().run(scanner, cursor, name);
                 }
-                System.out.println(auth_result);
+                else {
+                    System.out.println("Аккаунта с такими данным не найдено, попробуйте зарегистрироваться.\n\n");
+                }
             }
             else if (answer.equals("2")) {
                 System.out.println("Введите логин:\n");

diff --git a/src/finance_management/processsing.java b/src/finance_management/processsing.java
@@ -23,45 +23,55 @@ public void run(Scanner scanner, Connection cursor, String name) throws SQLExcep
                     4 - Выполнить расчет бюджета
                     5 - Подкрутить бюджет для категории
                     6 - СБП по имени
-                    7 - Выход из аккаунта, возврат к меню авторизации
+                    exit - Выход из аккаунта, возврат к меню авторизации
                     """);
 
         while (run) {
 
-
-
             System.out.println(menu);
 
             String answer = scanner.nextLine();
 
             if (answer.equals("1")) {
                 System.out.println("Укажите категорию и значение в формате \"Еда: 300\".");
-                String[] income_value = scanner.nextLine().split(":");
-                String category = income_value[0];
-                String value = income_value[1].strip();
-                String category_uuid = processsing_handler.checkCategory(cursor, category, user_uuid);
-                if (category_uuid != null) {
-                    processsing_handler.writeIncome(cursor, category_uuid, value);
-                    System.out.printf("Вы успешно записали доход %s по категории %s.\n", value, category);
+                String income_value_data = scanner.nextLine();
+                if (income_value_data.contains(":")) {
+                    String[] income_value = income_value_data.split(":");
+                    String category = income_value[0];
+                    String value = income_value[1].strip();
+                    String category_uuid = processsing_handler.checkCategory(cursor, category, user_uuid);
+                    if (category_uuid != null) {
+                        processsing_handler.writeIncome(cursor, category_uuid, value);
+                        System.out.printf("Вы успешно записали доход %s по категории %s.\n", value, category);
+                    }
+                    else {
+                        System.out.println("Операция не удалась, такой категории не существует. Создайте ее выбрав соответствующий пункт меню.\n");
+                    }
                 }
                 else {
-                    System.out.println("Операция не удалась, такой категории не существует. Создайте ее выбрав соответствующий пункт меню.\n");
+                    System.out.println("Значение не распознано. Требуемый формат ввода \"Еда: 300\".");
                 }
                 processsing_handler.calculateWallet(cursor, user_uuid);
             }
 
             else if (answer.equals("2")) {
                 System.out.println("Укажите категорию и значение в формате \"Еда: 300\".");
-                String[] income_value = scanner.nextLine().split(":");
-                String category = income_value[0];
-                String value = income_value[1].strip();
-                String category_uuid = processsing_handler.checkCategory(cursor, category, user_uuid);
-                if (category_uuid != null) {
-                    processsing_handler.writeExpense(cursor, category_uuid, value);
-                    System.out.printf("Вы успешно записали расход %s по категории %s.\n", value, category);
+                String income_value_data = scanner.nextLine();
+                if (income_value_data.contains(":")) {
+                    String[] income_value = income_value_data.split(":");
+                    String category = income_value[0];
+                    String value = income_value[1].strip();
+                    String category_uuid = processsing_handler.checkCategory(cursor, category, user_uuid);
+                    if (category_uuid != null) {
+                        processsing_handler.writeExpense(cursor, category_uuid, value);
+                        System.out.printf("Вы успешно записали расход %s по категории %s.\n", value, category);
+                    }
+                    else {
+                        System.out.println("Операция не удалась, такой категории не существует. Создайте ее выбрав соответствующий пункт меню.\n");
+                    }
                 }
                 else {
-                    System.out.println("Операция не удалась, такой категории не существует. Создайте ее выбрав соответствующий пункт меню.\n");
+                    System.out.println("Значение не распознано. Требуемый формат ввода \"Еда: 300\".");
                 }
                 processsing_handler.calculateWallet(cursor, user_uuid);
             }
@@ -227,9 +237,12 @@ else if (answer.equals("6")) {
 
             }
 
-            else if (answer.equals("7")) {
+            else if (answer.equals("exit")) {
                 run = false;
             }
+            else {
+                System.out.println("Требуется выбрать пункт меню 1, 2 и т.д. или exit для выхода к меню авторизации.");
+            }
         }
     }
 }