python 类的设计 （课程作业 / 水一篇）

课程作业， 嘘， 水一篇

python 类的设计（All Code）

ADD_PTES = "1.添加宠物"
FIND_PET = '2.查找宠物'
DEL_PET = '3.出售宠物'
LIST_PETS = '4.列出宠物'
EXIT_PETS = '5.退出宠物'

# 全局
PETS = []
PETS_NAME = []

# 添加
def add_pet():
    ID = input("请输入宠物编号：")
    b = 1
    while b:
        b = 0
        name = input("请输入宠物名称：")
        for s in PETS_NAME:
            if s == name:
                b = 1
                break
        if b:
            print("抱歉，已存在该名称的宠物啦~~~")
    PETS_NAME.append(name)
    color = input("请输入宠物的颜色：")
    category = input("请输入宠物种类:")
    price = input("请输入宠物价格：")
    pet = {'id': ID, 'name': name, 'color': color, 'category': category, 'price': price}
    PETS.append(pet)
    print("宠物添加成功！")

# 出售
def sale_pet():
    ID = input("请输入宠物编号：")
    for pet in PETS:
        if pet['id'] == ID:
            price = pet['price']
            name = pet['name']
            PETS.remove(pet)
            PETS_NAME.remove(name)
            print("价格：" + price +  "  出售成功！")
            break


# 查找
def search_pet():
    name = input("请输入宠物名称：")
    for pet in PETS:
        if pet['name'] == name:
            text = "编号：{},名称:{},颜色:{},种类:{},价格:{}".format(
                pet['id'],
                pet['name'],
                pet['color'],
                pet["category"],
                pet['price']
            )
            print(text)


# 列表
def list_pet():
    for pet in PETS:
        text = "编号：{},名称:{},颜色:{},种类:{},价格:{}".format(
            pet['id'],
            pet['name'],
            pet['color'],
            pet["category"],
            pet['price']
        )
        print(text)


def main():
    print('=' * 15)
    print(ADD_PTES)
    print(FIND_PET)
    print(DEL_PET)
    print(LIST_PETS)
    print(EXIT_PETS)
    print('=' * 15)

    while True:
        option = input("请输入选项：")
        if option == '1':
            add_pet()
        elif option == '2':
            search_pet()
        elif option == '3':
            sale_pet()
        elif option == '4':
            list_pet()
        elif option == '5':
            break
        else:
            print("请输入正确的选项")


main()

效果图：