Python宠物商店管理系统 课程设计

import time import pymysql.cursors class Pet: def __init__(self, pet_id, name, species, price, stock): self.pet_id = pet_id self.name = name self.species = species self.price = price self.stock = stock class Order: def __init__(self, order_id, pet, quantity, purchase_time): self.order_id = order_id self.pet = pet self.quantity = quantity self.purchase_time = purchase_time class Return: def __init__(self, return_id, pet_id, reason, quantity, return_time): self.return_id = return_id self.pet_id = pet_id self.reason = reason self.quantity = quantity self.return_time = return_time class PetStore: def __init__(self): # 数据库连接 self.connection = pymysql.connect( host="localhost", port=3306, user="root", passwd="123456", db="pet_store", charset="utf8", cursorclass=pymysql.cursors.DictCursor ) self.logged_in_user = None def register(self): """ 注册新用户 """ username = input("请输入用户名：") password = input("请输入密码：") try: with self.connection.cursor() as cursor: # 执行注册操作 sql = "INSERT INTO users (username, password) VALUES (%s, %s)" cursor.execute(sql, (username, password)) self.connection.commit() print("注册成功！") except Exception as e: print(f"注册失败：{e}") def login(self): """ 用户登录 """ username = input("请输入用户名：") password = input("请输入密码：") try: with self.connection.cursor() as cursor: # 执行登录查询 sql = "SELECT user_id FROM users WHERE username=%s AND password=%s" cursor.execute(sql, (username, password)) result = cursor.fetchone() if result: self.logged_in_user = result['user_id'] print("登录成功！") else: print("用户名或密码错误！") except Exception as e: print(f"登录失败：{e}") def show_pet_list(self): """ 显示宠物列表 """ try: with self.connection.cursor() as cursor: # 查询宠物信息 sql = "SELECT * FROM pets" cursor.execute(sql) pets = cursor.fetchall() print("宠物ID\t姓名\t种类\t价格\t库存") for pet in pets: print(f"{pet['pet_id']}\t{pet['name']}\t{pet['species']}\t{pet['price']}\t{pet['stock']}") except Exception as e: print(f"查询宠物列表失败：{e}") def buy_pet(self): """ 购买宠物 """ if not self.logged_in_user: print("请先登录！") return self.show_pet_list() pet_id = int(input("输入要购买的宠物ID：")) try: with self.connection.cursor() as cursor: # 查询宠物信息 sql = "SELECT * FROM pets WHERE pet_id=%s" cursor.execute(sql, (pet_id,)) pet_info = cursor.fetchone() if pet_info: quantity = int(input("输入购买数量：")) if 0 < quantity <= pet_info['stock']: try: # 减少库存 sql_update_stock = "UPDATE pets SET stock=stock-%s WHERE pet_id=%s" cursor.execute(sql_update_stock, (quantity, pet_id)) # 生成订单 purchase_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) sql_insert_order = "INSERT INTO orders (pet_id, user_id, quantity, purchase_time) VALUES (%s, %s, %s, %s)" cursor.execute(sql_insert_order, (pet_id, self.logged_in_user, quantity, purchase_time)) self.connection.commit() print(f"您成功购买了{quantity}只{pet_info['species']}！") except Exception as e: print(f"购买失败：{e}") else: print("无效的数量或库存不足。") else: print("无效的宠物ID。") except Exception as e: print(f"购买宠物失败：{e}") def show_order_list(self): """ 显示订单列表 """ if not self.logged_in_user: print("请先登录！") return try: with self.connection.cursor() as cursor: # 查询订单信息 sql = "SELECT orders.order_id, pets.species, pets.name, orders.quantity, orders.purchase_time FROM orders JOIN pets ON orders.pet_id = pets.pet_id WHERE orders.user_id=%s" cursor.execute(sql, (self.logged_in_user,)) orders = cursor.fetchall() print("订单ID\t宠物\t数量\t购买时间") for order in orders: print( f"{order['order_id']}\t{order['species']}（{order['name']}）\t{order['quantity']}\t{order['purchase_time']}") except Exception as e: print(f"查询订单列表失败：{e}") def show_user_list(self): """ 显示用户列表 """ try: with self.connection.cursor() as cursor: # 查询用户信息 sql = "SELECT * FROM users" cursor.execute(sql) users = cursor.fetchall() print("用户ID\t用户名") for user in users: print(f"{user['user_id']}\t{user['username']}") except Exception as e: print(f"查询用户列表失败：{e}") def close_connection(self): """ 关闭数据库连接 """ self.connection.close() print("数据库连接已关闭。") # 主程序 if __name__ == "__main__": pet_store = PetStore() while True: print("1. 注册") print("2. 登录") print("3. 显示宠物列表") print("4. 购买宠物") print("5. 显示订单列表") print("6. 查询用户表") print("7. 退出") choice = input("请选择操作：") if choice == "1": pet_store.register() elif choice == "2": pet_store.login() elif choice == "3": pet_store.show_pet_list() elif choice == "4": pet_store.buy_pet() elif choice == "5": pet_store.show_order_list() elif choice == "6": pet_store.show_user_list() # Assuming you have a method for querying the user table elif choice == "7": pet_store.close_connection() print("退出宠物商店。再见！") break else: print("无效的选择。请重试。")