from house_service import ( add_house, delete_house, update_house, query_house, push_operation_stack, push_browse_stack, pop_operation_stack, pop_browse_stack, get_recent_records, clear_stack, get_next_house_id, query_by_area, add_community_distance, show_adj_matrix ) # 引入上一轮优化后的数据保存方法 from house.house_operation import save_all_data def get_valid_input(prompt, cast_type=str): """通用输入工具：自动处理类型转换异常，输入为空时返回 None""" value = input(prompt).strip() if not value: return None try: return cast_type(value) except ValueError: print(f"错误：输入内容必须是有效的{cast_type.__name__}！") return False def print_house(house): """统一房源信息的打印格式""" print( f"ID：{house['id']} | 地址：{house['address']} | 租金：{house['price']}元/月 | 面积：{house['area']}㎡ | 户型：{house['house_type']}") def print_house_list(result, is_browse_history=False): """统一打印房源列表（支持普通列表和浏览历史记录）""" if not result: print("未找到相关记录！") return for i, item in enumerate(result, 1): house = item if not is_browse_history else query_house("id", item)[0] if query_house("id", item) else None if house: prefix = f"{i}. " if not is_browse_history else "" print(f"{prefix}ID:{house['id']} | 地址:{house['address']} | 租金:{house['price']}元/月") elif is_browse_history: print(f"{i}. ID:{item}（房源已删除）") # ==================== 业务功能模块拆分 ==================== def handle_add_house(): """处理新增房源""" print("\n----- 新增房源 -----") address = input("请输入房源地址：").strip() price = get_valid_input("请输入租金（元/月）：", float) if price is False: return area = get_valid_input("请输入面积（㎡）：", float) if area is False: return house_type = input("请输入户型（如：一居室、两居室）：").strip() if add_house(address, price, area, house_type): new_id = get_next_house_id() - 1 push_operation_stack(f"新增房源ID:{new_id}，地址:{address}") print("房源新增成功！") def handle_delete_house(): """处理删除房源""" print("\n----- 删除房源 -----") house_id = get_valid_input("请输入要删除的房源ID：", int) if house_id is False or house_id is None: return house = query_house("id", house_id) house_addr = house[0]["address"] if house else "未知地址" if delete_house(house_id): push_operation_stack(f"删除房源ID:{house_id}，地址:{house_addr}") print("房源删除成功！") def handle_update_house(): """处理修改房源""" print("\n----- 修改房源 -----") house_id = get_valid_input("请输入要修改的房源ID：", int) if house_id is False or house_id is None: return new_info = {} address = input("请输入新地址（不修改按回车）：").strip() if address: new_info["address"] = address price = get_valid_input("请输入新租金（不修改按回车）：", float) if price is False: return if price: new_info["price"] = price area = get_valid_input("请输入新面积（不修改按回车）：", float) if area is False: return if area: new_info["area"] = area house_type = input("请输入新户型（不修改按回车）：").strip() if house_type: new_info["house_type"] = house_type if new_info: if update_house(house_id, new_info): push_operation_stack(f"修改房源ID:{house_id}，修改内容:{new_info}") print("房源修改成功！") else: print("未输入任何修改内容！") def handle_query_house(): """处理查询房源""" print("\n----- 查询房源 -----") cond_type_map = {"1": "id", "2": "address", "3": "price", "4": "house_type"} print("查询条件类型：1-ID 2-地址 3-租金 4-户型") cond_choice = input("请输入条件类型编号（1-4）：").strip() if cond_choice not in cond_type_map: print("错误：无效的条件类型！") return condition_type = cond_type_map[cond_choice] condition_value = input(f"请输入{condition_type}查询值：").strip() result = query_house(condition_type, condition_value) if result: print("\n查询结果：") for house in result: print_house(house) if condition_type == "id": push_browse_stack(house["id"]) else: print("未找到符合条件的房源！") def handle_operation_records(): """处理操作记录管理""" while True: print("\n----- 操作记录管理 -----") print("1. 查看最新10条操作记录\n2. 获取最新1条操作记录（并删除）\n3. 清空所有操作记录\n4. 返回主菜单") c = input("请输入操作编号：").strip() if c == "1": records = get_recent_records("operation", 10) if records: print("\n最新10条操作记录（从新到旧）：") for i, r in enumerate(records, 1): print(f"{i}. {r}") else: print("暂无操作记录！") elif c == "2": latest = pop_operation_stack() print(latest if latest else "暂无操作记录！") elif c == "3": clear_stack("operation") print("操作记录已清空！") elif c == "4": break def handle_browse_history(): """处理浏览历史管理""" while True: print("\n----- 浏览历史管理 -----") print("1. 查看最新10条浏览历史\n2. 获取最新1条浏览房源（并删除）\n3. 清空所有浏览历史\n4. 返回主菜单") c = input("请输入操作编号：").strip() if c == "1": ids = get_recent_records("browse", 10) print_house_list(ids, is_browse_history=True) elif c == "2": hid = pop_browse_stack() if hid: h = query_house("id", hid) if h: print(f"最新浏览：ID:{hid} | 地址:{h[0]['address']}") else: print(f"最新浏览ID:{hid}（已删除）") else: print("暂无浏览历史！") elif c == "3": clear_stack("browse") print("浏览历史已清空！") elif c == "4": break def handle_address_area(): """V3.0 处理地址解析与区域查询""" while True: print("\n----- 地址解析与区域查询 -----") print("1. 按区域关键词查询房源\n2. 返回主菜单") c = input("请输入操作编号：").strip() if c == "1": keyword = input("请输入区域关键词（如：朝阳、花园）：").strip() res = query_by_area(keyword) print_house_list(res) elif c == "2": break def handle_adj_matrix(): """V3.0 处理小区邻接矩阵管理""" while True: print("\n----- 小区邻接矩阵管理 -----") print("1. 设置小区间距离\n2. 查看邻接矩阵\n3. 返回主菜单") c = input("请输入操作编号：").strip() if c == "1": c1 = input("请输入小区1名称：").strip() c2 = input("请输入小区2名称：").strip() dis = get_valid_input("请输入距离（米）：", int) if dis and dis is not False: if add_community_distance(c1, c2, dis): print("距离设置成功！") else: print("设置失败，请检查小区名称是否正确！") elif c == "2": show_adj_matrix() elif c == "3": break # ==================== 主程序 ==================== def print_menu(): print("\n===== 房屋出租系统 V3.0 =====") print("1. 新增房源") print("2. 删除房源") print("3. 修改房源") print("4. 查询房源") print("5. 查看/管理操作记录") print("6. 查看/管理浏览历史") print("7. 地址解析与区域查询") print("8. 小区邻接矩阵管理") print("9. 退出系统") print("==============================") def main(): action_map = { "1": handle_add_house, "2": handle_delete_house, "3": handle_update_house, "4": handle_query_house, "5": handle_operation_records, "6": handle_browse_history, "7": handle_address_area, "8": handle_adj_matrix, } while True: print_menu() choice = input("请输入操作编号（1-9）：").strip() if choice == "9": print("正在保存数据...") save_all_data() # 退出前保存内存中的数据到磁盘 print("感谢使用房屋出租系统 V3.0，再见！") break action = action_map.get(choice) if action: action() else: print("错误：请输入1-9之间的有效编号！") if __name__ == "__main__": main()