Python-сообщество

  
goods = {'торт': ['мука, яйцо, сливки, шоколад', 150, 2000],
         'пирожное': ['тесто слоеное, молоко, сливочное масло', 120, 1500],
         'маффин': ['порошок яичный, порошок какао, ванилин', 200, 1000]}
my_cart = {}
 
def goods_demo(demo_type=0, product=None):
    p = 'Товар:'
    s = 'Состав: '
    c = 'Цена за 100 гр.: '
    k = 'Количество (г): '
    if demo_type == 0:
        for i in goods:
            print(i)
        return
    elif demo_type == 1:
        return f'{s}{goods[product][0]}'
    elif demo_type == 2:
        return f'{c}{goods[product][1]}'
    elif demo_type == 3:
        return f'{k}{goods[product][2]}'
    elif demo_type == 4:
        return f'{s}{goods[product][0]}\n' \
               f'{c}{goods[product][1]}\n' \
               f'{k}{goods[product][2]}'
    elif demo_type == 6:
        print('\nОстаток кондитерской:')
        for i in goods:
            print(f'{p} "{i}" {k}{goods[i][2]}')
        return
 
def ask_action_type(product):
    while True:
        try:
            action = int(input('Что вы хотели бы узнать о товаре?\n '
                               '(1)-состав, (2)-цена, (3)-количество, '
                               '(4)-вся инф., (5)-покупка\n>>> '))
            if action == 5:
                return action
            elif 1 <= action <= 4:
                return goods_demo(action, product)
        except:
            print('Введите число, сопоставленное с действием.')
            continue
 
def ask_continue():
    while True:
        s = input('Продолжить покупки? д|н\n>>> ').lower()
        if s == 'д':
            return True
        elif s == 'н':
            return False
 
def add_to_cart(product, price, weight):
    if product in my_cart.keys():
        my_cart[product][0] += price
        my_cart[product][1] += weight
    else:
        my_cart.update({product: [price, weight]})
 
def buy(product):
    while True:
        try:
            weight = abs(int(input('Сколько грамм купить?\n>>> ')))
        except:
            print('Нужно ввести целое число.')
            continue
        if goods[product][2] - weight >= 0:
            if weight > 0:
                goods[product][2] -= weight
                price = goods[product][1] * (weight * 0.01)
                return add_to_cart(product, price, weight)
            else:
                print('Нужно число больше нуля.')
                continue
        else:
            print(f'На складе нет такого количества! '
                  f'В наличии: {goods[product][2]}')
            continue
 
def show_total():
    p = 'Товар:'
    o = 'Общая стоимость товаров: '
    w = 'Вес в граммах: '
    summ = 0
    print('В вашей корзине: ')
    for i in my_cart:
        summ += my_cart[i][0]
        print(f'{p} "{i}" {w}{my_cart[i][1]}')
    print(f'{o}{summ} у.е.')
    if goods != {}:
        return goods_demo(6)
 
def check_goods():
    global goods
    new = goods.copy()
    for i in goods:
        if goods[i][2] == 0:
            new.pop(i)
    goods = new.copy()
    if goods == {}:
        print('Все товары закончились!')
        return False
    else:
        return True
 
def ask_product():
    while True:
        product = input('Какой из товаров вас интересует?\n>>> ').lower()
        if product in goods.keys():
            return product
        else:
            print('Такого товара нет, введите товар из списка выше.')
            continue
 
def main():
    print('Добро пожаловать в кондитерскую!')
    run = True
    while run:
        if check_goods():
            print('В продаже имеется:')
            goods_demo()
            user_choice = ask_product()
            user_interest = ask_action_type(user_choice)
            if user_interest == 5:
                buy(user_choice)
            else:
                print(f'Товар: "{user_choice}"\n{user_interest}')
            run = ask_continue()
        else:
            break
    show_total()
    print('До свидания.')
 
if __name__ == '__main__':
    main()