接单第一天

ai龙要努力

import pymysql
import random

# number = int(input("Enter a number: "))
connection = ''
try:
    connection = pymysql.connect(
        host='localhost',
        user='root',
        password='123456',
        db='test',
        charset='utf8mb4',

    )

    connection.cursor().execute("""
        CREATE TABLE IF NOT EXISTS menu (
            id INT AUTO_INCREMENT PRIMARY KEY,
            name VARCHAR(255) NOT NULL,
            salary FLOAT NOT NULL
        )
        """)
except Exception as e:
    print("database error :", e)
while True:
    username = input("请输入用户名: ")
    password = input("请输入用户名密码: ")
    if username == "admin" and password == "123456":
        print(f"{username}用户登录成功")
        break
    else:
        print("登录失败,请在下面从新输入")
while True:
    print("===========菜单功能==============")
    print("1 新增菜单信息\n"
          "2 查询菜单信息\n"
          "3 删除菜单信息\n"
          "4 修改菜单信息\n"
          "5 查询所以菜单\n"
          "请输入序号：")
    print("============菜单功能=============")
    number = int(input("Enter a number: "))
    if number == 1:
        name = input("请输入菜单名称: ")
        salary = random.randint(1, 100)
        with connection.cursor() as cursor:
            cursor.execute("INSERT INTO menu (name, salary) VALUES (%s, %s)", (name, salary))
            connection.commit()

    elif number == 2:
        number = int(input("请输入修改菜单名称的id: "))
        with connection.cursor() as cursor:
            cursor.execute(f"SELECT id,name,salary FROM menu where id ={number}")
            datas = list(cursor.fetchall())
            for item in datas:
                print(item[0], item[1], item[2])

    elif number == 3:
        number = int(input("请输入修改菜单名称的id: "))
        with connection.cursor() as cursor:
            cursor.execute("DELETE FROM menu WHERE id = %s", (number,))
        connection.commit()


    elif number == 4:
        new_id = int(input("请输入菜单id: "))
        new_name = input("请输入菜单名称: ")
        with connection.cursor() as cursor:
            cursor.execute("UPDATE menu SET name = %s WHERE id = %s", (new_name, new_id))
        connection.commit()

    elif number == 5:
        with connection.cursor() as cursor:
            cursor.execute("SELECT id,name FROM menu")
            datas = list(cursor.fetchall())
            for item in datas:
                print(item[0], item[1])
    else:
        print("退出菜单程序")
        exit()