MySQL 实现点餐系统的预定功能

CREATE TABLE users (
    id INT PRIMARY KEY AUTO_INCREMENT,
    username VARCHAR(50) NOT NULL,
    password VARCHAR(50) NOT NULL
);

CREATE TABLE restaurants (
    id INT PRIMARY KEY AUTO_INCREMENT,
    name VARCHAR(100) NOT NULL,
    address VARCHAR(200) NOT NULL
);

CREATE TABLE bookings (
    id INT PRIMARY KEY AUTO_INCREMENT,
    user_id INT NOT NULL,
    restaurant_id INT NOT NULL,
    booking_time DATETIME NOT NULL,
    party_size INT NOT NULL,
    FOREIGN KEY (user_id) REFERENCES users(id),
    FOREIGN KEY (restaurant_id) REFERENCES restaurants(id)
);

import mysql.connector

# 建立数据库连接
db = mysql.connector.connect(
    host="localhost",
    user="root",
    password="password",
    database="restaurant_booking"
)

# 创建游标对象
cursor = db.cursor()

# 获取用户输入的预定信息
username = input("请输入用户名：")
password = input("请输入密码：")
restaurant_id = input("请输入餐厅ID：")
booking_time = input("请输入预定时间（YYYY-MM-DD HH:MM:SS）：")
party_size = input("请输入预定人数：")

# 查询用户ID
cursor.execute(f"SELECT id FROM users WHERE username = '{username}' AND password = '{password}'")
user_id = cursor.fetchone()
if not user_id:
    print("用户名或密码错误")
    exit()

# 添加预定信息到数据库
cursor.execute(f"INSERT INTO bookings (user_id, restaurant_id, booking_time, party_size) VALUES ({user_id[0]}, {restaurant_id}, '{booking_time}', {party_size})")
db.commit()

print("预定成功")

# 关闭数据库连接
db.close()