Question

I have this data:

inventory = { HAMMER: (10,100),
              SCREW: (1, 1000),
              NAIL: (1, 1000),
              SCREWDRIVER: (8, 100),
              DRILL: (50, 20),
              WORKBENCH: (150, 5),
              HANDSAW: (15, 50),
              CHAINSAW: (80, 30)
            }

Which I have written the following function:

def get_items(cheapness):
    """ Return a list of (item, (price, count)) tuples that are the given
    cheapness. Note that the second element of the tuple is another tuple. """

    if cheapness == 'CHEAP':
        return [(key, value) for (key,value) in inventory.items() if value in ((cost,quantity) for (cost, quantity) in inventory.values() if cost < 20)]
    elif cheapness == 'MODERATE':
        return [(key, value) for (key,value) in inventory.items() if value in ((cost,quantity) for (cost, quantity) in inventory.values() if 20 < cost < 100)]
    elif cheapness == 'EXPENSIVE':
        return [(key, value) for (key,value) in inventory.items() if value in ((cost,quantity) for (cost, quantity) in inventory.values() if cost > 100)]

My problem is when I do a print type(get_items(CHEAP)), I get <type 'NoneType'>. How do I return the results as a list?

Answer 1

Say what you mean:

def get_items(cheapness):
    """ Return a list of (item, (price, count)) tuples that are the given
    cheapness. Note that the second element of the tuple is another tuple. """

    if cheapness == 'CHEAP':
        return [(item, (price,count)) for (item, (price,count)) in inventory.items() if price < 20]
    elif cheapness == 'MODERATE':
        return [(item, (price,count)) for (item, (price,count)) in inventory.items() if price > 20 and price < 100]
    elif cheapness == 'EXPENSIVE':
        return [(item, (price,count)) for (item, (price,count)) in inventory.items() if price > 100]