Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from collections import deque
- def stock_availability(list_of_boxes, command, *args):
- if command == 'delivery':
- list_of_boxes.extend(args[:])
- elif command == 'sell':
- list_of_boxes = deque(list_of_boxes)
- if not args:
- list_of_boxes.popleft()
- elif type(args[0]) == int:
- for _ in range(args[0]):
- if list_of_boxes:
- list_of_boxes.popleft()
- else:
- for element in args:
- while element in list_of_boxes:
- list_of_boxes.remove(element)
- return list(list_of_boxes)
- print(stock_availability(["choco", "vanilla", "banana"], "delivery", "caramel", "berry"))
- print(stock_availability(["chocolate", "vanilla", "banana"], "delivery", "cookie","banana"))
- print(stock_availability(["chocolate", "vanilla", "banana"], "sell"))
- print(stock_availability(["chocolate", "vanilla", "banana"], "sell", 3))
- print(stock_availability(["chocolate", "chocolate", "banana"], "sell", "chocolate"))
- print(stock_availability(["cookie", "chocolate", "banana"], "sell", "chocolate"))
- print(stock_availability(["chocolate", "vanilla", "banana"], "sell", "cookie"))
Advertisement
Add Comment
Please, Sign In to add comment