diff --git a/src/webqueue2api/parser/queue.py b/src/webqueue2api/parser/queue.py index 9351591..5276b47 100644 --- a/src/webqueue2api/parser/queue.py +++ b/src/webqueue2api/parser/queue.py @@ -26,21 +26,24 @@ class Queue: QueueDoesNotExistError: If a queue's directory does not exist on the filesystem. """ - def __init__(self, name: str) -> None: + def __init__(self, name: str, headers_only: bool = True) -> None: self.name = name self.path = Path(config.queue_directory, self.name) if not self.path.exists(): raise QueueDoesNotExistError(str(self.path)) - self.items = self.get_items() + self.items = self.get_items(headers_only=headers_only) self.json_data = { "name": self.name, } - def get_items(self) -> list: + def get_items(self, headers_only: bool) -> list: """Returns a list of items for this Queue + Args: + headers_only (bool): If True, loads Item headers. + Returns: list: a list of items for this Queue """ @@ -52,7 +55,7 @@ def get_items(self) -> list: is_file = True if os.path.isfile(item_path) else False if is_file and is_valid_item_name(item): - items.append(Item(self.name, item)) + items.append(Item(self.name, item, headers_only)) return items @@ -92,6 +95,7 @@ def is_valid_item_name(name: str) -> bool: Args: name (str): The name to test. + headers_only (bool, optional): Whether or not to parse headers only. Defaults to True. Returns: bool: Name is valid item name.