|
|
from enum import Enum
|
|
|
import time
|
|
|
import re
|
|
|
import os
|
|
|
import logging
|
|
|
|
|
|
Model = Enum('Model', ("小汽车", "小卡", "中卡", "大卡"))
|
|
|
Color = Enum('Color', ("白色", "黑色", "灰色", "蓝色", "红色", "黄色"))
|
|
|
unit_price = 5 # 每小时单价为5元
|
|
|
|
|
|
|
|
|
class Car(object): # ParkManage
|
|
|
def __init__(self, parknum, carnum, model, color, intime=None):
|
|
|
# super(Car, self).__init__()
|
|
|
self.parknum = parknum
|
|
|
self.carnum = carnum
|
|
|
self.model = model
|
|
|
self.color = color
|
|
|
self.intime = intime
|
|
|
|
|
|
def __setitem__(self, key, value):
|
|
|
self.__dict__[key] = value
|
|
|
|
|
|
def __getitem__(self, key):
|
|
|
return self.__dict__[key]
|
|
|
|
|
|
def __str__(self):
|
|
|
return "%4s %-8s %-6s %-s %s" % \
|
|
|
(self.parknum, self.carnum, self.model.name, self.color.name, self.intime)
|
|
|
|
|
|
|
|
|
class ParkManage(object):
|
|
|
def __init__(self, max_car=150):
|
|
|
self.max_car = max_car
|
|
|
self.carlist = []
|
|
|
# self.carnum = len(self.carlist)
|
|
|
|
|
|
def check_empty(self, model=1):
|
|
|
"""
|
|
|
查找空闲车位，model=1默认按小汽车查找
|
|
|
:return:{list}空闲车位号列表
|
|
|
"""
|
|
|
if model == 1:
|
|
|
# t_num = [parkedCar["parknum"] for parkedCar in self.carlist]
|
|
|
"""
|
|
|
t_list = []
|
|
|
for i in range(self.max_car):
|
|
|
if i not in t_num:
|
|
|
t_list.append(i)
|
|
|
return t_list
|
|
|
"""
|
|
|
# return [i for i in range(self.max_car) if i not in t_num]
|
|
|
return [i for i in range(self.max_car) if i not in
|
|
|
[parkedCar["parknum"] for parkedCar in self.carlist]]
|
|
|
|
|
|
def inquire_empty(self):
|
|
|
"""
|
|
|
查询空闲车位，调用即可进行输出。
|
|
|
:return: None
|
|
|
"""
|
|
|
t_i = 0
|
|
|
for t_empty in self.check_empty():
|
|
|
print("%3s " % t_empty, end='')
|
|
|
t_i += 1
|
|
|
if len(self.check_empty()) > 15 and t_i % 15 == 0: # 每15个一行
|
|
|
print('')
|
|
|
|
|
|
def check_car_num(self, carnum):
|
|
|
"""
|
|
|
基于正则表达式判断车牌号是否合法，合法为True
|
|
|
:return:{bool}True/False
|
|
|
"""
|
|
|
# return True # TODO DEBUG
|
|
|
car_license = re.compile(u"^[\u4e00-\u9fa5][A-Z][A-Z0-9]{5}$") # e.g.: 苏BCD123
|
|
|
if car_license.match(carnum):
|
|
|
return True
|
|
|
else:
|
|
|
return False
|
|
|
|
|
|
def park(self, isAdmin):
|
|
|
"""
|
|
|
输入车辆信息，停车。管理员isAdmin==1时询问是否继续停车。
|
|
|
:return:None
|
|
|
"""
|
|
|
if len(self.carlist) >= self.max_car:
|
|
|
print("对不起，当前车库已满！")
|
|
|
return None
|
|
|
while True:
|
|
|
print('当前空闲车位如下：')
|
|
|
self.inquire_empty()
|
|
|
print('')
|
|
|
|
|
|
# 变量初始化
|
|
|
flag_right_input = False
|
|
|
flag_error = 0 # 清除错误flag
|
|
|
temp_parknum = None
|
|
|
temp_carnum = None
|
|
|
temp_color = None
|
|
|
temp_model = None
|
|
|
try:
|
|
|
temp_parknum = int(input('请输入所停车位:'))
|
|
|
temp_carnum = input('请输入车牌号:')
|
|
|
temp_model = int(input('请输入车型序号（1小汽车, 2小卡, 3中卡, 4大卡）:'))
|
|
|
temp_color = int(input('请输入车辆颜色序号（1白色, 2黑色, 3灰色, 4蓝色, 5红色, 6黄色）:'))
|
|
|
|
|
|
# 以下为输入信息是否有效的判断
|
|
|
if 0 <= temp_parknum <= self.max_car: # 判断车位是否在合理范围内
|
|
|
if 1 <= temp_model <= 4:
|
|
|
temp_model = Model(temp_model) # 转化为Model枚举类型
|
|
|
if 1 <= temp_color <= 6:
|
|
|
temp_color = Color(temp_color) # 转化Color枚举类型
|
|
|
if self.check_car_num(temp_carnum): # 判断车牌号是否有效
|
|
|
for parkedCar in self.carlist:
|
|
|
if parkedCar.parknum == temp_parknum: # 判断车位是否已被占用
|
|
|
flag_error = 1
|
|
|
break
|
|
|
elif parkedCar.carnum == temp_carnum: # 判断车牌是否已重复
|
|
|
flag_error = 2
|
|
|
break
|
|
|
else:
|
|
|
flag_right_input = True # 以上判断均通过
|
|
|
"""
|
|
|
else:
|
|
|
error_input = 3
|
|
|
else:
|
|
|
error_input = 4
|
|
|
else:
|
|
|
error_input = 5
|
|
|
"""
|
|
|
|
|
|
except Exception as e:
|
|
|
# logging.exception(e) # for debug
|
|
|
flag_right_input = False # 输入类型错误
|
|
|
# error_input = 6
|
|
|
finally:
|
|
|
# if error_input != 0:
|
|
|
if not flag_right_input:
|
|
|
if flag_error == 1:
|
|
|
print('当前车位已被占用！')
|
|
|
elif flag_error == 2:
|
|
|
print('车牌号重复！')
|
|
|
else:
|
|
|
print("输入错误！")
|
|
|
retry_choice = input('按<F>键返回主菜单，按其他任意键重试')
|
|
|
if retry_choice == 'F' or retry_choice == 'f':
|
|
|
return None # 返回主菜单
|
|
|
else:
|
|
|
os.system('cls') # 清屏
|
|
|
else:
|
|
|
# 通过各项判断，下面新增车辆
|
|
|
temp_car = Car(temp_parknum, temp_carnum, temp_model, temp_color)
|
|
|
temp_car["intime"] = time.ctime() # 调用当前时间
|
|
|
self.carlist.append(temp_car)
|
|
|
print("%s 停车入库成功！" % temp_carnum)
|
|
|
|
|
|
if not isAdmin: # 用户返回主菜单
|
|
|
print('按任意键返回主菜单......')
|
|
|
os.system("pause")
|
|
|
return None
|
|
|
if isAdmin: # 管理员询问是否继续新增车辆
|
|
|
while True:
|
|
|
continue_choice = input("是（Y）否（N）继续新增车辆？")
|
|
|
if continue_choice == 'Y' or continue_choice == 'y':
|
|
|
os.system('cls') # 清屏
|
|
|
break
|
|
|
elif continue_choice == 'N' or continue_choice == 'n':
|
|
|
return None
|
|
|
else:
|
|
|
print('输入错误，请重试！')
|
|
|
|
|
|
"""
|
|
|
def add_car(self, car):
|
|
|
# 从car新增车辆到carlist中
|
|
|
# if self.check_car(car) is not None:
|
|
|
# print("车辆已存在，请重试！")
|
|
|
# else:
|
|
|
car["intime"] = time.ctime() # 调用当前时间
|
|
|
self.carlist.append(car)
|
|
|
print("%s 停车入库成功！" % car.carnum)
|
|
|
"""
|
|
|
|
|
|
def display(self, carlist=None):
|
|
|
"""
|
|
|
显示车辆信息
|
|
|
:param:{list}carlist, 默认为self.carlist
|
|
|
:return:None
|
|
|
"""
|
|
|
if carlist is None:
|
|
|
carlist = self.carlist
|
|
|
if len(self.carlist) == 0:
|
|
|
print("停车场内暂无车辆。")
|
|
|
return None
|
|
|
|
|
|
print("车位 车牌号 车型 颜色 入场时间 ")
|
|
|
i = 0
|
|
|
for parkedCar in carlist:
|
|
|
print(parkedCar)
|
|
|
i += 1
|
|
|
if len(carlist) > 10 and i % 10 == 0: # 超过10辆车则每10辆一页
|
|
|
print("按任意键查看下一页......")
|
|
|
os.system("pause")
|
|
|
print("车位 车牌号 车型 颜色 入场时间 ")
|
|
|
# os.system('pause')
|
|
|
|
|
|
def check_car(self, keyword, sort=1):
|
|
|
"""
|
|
|
用于查找车辆。默认sort=1,按照车位查找，若存在则返回满足条件的车辆信息。
|
|
|
:param:keyword关键词，sort查找方式
|
|
|
:return:{class Car}/{list}
|
|
|
"""
|
|
|
if sort == 1: # 按车位
|
|
|
for parkedCar in self.carlist:
|
|
|
if parkedCar.parknum == keyword:
|
|
|
return parkedCar
|
|
|
elif sort == 2: # 按车牌
|
|
|
for parkedCar in self.carlist:
|
|
|
if parkedCar.carnum == keyword:
|
|
|
return parkedCar
|
|
|
elif 3 <= sort <= 4:
|
|
|
if sort == 3: # 按车型
|
|
|
key = "model"
|
|
|
elif sort == 4: # 按颜色
|
|
|
key = "color"
|
|
|
else:
|
|
|
return None
|
|
|
t_list = []
|
|
|
for parkedCar in self.carlist:
|
|
|
if parkedCar[key] == keyword:
|
|
|
t_list.append(parkedCar)
|
|
|
return t_list
|
|
|
elif sort == 5: # TODO 按入场时间
|
|
|
# (developing)
|
|
|
pass
|
|
|
else:
|
|
|
return None
|
|
|
|
|
|
def inquire(self, choice=-1):
|
|
|
"""
|
|
|
查询信息
|
|
|
复用于取车的查找，修改信息的查找。
|
|
|
Developing:按照入场时间查询（多少时间内）。
|
|
|
:param choice:查询方式选择，默认为-1.
|
|
|
:return:None/{class Car}parkedCar
|
|
|
"""
|
|
|
# choice = -1 # DEBUG
|
|
|
while choice < 0 or choice > 6:
|
|
|
print("""
|
|
|
1)空闲车位查询
|
|
|
--车辆信息查询--
|
|
|
2)按车位查询车辆
|
|
|
3)按车牌查询车辆
|
|
|
4)按车型查询车辆
|
|
|
5)按颜色查询车辆
|
|
|
6)按入场时间查询(developing)
|
|
|
---------------------------
|
|
|
0)返回主菜单
|
|
|
""")
|
|
|
try:
|
|
|
choice = int(input("请输入操作码："))
|
|
|
except Exception:
|
|
|
print("错误操作码，请重试！")
|
|
|
continue
|
|
|
if choice < 0 or choice > 6:
|
|
|
print("错误操作码，请重试！")
|
|
|
|
|
|
if choice == 0:
|
|
|
return None # 返回主菜单
|
|
|
# 空闲车位查询：
|
|
|
elif choice == 1:
|
|
|
if len(self.carlist) >= self.max_car:
|
|
|
print("对不起，当前车库已满！")
|
|
|
return None
|
|
|
else:
|
|
|
print('空闲车位如下：')
|
|
|
self.inquire_empty()
|
|
|
print('') # 多空一行
|
|
|
else:
|
|
|
if len(self.carlist) == 0:
|
|
|
print("停车场内暂无车辆。")
|
|
|
return None
|
|
|
try:
|
|
|
# 按车位查询车辆：
|
|
|
if choice == 2:
|
|
|
keyword = int(input("请输入车位号："))
|
|
|
t_result = self.check_car(keyword, sort=1)
|
|
|
if t_result:
|
|
|
print("车位 车牌号 车型 颜色 入场时间 ")
|
|
|
print(t_result)
|
|
|
return t_result
|
|
|
else:
|
|
|
print("停车场中无此车辆。")
|
|
|
|
|
|
# 按车牌查询车辆：
|
|
|
elif choice == 3:
|
|
|
keyword = input("请输入车牌号：")
|
|
|
t_result = self.check_car(keyword, sort=2)
|
|
|
if t_result:
|
|
|
print("车位 车牌号 车型 颜色 入场时间 ")
|
|
|
print(t_result)
|
|
|
return t_result
|
|
|
else:
|
|
|
print("停车场中无此车辆。")
|
|
|
|
|
|
# 按车型查询车辆：
|
|
|
elif choice == 4:
|
|
|
keyword = Model(int(input("请输入车辆的车型（1小汽车, 2小卡, 3中卡, 4大卡）：")))
|
|
|
t_result = self.check_car(keyword, sort=3)
|
|
|
if t_result:
|
|
|
self.display(t_result)
|
|
|
else:
|
|
|
print("停车场中无此车型车辆。")
|
|
|
|
|
|
# 按颜色查询车辆：
|
|
|
elif choice == 5:
|
|
|
keyword = Color(int(input("请输入车辆的颜色（1白色, 2黑色, 3灰色, 4蓝色, 5红色, 6黄色）：")))
|
|
|
t_result = self.check_car(keyword, sort=4)
|
|
|
if t_result:
|
|
|
self.display(t_result)
|
|
|
else:
|
|
|
print("停车场中无此颜色车辆。")
|
|
|
|
|
|
# 按入场时间查询：
|
|
|
elif choice == 6:
|
|
|
# TODO (developing) 按入场时间查询
|
|
|
pass
|
|
|
"""
|
|
|
keyword = input("请输入入场时间：")
|
|
|
t_result = self.check_car(keyword, sort=5)
|
|
|
if t_result:
|
|
|
print(t_result)
|
|
|
else:
|
|
|
print("停车场中无此车辆。")
|
|
|
"""
|
|
|
else:
|
|
|
pass
|
|
|
except Exception as e: # 输入异常处理
|
|
|
# logging.exception(e) # DEBUG
|
|
|
print("输入错误，按任意键返回......")
|
|
|
return None
|
|
|
|
|
|
def pickup(self):
|
|
|
"""
|
|
|
取车
|
|
|
:return:None
|
|
|
"""
|
|
|
if len(self.carlist) == 0:
|
|
|
print("停车场内暂无车辆。")
|
|
|
return None
|
|
|
# 变量初始化
|
|
|
exit_car = None
|
|
|
inquire_choice = -1
|
|
|
while inquire_choice < 0 or inquire_choice > 2:
|
|
|
try:
|
|
|
print("-----------取车-----------")
|
|
|
inquire_choice = int(input("1)按车位查找\n2)按车牌号查找\n0)返回主菜单\n\n请输入操作码:"))
|
|
|
except:
|
|
|
continue
|
|
|
if inquire_choice == 0: # 返回主菜单
|
|
|
return None
|
|
|
elif inquire_choice == 1: # 按车位
|
|
|
exit_car = self.inquire(choice=2)
|
|
|
elif inquire_choice == 2: # 按车牌号
|
|
|
exit_car = self.inquire(choice=3)
|
|
|
if exit_car is None:
|
|
|
return None
|
|
|
exit_choice = input("确定请输入“Y”，其他任意输入返回主菜单：")
|
|
|
if exit_choice == 'Y' or exit_choice == 'y':
|
|
|
exit_time = time.ctime()
|
|
|
park_time = time.mktime(time.strptime(exit_time)) - time.mktime(time.strptime(exit_car.intime))
|
|
|
m, s = divmod(park_time, 60)
|
|
|
h, m = divmod(m, 60)
|
|
|
str_time = "%02d:%02d:%02d" % (h, m, s) # 得到时分秒字符串
|
|
|
global unit_price
|
|
|
price = h * unit_price
|
|
|
print("车牌号：%s \n停车时长：%s\n请交费%3d元 " % (exit_car.carnum, str_time, price))
|
|
|
self.carlist.remove(exit_car)
|
|
|
os.system("pause") # 预留之后完善计费功能
|
|
|
print("结算成功，欢迎您再次光临！")
|
|
|
else:
|
|
|
return None
|
|
|
|
|
|
def edit(self):
|
|
|
"""
|
|
|
编辑车辆信息
|
|
|
:return:None
|
|
|
"""
|
|
|
if len(self.carlist) == 0:
|
|
|
print("停车场内暂无车辆。")
|
|
|
return None
|
|
|
|
|
|
# 变量初始化
|
|
|
edit_car = None
|
|
|
inquire_choice = -1
|
|
|
while inquire_choice < 0 or inquire_choice > 2:
|
|
|
try:
|
|
|
print("-----车辆信息编辑-----")
|
|
|
inquire_choice = int(input("1)按车位查找\n2)按车牌号查找\n0)返回主菜单\n\n请输入操作码:"))
|
|
|
except:
|
|
|
continue
|
|
|
if inquire_choice == 0: # 返回主菜单
|
|
|
return None
|
|
|
elif inquire_choice == 1: # 按车位
|
|
|
edit_car = self.inquire(choice=2)
|
|
|
elif inquire_choice == 2: # 按车牌号
|
|
|
edit_car = self.inquire(choice=3)
|
|
|
if edit_car is None:
|
|
|
return None
|
|
|
edit_choice = input("确定请输入“Y”，其他任意输入返回主菜单：")
|
|
|
if edit_choice == 'Y' or edit_choice == 'y':
|
|
|
index = self.carlist.index(edit_car) # 获取索引
|
|
|
while True:
|
|
|
try:
|
|
|
flag_right = False # right flag
|
|
|
change_choice = input(
|
|
|
"\n1)车位号\n2)车牌号\n3)车型\n4)车颜色\n5)入场时间\n0)不修改返回主菜单\n\n请输入您要修改的信息序号：")
|
|
|
if change_choice == '0':
|
|
|
return None
|
|
|
elif change_choice == '1':
|
|
|
new_info = int(input("请输入新的车位号："))
|
|
|
if 0 <= new_info <= self.max_car: # 在范围内
|
|
|
for parkedCar in self.carlist:
|
|
|
if parkedCar.parknum == new_info: # 判断车位是否已被占用
|
|
|
print(" %d号车位已被占用！" % new_info)
|
|
|
break
|
|
|
else:
|
|
|
self.carlist[index]["parknum"] = new_info
|
|
|
flag_right = True
|
|
|
print("车位号修改成功！")
|
|
|
# break
|
|
|
elif change_choice == '2':
|
|
|
new_info = input("请输入新的车牌号：")
|
|
|
if self.check_car_num(new_info): # 判断车牌号是否有效
|
|
|
for parkedCar in self.carlist:
|
|
|
if parkedCar.carnum == new_info: # 判断车牌是否已重复
|
|
|
print("车牌号重复！")
|
|
|
break
|
|
|
else:
|
|
|
self.carlist[index]["carnum"] = new_info
|
|
|
flag_right = True
|
|
|
print("车牌号修改成功！")
|
|
|
# break
|
|
|
elif change_choice == '3':
|
|
|
new_info = int(input("请输入新的车型（1小汽车, 2小卡, 3中卡, 4大卡）:"))
|
|
|
if 1 <= new_info <= 4:
|
|
|
new_info = Model(new_info) # 转化为Model枚举类型
|
|
|
self.carlist[index]["model"] = new_info
|
|
|
flag_right = True
|
|
|
print("车型修改成功！")
|
|
|
elif change_choice == '4':
|
|
|
new_info = int(input("请输入新的车颜色（1白色, 2黑色, 3灰色, 4蓝色, 5红色, 6黄色）："))
|
|
|
if 1 <= new_info <= 6:
|
|
|
new_info = Color(new_info) # 转化Color枚举类型
|
|
|
self.carlist[index]["color"] = new_info
|
|
|
flag_right = True
|
|
|
print("颜色修改成功！")
|
|
|
elif change_choice == '5':
|
|
|
new_info = input("请输入新的入场时间：(参考格式：2019.3.4 12:00:00)")
|
|
|
print("DEVELOPING...")
|
|
|
# TODO 修改入场时间
|
|
|
pass
|
|
|
break
|
|
|
else:
|
|
|
pass
|
|
|
except:
|
|
|
print("输入错误，请重试！")
|
|
|
continue
|
|
|
|
|
|
if flag_right:
|
|
|
break
|
|
|
else:
|
|
|
print("请重试！")
|
|
|
continue
|
|
|
else:
|
|
|
return None
|
|
|
|
|
|
def statistics(self):
|
|
|
"""
|
|
|
统计车辆信息
|
|
|
:return:None
|
|
|
"""
|
|
|
print("""
|
|
|
本停车场总车位数：{total:3d}
|
|
|
当前已驶入车辆数：{exist:3d} 空闲车位数：{empty:3d}
|
|
|
-----------------------------------------------
|
|
|
""".format(total=self.max_car, exist=len(self.carlist), empty=self.max_car - len(self.carlist)),
|
|
|
end='')
|
|
|
|
|
|
# 按车型、颜色统计车辆信息
|
|
|
cnt_model = [0, 0, 0, 0]
|
|
|
cnt_color = [0, 0, 0, 0, 0, 0]
|
|
|
if len(self.carlist) != 0:
|
|
|
for parkedCar in self.carlist:
|
|
|
cnt_model[parkedCar.model.value - 1] += 1
|
|
|
cnt_color[parkedCar.color.value - 1] += 1
|
|
|
print("""
|
|
|
* 按车型统计：
|
|
|
小汽车 {:3d}, 小卡 {:3d}, 中卡 {:3d}, 大卡 {:3d}
|
|
|
|
|
|
* 按颜色统计：
|
|
|
白色 {:3d}, 黑色 {:3d}, 灰色 {:3d}
|
|
|
蓝色 {:3d}, 红色 {:3d}, 黄色 {:3d}
|
|
|
""".format(cnt_model[0], cnt_model[1], cnt_model[2], cnt_model[3],
|
|
|
cnt_color[0], cnt_color[1], cnt_color[2],
|
|
|
cnt_color[3], cnt_color[4], cnt_color[5], ), end='')
|
|
|
|
|
|
if len(self.carlist) == 0:
|
|
|
print("""
|
|
|
-----------------------------------------------
|
|
|
""")
|
|
|
else:
|
|
|
# TODO 按入场时间统计
|
|
|
print("""
|
|
|
* 按入场时间统计：
|
|
|
（正在开发中 Developing...）
|
|
|
|
|
|
-----------------------------------------------""")
|
|
|
|
|
|
# TODO 车辆信息按顺序输出
|
|
|
|
|
|
# choice = input("""
|
|
|
# 1)按车位排序 2)按入场时间排序
|
|
|
# 请输入操作码（其他任意输入返回主菜单）:""")
|
|
|
# if choice == '1':
|
|
|
# pass
|
|
|
# temp_carlist = []
|
|
|
# for parkedCar in self.carlist:
|
|
|
# pass
|
|
|
#
|
|
|
# self.display(temp_carlist)
|
|
|
# elif choice == '2':
|
|
|
# pass
|
|
|
# else:
|
|
|
# return None
|
|
|
# |
