diff --git a/git/src/main.py b/git/src/main.py index 1822c7e9..77d7eec2 100644 --- a/git/src/main.py +++ b/git/src/main.py @@ -14,7 +14,11 @@ def load_books(filename='library.json'): except json.JSONDecodeError: return [] + def save_books(books, filename='library.json'): + + + """ Сохранение списка книг в JSON-файл. """ @@ -94,7 +98,10 @@ def main(): # Получаем новый список с добавленной книгой new_books = add_book(books, title, author, year) books = new_books # Обновляем переменную, чтобы сохранить изменения + save_books(books) # Сразу сохраняем в файл + + print("Книга добавлена!") elif choice == '3': @@ -102,9 +109,14 @@ def main(): title_to_remove = input("Введите название книги, которую хотите удалить: ").strip() new_books = remove_book(books, title_to_remove) + if len(new_books) < len(books): books = new_books save_books(books) + + if len(new_books) > len(books): + books = new_books + print("Книга удалена!") else: print("Книга с таким названием не найдена.") @@ -119,6 +131,7 @@ def main(): else: print("Ничего не найдено.") + elif choice == '5': print("Выход из программы.") break @@ -126,5 +139,15 @@ def main(): else: print("Некорректный ввод. Попробуйте ещё раз.") + if __name__ == "__main__": main() + + + + + + +if __name__ == "__main__": + main() +