Skip to content

Commit

Permalink
Add getQueueCounts and loadQueues functions to utilities
Browse files Browse the repository at this point in the history
  • Loading branch information
campb303 committed Jun 2, 2021
1 parent 45813b6 commit 487f9df
Showing 1 changed file with 44 additions and 1 deletion.
45 changes: 44 additions & 1 deletion src/webqueue2api/parser/utilities.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,4 +46,47 @@ def getValidQueues() -> list:
if isDirectory and isValid:
queues.append(file)

return queues
return queues

def getQueueCounts() -> list:
"""Returns a list of dictionaries with the number of items in each queue.
Example:
[
{
name: "me",
number_of_items: 42
},
{
name: "bidc",
number_of_items: 3
}
]
Returns:
list: Dictionaries with the number of items in each queue.
"""
queueInfo = []
for queue in getValidQueues():
possibleItems = os.listdir(queueDirectory + "/" + queue)
validItems = [isValidItemName for file in possibleItems]
queueInfo.append( {"name": queue, "number_of_items": len(validItems)} )

# Sorts list of queue info alphabetically
sortedQueueInfo = sorted(queueInfo, key = lambda queueInfoList: queueInfoList['name'])

return sortedQueueInfo


def loadQueues() -> list:
"""Return a list of Queues for each queue.
Returns:
list: list of Queues for each queue.
"""
queues = []

for queue in getValidQueues():
queues.append(Queue(queue))

return queuesG

0 comments on commit 487f9df

Please sign in to comment.