From 6683fd1c76373d5627f96e323d9c3f667f6a0837 Mon Sep 17 00:00:00 2001 From: ple5f8o2u <2440362584@qq.com> Date: Mon, 27 May 2024 17:14:39 +0800 Subject: [PATCH] ADD file via upload --- hotelManager.py | 278 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 278 insertions(+) create mode 100644 hotelManager.py diff --git a/hotelManager.py b/hotelManager.py new file mode 100644 index 0000000..7e790b8 --- /dev/null +++ b/hotelManager.py @@ -0,0 +1,278 @@ +users = [ + {'username': 'woo1', 'password': '123'}, + {'username': 'woo2', 'password': '123'}, + {'username': 'woo3', 'password': '123'} +] +room_data = [] +# 登录函数 +def login(): + print("===== 酒店管理系统登录 =====") + attempts = 3 + while attempts > 0: + username = input("请输入用户名: ") + password = input("请输入密码: ") + user = next((user for user in users if user['username'] == username), None) + if user is None: + print("用户名不存在，请重新输入。") + attempts -= 1 + elif user['password'] != password: + print("密码错误，请重新输入。") + attempts -= 1 + else: + print("登录成功！") + return True + print("登录失败！程序退出。") + return False + + +# 主界面函数 +def main_menu(): + print("-----------------------------") + print("| 酒店管理系统 |") + print("-----------------------------") + print("| 1. 录入房间信息 |") + print("| 2. 查找房间 |") + print("| 3. 删除房间信息 |") + print("| 4. 修改房间信息 |") + print("| 5. 房间信息排序显示 |") + print("| 6. 统计房间总个数 |") + print("| 7. 显示所有房间信息 |") + print("| 0. 退出系统 |") + print("-----------------------------") + choice = input("请输入选项(0-7): ") + return choice +def hotel_management_system(): + + if not login(): + return + room_data = read_room_data(); + while True: + choice = main_menu() + if choice == "1": + add_room(room_data) + elif choice == "2": + search_room(room_data) + elif choice == "3": + delete_room(room_data) + elif choice == "4": + modify_room(room_data) + elif choice == "5": + sort_room_info(room_data) + elif choice == "6": + count_rooms(room_data) + elif choice == "7": + display_all_rooms(room_data) + elif choice == "0": + exit_system(room_data) + else: + print("无效的选择。") +def read_room_data(): + temp_data = [] + try: + with open("room_data.txt", "r") as file: + for line in file: + room = line.strip().split(",") + room[0] = int(room[0]) + temp_data.append(room) + print("成功读取房间信息。") + except FileNotFoundError: + print("未找到房间信息文件。") + return temp_data +def print_room_info(room): + print("房间号:", room[0]) + print("房间地址:", room[1]) + print("房间类型:", room[2]) + print("房间价格:", room[3]) + print("房间状态:", room[4]) + print("--------------------------") +def save_room_data(room_data): + # try: + # with open("room_data.txt", "w") as file: + # for room in room_data: + # file.write(','.join(room) + "\n") + # print("房间信息已成功保存。") + # except: + # print("保存房间信息时发生错误。") + with open("room_data.txt", "w") as file: + for room in room_data: + file.write(",".join(str(item) for item in room) + "\n") + +# 酒店管理系统主程序 + + +def add_room(room_data): + room_number = int(input("请输入房间号: ")) + room_address = input("请输入房间地址: ") + room_type = input("请输入房间类型: ") + room_price = float(input("请输入房间价格: ")) + room_status = input("请输入房间状态: ") + new_room = [room_number,room_address,room_type,room_price,room_status] + # new_room = f"{room_number},{room_address},{room_type},{room_price},{room_status}" + room_data.append(new_room) + print("房间信息已成功录入。") + return_last_menu = input("是否继续录入房间信息(1/0)") + if return_last_menu =="1": + add_room(room_data); + return + + +def search_room(room_data): + search_choice = input("请选择查找方式 (1.房间号 2.房间类型 3.房间状态): ") + if search_choice == "1": + room_number = int(input("请输入房间号: ")) + for room in room_data: + if room[0] == room_number: + print("找到匹配的房间信息:") + print_room_info(room) + return_last_menu = input("是否继续查询房间信息(1/0)") + if return_last_menu == "1": + search_room(room_data) + return + print("未找到匹配的房间信息。") + elif search_choice == "2": + room_type = input("请输入房间类型: ") + found_rooms = [room for room in room_data if room[2] == room_type] + if found_rooms: + print("找到匹配的房间信息:") + for room in found_rooms: + print_room_info(room) + return_last_menu = input("是否继续查询房间信息(1/0)") + if return_last_menu == "1": + search_room(room_data) + return + else: + print("未找到匹配的房间信息。") + elif search_choice == "3": + room_status = input("请输入房间状态: ") + found_rooms = [room for room in room_data if room[4] == room_status] + if found_rooms: + print("找到匹配的房间信息:") + for room in found_rooms: + print_room_info(room) + return_last_menu = input("是否继续查询房间信息(1/0)") + if return_last_menu == "1": + search_room(room_data) + return + else: + print("未找到匹配的房间信息。") + else: + print("无效的选择。") +def delete_room(room_data): + delete_choice = input("请选择删除方式 (1.房间号 2.房间地址): ") + if delete_choice == "1": + room_number = int(input("请输入要删除的房间号: ")) + found_rooms = [room for room in room_data if room[0] == room_number] + if found_rooms: + confirm = input("找到匹配的房间信息，确认要删除吗？(Y/N): ") + if confirm.lower() == "y": + room_data.remove(found_rooms[0]) + print("房间信息已成功删除。") + return_last_menu = input("是否继续删除房间信息(1/0)") + if return_last_menu == "1": + delete_room(room_data) + return + else: + print("已取消删除操作。") + else: + print("未找到要删除的房间信息。") + elif delete_choice == "2": + room_address = input("请输入要删除的房间地址: ") + found_rooms = [room for room in room_data if room[1] == room_address] + if found_rooms: + confirm = input("找到匹配的房间信息，确认要删除吗？(Y/N): ") + if confirm.lower() == "y": + for room in found_rooms: + room_data.remove(room) + print("房间信息已成功删除。") + return_last_menu = input("是否继续删除房间信息(1/0)") + if return_last_menu == "1": + delete_room(room_data) + return + else: + print("已取消删除操作。") + else: + print("未找到要删除的房间信息。") + else: + print("无效的选择。") +def modify_room(room_data): + room_number = int(input("请输入要修改的房间号: ")) + found_rooms = [room for room in room_data if room[0] == room_number] + if found_rooms: + print("找到匹配的房间信息:") + print_room_info(found_rooms[0]) + modify_choice = input("请选择要修改的信息 (1.房间状态 2.房间类型 3.房间价格): ") + if modify_choice == "1": + new_status = input("请输入新的房间状态: ") + for room in room_data: + if room[0] == room_number: + room[4] = new_status + print_room_info(room) + print("房间信息已成功修改。") + + return_last_menu = input("是否继续修改房间信息(1/0)") + if return_last_menu == "1": + modify_room(room_data) + elif modify_choice == "2": + new_type = input("请输入新的房间类型: ") + for room in room_data: + if room[0] == room_number: + room[2] = new_type + print_room_info(room) + print("房间信息已成功修改。") + return_last_menu = input("是否继续修改房间信息(1/0)") + if return_last_menu == "1": + modify_room(room_data) + elif modify_choice == "3": + new_price = float(input("请输入新的房间价格: ")) + for room in room_data: + if room[0] == room_number: + room[3] = new_price + print_room_info(room) + print("房间信息已成功修改。") + return_last_menu = input("是否继续修改房间信息(1/0)") + if return_last_menu == "1": + modify_room(room_data) + else: + print("无效的选择。") + else: + print("未找到要修改的房间信息。") + +def sort_room_info(room_data): + sort_choice = input("请选择排序方式 (1.根据房间号 2.根据房间价格): ") + if sort_choice == "1": + sorted_data = sorted(room_data, key=lambda x: x[0]) # 根据房间号排序 + elif sort_choice == "2": + sorted_data = sorted(room_data, key=lambda x: x[3]) # 根据房间价格排序 + else: + print("无效的选择。") + return + + order_choice = input("请选择排序顺序 (1.升序 2.降序): ") + if order_choice == "2": + sorted_data.reverse() # 降序 + print("排序后的房间信息:") + for room in sorted_data: + print_room_info(room) + return_last_menu = input("是否继续排序房间信息(1/0)") + if return_last_menu == "1": + sort_room_info(room_data) + return +def count_rooms(room_data): + total_rooms = len(room_data) + print("房间总个数:", total_rooms) + return_last_menu = input("返回主界面(1):") + return +def display_all_rooms(room_data): + print("所有房间信息:") + for room in room_data: + print_room_info(room) + return_last_menu = input("返回主界面(1):") + return +def exit_system(room_data): + save_choice = input("是否保存房间信息并退出？(Y/N): ") + if save_choice.lower() == "y": + save_room_data(room_data) + print("感谢使用酒店管理系统，再见！") + exit() + +hotel_management_system() \ No newline at end of file