185 lines
6.6 KiB
Python
185 lines
6.6 KiB
Python
from house_service import *
|
||
|
||
def print_menu():
|
||
print("\n===== 房屋出租系统 V3.0 =====")
|
||
print("1. 新增房源")
|
||
print("2. 删除房源")
|
||
print("3. 修改房源")
|
||
print("4. 查询房源")
|
||
print("5. 查看/管理操作记录")
|
||
print("6. 查看/管理浏览历史")
|
||
# V3.0 新增菜单
|
||
print("7. 地址字符串解析测试")
|
||
print("8. 按区域关键词模糊查询")
|
||
print("9. 初始化小区邻接矩阵")
|
||
print("10. 更新小区间距离")
|
||
print("11. 退出系统")
|
||
print("============================")
|
||
|
||
def op_record_menu():
|
||
while True:
|
||
print("\n----- 操作记录管理 -----")
|
||
print("1. 查看最近10条")
|
||
print("2. 取出最新一条")
|
||
print("3. 清空记录")
|
||
print("4. 返回")
|
||
c = input("请选择：")
|
||
if c == "1":
|
||
lst = get_recent_records("operation",10)
|
||
print(lst if lst else "暂无记录")
|
||
elif c == "2":
|
||
print(pop_operation_stack() or "暂无记录")
|
||
elif c == "3":
|
||
clear_stack("operation")
|
||
elif c == "4":
|
||
break
|
||
else:
|
||
print("输入有误")
|
||
|
||
|
||
def browse_history_menu():
|
||
while True:
|
||
print("\n----- 浏览历史管理 -----")
|
||
print("1. 查看最近10条")
|
||
print("2. 取出最新一条")
|
||
print("3. 清空历史")
|
||
print("4. 返回")
|
||
c = input("请选择：")
|
||
if c == "1":
|
||
id_list = get_recent_records("browse",10)
|
||
if not id_list:
|
||
print("暂无浏览历史")
|
||
continue
|
||
for idx,hid in enumerate(id_list,1):
|
||
info = query_house("id", hid)
|
||
if info:
|
||
h = info[0]
|
||
print(f"{idx}. ID:{hid} 地址:{h['address']}")
|
||
else:
|
||
print(f"{idx}. ID:{hid} 房源已不存在")
|
||
elif c == "2":
|
||
hid = pop_browse_stack()
|
||
if not hid:
|
||
print("暂无浏览历史")
|
||
continue
|
||
info = query_house("id", hid)
|
||
print(f"最近浏览ID:{hid}")
|
||
elif c == "3":
|
||
clear_stack("browse")
|
||
elif c == "4":
|
||
break
|
||
else:
|
||
print("输入有误")
|
||
|
||
|
||
def main():
|
||
while True:
|
||
print_menu()
|
||
choice = input("请输入功能编号：")
|
||
|
||
if choice == "1":
|
||
addr = input("输入房源地址：")
|
||
try:
|
||
price = float(input("输入月租金："))
|
||
area = float(input("输入面积："))
|
||
except:
|
||
print("租金面积必须是数字！")
|
||
continue
|
||
htype = input("输入户型：")
|
||
if add_house(addr,price,area,htype):
|
||
new_id = get_next_house_id() - 1
|
||
push_operation_stack(f"新增房源ID:{new_id} 地址:{addr}")
|
||
|
||
elif choice == "2":
|
||
try:
|
||
hid = int(input("输入要删除房源ID："))
|
||
res = query_house("id",hid)
|
||
addr = res[0]["address"] if res else "未知地址"
|
||
if delete_house(hid):
|
||
push_operation_stack(f"删除房源ID:{hid} 地址:{addr}")
|
||
except:
|
||
print("ID必须是整数")
|
||
|
||
elif choice == "3":
|
||
try:
|
||
hid = int(input("输入要修改房源ID："))
|
||
edit = {}
|
||
a = input("新地址(回车不修改)：")
|
||
if a: edit["address"] = a
|
||
p = input("新租金(回车不修改)：")
|
||
if p: edit["price"] = float(p)
|
||
ar = input("新面积(回车不修改)：")
|
||
if ar: edit["area"] = float(ar)
|
||
t = input("新户型(回车不修改)：")
|
||
if t: edit["house_type"] = t
|
||
if edit:
|
||
if update_house(hid, edit):
|
||
push_operation_stack(f"修改房源ID:{hid} 变更:{edit}")
|
||
else:
|
||
print("未填写任何修改内容")
|
||
except:
|
||
print("输入格式错误")
|
||
|
||
elif choice == "4":
|
||
print("1-ID 2-地址 3-租金 4-户型")
|
||
sel = input("选择查询类型：")
|
||
map_dic = {"1":"id","2":"address","3":"price","4":"house_type"}
|
||
if sel not in map_dic:
|
||
print("选择无效")
|
||
continue
|
||
val = input("输入查询值：")
|
||
res_list = query_house(map_dic[sel], val)
|
||
if not res_list:
|
||
print("未找到房源")
|
||
continue
|
||
for h in res_list:
|
||
print(f"ID:{h['id']} 地址:{h['address']} 租金:{h['price']}")
|
||
if map_dic[sel] == "id":
|
||
push_browse_stack(h["id"])
|
||
|
||
elif choice == "5":
|
||
op_record_menu()
|
||
elif choice == "6":
|
||
browse_history_menu()
|
||
|
||
#3.0
|
||
elif choice == "7":
|
||
# 测试地址字符串解析
|
||
full_addr = input("输入完整地址（如：四川省德阳市什邡市XX小区）：")
|
||
res = parse_address_string(full_addr)
|
||
print("解析结果：", res)
|
||
|
||
elif choice == "8":
|
||
# 按区域关键词模糊查询
|
||
keyword = input("输入区域关键词（如：什邡、德阳）：")
|
||
res_list = fuzzy_query_by_district(keyword)
|
||
if not res_list:
|
||
print("未找到该区域房源")
|
||
else:
|
||
for h in res_list:
|
||
print(f"ID:{h['id']} 地址:{h['address']}")
|
||
|
||
elif choice == "9":
|
||
# 初始化小区邻接矩阵
|
||
community_input = input("输入所有小区名称，用英文逗号分隔：")
|
||
community_list = community_input.split(",")
|
||
init_community_matrix(community_list)
|
||
|
||
elif choice == "10":
|
||
# 更新两个小区距离
|
||
c1 = input("输入小区A名称：")
|
||
c2 = input("输入小区B名称：")
|
||
try:
|
||
dis = float(input("输入两小区距离："))
|
||
update_matrix_distance(c1, c2, dis)
|
||
except:
|
||
print("距离必须是数字")
|
||
|
||
elif choice == "11":
|
||
print("系统退出成功！")
|
||
break
|
||
else:
|
||
print("请输入有效编号")
|
||
|
||
if __name__ == "__main__":
|
||
main() |