diff --git a/src/finance_management/authentication.java b/src/finance_management/authentication.java index ab03fc7..0c25c88 100644 --- 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 index 2a11160..6793237 100644 --- 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 для выхода к меню авторизации."); + } } } } \ No newline at end of file