|
|
import os
|
|
|
import logging
|
|
|
import re # 正则表达式
|
|
|
import time
|
|
|
|
|
|
# Sub module
|
|
|
from Car import Car, ParkManage, Model, Color
|
|
|
from User import User
|
|
|
from IO import *
|
|
|
|
|
|
|
|
|
# from GUI import * # TODO
|
|
|
|
|
|
def menu_select(isAdmin, park):
|
|
|
"""显示菜单，返回str型操作码"""
|
|
|
if isAdmin == True:
|
|
|
# 管理员模式
|
|
|
while True:
|
|
|
os.system("cls")
|
|
|
print("""
|
|
|
***欢迎使用停车场管理系统***
|
|
|
## 管理员模式 ##
|
|
|
|
|
|
---北京第三区交通委提醒您---
|
|
|
| 道路千万条，安全第一条 |
|
|
|
| 行车不规范，亲人两行泪 |
|
|
|
---------------------------
|
|
|
本停车场总车位数：%3d 当前空闲车位数：%3d
|
|
|
|
|
|
1)停车
|
|
|
2)取车
|
|
|
3)显示车辆信息
|
|
|
4)查询车辆信息
|
|
|
5)编辑车辆信息
|
|
|
6)统计车辆信息
|
|
|
7)从文件加载车辆
|
|
|
|
|
|
8)切换至用户模式
|
|
|
9)关于
|
|
|
0)退出系统
|
|
|
""" % (park.max_car, park.max_car - len(park.carlist)))
|
|
|
c = input("请输入操作码(0~9):")
|
|
|
if re.match(r'(^\d$)', c):
|
|
|
if 0 <= int(c) <= 9: # c >= 0 and c <= 9
|
|
|
return c
|
|
|
else:
|
|
|
print("输入错误，请重试！")
|
|
|
else:
|
|
|
print("输入错误，请重试！")
|
|
|
'''
|
|
|
try:
|
|
|
c = int(input("请输入操作码(0~8):"))
|
|
|
except Exception as e:
|
|
|
logging.exception(e)
|
|
|
c = -1
|
|
|
continue
|
|
|
'''
|
|
|
|
|
|
|
|
|
elif isAdmin == False:
|
|
|
# 用户模式
|
|
|
while True:
|
|
|
os.system("cls")
|
|
|
print("""
|
|
|
***欢迎使用停车场管理系统***
|
|
|
## 用户模式 ##
|
|
|
|
|
|
---北京第三区交通委提醒您---
|
|
|
| 道路千万条，安全第一条 |
|
|
|
| 行车不规范，亲人两行泪 |
|
|
|
---------------------------
|
|
|
本停车场总车位数：%3d 当前空闲车位数：%3d
|
|
|
|
|
|
1)停车
|
|
|
2)取车
|
|
|
3)查询车辆信息
|
|
|
|
|
|
4)切换至管理员模式
|
|
|
5)关于
|
|
|
0)退出系统
|
|
|
""" % (park.max_car, park.max_car - len(park.carlist)))
|
|
|
c = input("请输入操作码(0~5):")
|
|
|
if re.match(r'(^\d$)', c):
|
|
|
if 0 <= int(c) <= 5: # c >= 0 and c <= 5
|
|
|
return c
|
|
|
else:
|
|
|
print("输入错误，请重试！")
|
|
|
else:
|
|
|
print("输入错误，请重试！")
|
|
|
|
|
|
'''
|
|
|
try:
|
|
|
c = int(input("请输入操作码(0~5):"))
|
|
|
except Exception as e:
|
|
|
print('请输入正确的操作码！')
|
|
|
logging.exception(e)
|
|
|
c = -1
|
|
|
continue
|
|
|
'''
|
|
|
|
|
|
|
|
|
def main():
|
|
|
park = ParkManage() # 建立并初始化车库
|
|
|
user = User() # 初始化用户
|
|
|
park = load_from_file(park, filename="cars.csv")[0] # 从文件初始化车辆信息
|
|
|
|
|
|
# user.login()
|
|
|
# isAdmin = True # TODO DEBUG 测试用
|
|
|
isAdmin = False # 程序启动时直接进入用户模式
|
|
|
|
|
|
while True:
|
|
|
choice = menu_select(isAdmin, park=park)
|
|
|
if not isAdmin:
|
|
|
# 用户
|
|
|
if choice == '0':
|
|
|
os.system("cls") # 清屏
|
|
|
write_to_file(park, filename="cars.csv") # 将车辆信息写入文件
|
|
|
print("Goodbye.\nHave a nice day!\n期待与您下一次的相遇！ \n\n系统3秒后自动关闭...")
|
|
|
time.sleep(3)
|
|
|
exit(0)
|
|
|
elif choice == '1': # 停车
|
|
|
os.system("cls") # 清屏
|
|
|
park.park(isAdmin)
|
|
|
elif choice == '2': # 取车
|
|
|
os.system("cls")
|
|
|
park.pickup()
|
|
|
os.system("pause")
|
|
|
elif choice == '3': # 查车
|
|
|
os.system("cls")
|
|
|
park.inquire()
|
|
|
os.system("pause")
|
|
|
elif choice == '4': # 切换到管理员模式
|
|
|
# isAdmin = True # DEBUG
|
|
|
os.system("cls")
|
|
|
isAdmin = user.login() # 管理员登录
|
|
|
continue
|
|
|
elif choice == '5': # 关于
|
|
|
os.system("cls") # 清屏
|
|
|
print("""
|
|
|
停车场管理系统
|
|
|
Version:0.6.0
|
|
|
Copyright by Hzj.
|
|
|
All rights reserved.
|
|
|
|
|
|
""")
|
|
|
os.system("pause")
|
|
|
else:
|
|
|
pass
|
|
|
|
|
|
elif isAdmin:
|
|
|
# 管理员
|
|
|
if choice == '0': # 退出
|
|
|
os.system("cls") # 清屏
|
|
|
write_to_file(park, filename="cars.csv") # 将车辆信息写入文件
|
|
|
print("Goodbye.\nHave a nice day!\n期待与您下一次的相遇！ \n\n系统3秒后自动关闭...")
|
|
|
time.sleep(3)
|
|
|
exit(0)
|
|
|
elif choice == '1': # 停车
|
|
|
os.system("cls") # 清屏
|
|
|
park.park(isAdmin)
|
|
|
elif choice == '2': # 取车
|
|
|
os.system("cls") # 清屏
|
|
|
park.pickup()
|
|
|
os.system("pause")
|
|
|
elif choice == '3': # 显示车辆信息
|
|
|
os.system("cls") # 清屏
|
|
|
park.display()
|
|
|
os.system("pause")
|
|
|
elif choice == '4': # 查询车辆信息
|
|
|
os.system("cls")
|
|
|
park.inquire()
|
|
|
os.system("pause")
|
|
|
elif choice == '5': # 编辑车辆信息
|
|
|
os.system("cls") # 清屏
|
|
|
park.edit()
|
|
|
os.system("pause")
|
|
|
elif choice == '6': # 统计车辆信息
|
|
|
os.system("cls") # 清屏
|
|
|
park.statistics()
|
|
|
os.system("pause")
|
|
|
elif choice == '7': # 从文件加载车辆
|
|
|
os.system("cls") # 清屏
|
|
|
park = load_new_file(park=park)
|
|
|
os.system("pause")
|
|
|
elif choice == '8': # 切换到用户模式
|
|
|
isAdmin = False
|
|
|
elif choice == '9': # 关于
|
|
|
os.system("cls") # 清屏
|
|
|
print("""
|
|
|
停车场管理系统
|
|
|
Version:0.6.0
|
|
|
Copyright by Hzj.
|
|
|
All rights reserved.
|
|
|
|
|
|
""")
|
|
|
os.system("pause")
|
|
|
else:
|
|
|
pass
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
main() |
