food.py

food.py

#!/usr/bin/python3
from collections import defaultdict 
import sys

individual_indicator = "east"

by_name = defaultdict(list)
by_item = defaultdict(list)
individual = defaultdict(list)

current_name = None
count = 0

for line in sys.stdin:
    l = line.rstrip("\n")
    l = l.strip()

    if not l.startswith("-"):
        current_name = l
    else:
        if individual_indicator in current_name:
            individual[current_name].append(l.strip())
        else:
            item = l[1:].strip()
            item_count = 1

            tokens = item.split(" ")
            
            if len(tokens) > 1:
                if tokens[-1].isnumeric():
                    item_count =  int(tokens[-1])
                    item = " ".join(tokens[:-1])

            for i in range(item_count):
                by_name[current_name].append(item)
                by_item[item].append(current_name)
                count += 1

print(f"สรุป: {count}")
for item, names in by_item.items():
    print(f"{item} {len(names)}")

print()
print("---")
print()

for item, names in by_item.items():
    print(f"{item} {len(names)} => {names}")

print()
print("---")
print("ไปรับเองที่บ้่านป้า")
print("---")
print("")
for name, s in individual.items():
    print(f"{name}")
    for l in s:
        print(f"{l}")