main.py - master - lerka59rus/PYAPI_Homework

main.py
102 строки3 KB
Валерия Житлухина
Домашнее задание
23 июн 2025, 13:03
23 июн 2025, 13:0334f7e9b
def read_cook_book(file_path):
    cook_book = {}
    with open(file_path, 'r', encoding='utf-8') as f:
        lines = [line.strip() for line in f.readlines()]

    i = 0
    while i < len(lines):
        if not lines[i]:
            i += 1
            continue

        dish_name = lines[i]
        i += 1

        ingredient_count = int(lines[i])
        i += 1

        ingredients = []
        for _ in range(ingredient_count):
            if i >= len(lines):
                break

            parts = lines[i].split('|')
            if len(parts) != 3:
                i += 1
                continue

            ingredient = {
                'ingredient_name': parts[0].strip(),
                'quantity': int(parts[1].strip()),
                'measure': parts[2].strip()
            }
            ingredients.append(ingredient)
            i += 1

        cook_book[dish_name] = ingredients

    return cook_book

def get_shop_list_by_dishes(dishes, person_count, cook_book):
    shop_list = {}
    for dish in dishes:
        if dish not in cook_book:
            continue

        for ingredient in cook_book[dish]:
            name = ingredient['ingredient_name']
            new_quantity = ingredient['quantity'] * person_count

            if name in shop_list:
                shop_list[name]['quantity'] += new_quantity
            else:
                shop_list[name] = {
                    'measure': ingredient['measure'],
                    'quantity': new_quantity
                }

    return shop_list

def merge_files(files, output_file):
    # Получаем список файлов с количеством строк
    file_info = []
    for file_name in files:
        with open(file_name, 'r', encoding='utf-8') as f:
            lines = f.readlines()
            file_info.append((file_name, len(lines)))

    file_info.sort(key=lambda x: x[1])

    with open(output_file, 'w', encoding='utf-8') as result_file:
        for file_name, line_count in file_info:
            with open(file_name, 'r', encoding='utf-8') as f:
                content = f.read()

            result_file.write(f"{file_name}\n")
            result_file.write(f"{line_count}\n")
            result_file.write(content)
            if not content.endswith('\n'):
                result_file.write('\n')
            result_file.write('\n')

if __name__ == "__main__":
    # Задача 1: Чтение cook_book
    cook_book = read_cook_book('recipes.txt')
    print("cook_book:")
    for dish, ingredients in cook_book.items():
        print(f"{dish}:")
        for ing in ingredients:
            print(f"  {ing}")
    print("\n")

    # Задача 2: Генерация списка покупок
    shop_list = get_shop_list_by_dishes(['Запеченный картофель', 'Омлет'], 2, cook_book)
    print("shop_list:")
    for ingredient, details in shop_list.items():
        print(f"{ingredient}: {details}")
    print("\n")

    # Задача 3: Объединение файлов
    files_to_merge = ['files/1.txt', 'files/2.txt', 'files/3.txt']
    merge_files(files_to_merge, 'result.txt')
    print("Файлы объединены в result.txt")