Calculate total amount of products in middle shelves from [REDACTED] SH job

sum-middle-shelves.py

from hubstorage import HubstorageClient

hs = HubstorageClient('<API_KEY>')

class Shelf():
    def __init__(self):
        self.children = defaultdict(Shelf)
        self.products = 0
    def __iter__(self):
        for child in self.children.values():
            for item in child:
                yield item
        yield self.products

category_tree = Shelf()

for item in hs.get_job('3784/5/386').items.iter_values():
    top = category_tree
    for shelf in item['category'].split(' > '):
        top.products = min(top.products + 1, 24 * 400)
        top = top.children[shelf]

sum(category_tree)