From 4d93c08e447f8d1d8d2a3206267856014adf4de7 Mon Sep 17 00:00:00 2001 From: alyx Date: Fri, 24 Nov 2023 02:22:27 -0500 Subject: Wow, I forgot how to write python lmao --- src/bbss/lists.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'src/bbss/lists.py') diff --git a/src/bbss/lists.py b/src/bbss/lists.py index 3d4779e..234d0f8 100644 --- a/src/bbss/lists.py +++ b/src/bbss/lists.py @@ -1,11 +1,12 @@ from dataclasses import dataclass from collections.abc import Iterable, Iterator +from typing import Optional import requests @dataclass(frozen=True) class ListFileEntry: entry: str - comment: str + comment: Optional[str] class ListFileIterator(Iterator[ListFileEntry]): def __init__(self, entries: list[ListFileEntry]): @@ -23,13 +24,13 @@ class ListFile(Iterable[ListFileEntry]): self._contents = contents acc = [] comment_acc = "" - for line in str.splitlines(): + for line in contents.splitlines(): if line.startswith("#"): continue elif line.startswith("##"): - comment_acc += line[2:].split() + comment_acc += line[2:].strip() else: - acc += ListFileEntry(comment = comment_acc if comment_acc else None, entry = line) + acc.append(ListFileEntry(comment = comment_acc if comment_acc else None, entry = line)) self._entries = acc -- cgit v1.2.3-54-g00ecf