python实战项目火车票

开始
↓
乘客登录/注册
↓
查询列车时刻表
↓
选择列车
↓
预订火车票
↓
查看预订记录
↓
退票（可选）
↓
结束

# 乘客信息
passengers = {}

# 列车时刻表
schedule = [
    {"origin": "北京", "destination": "上海", "departure": "08:00", "arrival": "12:00", "price": 500},
    {"origin": "上海", "destination": "广州", "departure": "10:00", "arrival": "14:00", "price": 600},
]

# 预订记录
bookings = []

# 乘客登录
def login(username):
    if username in passengers:
        return passengers[username]
    else:
        return None

# 查询时刻表
def get_schedule(origin, destination):
    for train in schedule:
        if train["origin"] == origin and train["destination"] == destination:
            return train
    return None

# 预订火车票
def book_ticket(passenger, train):
    booking = {
        "passenger": passenger["name"],
        "id_number": passenger["id_number"],
        "train": train["name"],
        "departure": train["origin"],
        "arrival": train["destination"],
        "time": datetime.now(),
    }
    bookings.append(booking)

# 查看预订记录
def get_bookings(passenger):
    return [booking for booking in bookings if booking["passenger"] == passenger["name"]]

# 退票
def cancel_booking(booking):
    bookings.remove(booking)